Merge pull request #8200 from leofang/jitify_cc70

Fix jitify warmup kernel
cupy · Feb 28, 2024 · 20ccd63 · 20ccd63
2 parents 2fcc6d4 + 4f06252
commit 20ccd63
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/cupy/cuda/jitify.pyx b/cupy/cuda/jitify.pyx
@@ -42,7 +42,7 @@ cdef extern from 'cupy_jitify.h' namespace "jitify::detail" nogil:
 # users.
 cdef extern from *:
     """
-    const int build_num = 2;
+    const int build_num = 3;
     """
     const int build_num
 
@@ -139,7 +139,10 @@ cdef str warmup_kernel = r"""cupy_jitify_exercise
 
 #include <cub/block/block_reduce.cuh>
 #include <cub/block/block_load.cuh>
+#if defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 700
+// not supported before CC 7.0
 #include <cuda/barrier>
+#endif
 #include <cooperative_groups.h>
 #include <cooperative_groups/memcpy_async.h>