From 595382fcfed19aa16af0fcd438f49c2c252fa412 Mon Sep 17 00:00:00 2001
From: Kenichi Maehashi <939877+kmaehashi@users.noreply.github.com>
Date: Wed, 5 Feb 2025 00:06:26 +0900
Subject: [PATCH] Merge pull request #8920 from leofang/sm120

Minor updates for sm120
---
 tests/cupy_tests/fft_tests/test_callback.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/tests/cupy_tests/fft_tests/test_callback.py b/tests/cupy_tests/fft_tests/test_callback.py
index 4df1ef2200b..1b63cab007d 100644
--- 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