Merge pull request cupy#8920 from leofang/sm120

Minor updates for sm120
chainer-ci · Feb 4, 2025 · 595382f · 595382f
1 parent aa5ae4c
commit 595382f
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/cupy_tests/fft_tests/test_callback.py b/tests/cupy_tests/fft_tests/test_callback.py
@@ -9,6 +9,12 @@
 
 import cupy
 from cupy import testing
+from cupy.cuda import cufft, Device
+
+
+pytestmark = pytest.mark.skipif(
+    cufft.getVersion() == 11303 and Device().compute_capability == '120',
+    reason="cuFFT static callbacks in CUDA 12.8.0 do not support sm120")
 
 
 @contextlib.contextmanager