Skip to content

Commit 939675a

Browse files
committed
Drop compute target 2.x at CUDA 11.8 for AArch64 compatibility (closes gh-415)
1 parent 6b0fe7a commit 939675a

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

src/wrapper/wrap_cudadrv.cpp

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1030,10 +1030,12 @@ BOOST_PYTHON_MODULE(_driver)
10301030
.value("COMPUTE_12", CU_TARGET_COMPUTE_12)
10311031
.value("COMPUTE_13", CU_TARGET_COMPUTE_13)
10321032
#endif
1033-
#if (CUDAPP_CUDA_VERSION >= 3000) && CUDAPP_CUDA_VERSION < 12000
1033+
// Aarch64 CUDA seems to have dropped these at CUDA 11.8 already.
1034+
// https://github.com/inducer/pycuda/issues/415
1035+
#if (CUDAPP_CUDA_VERSION >= 3000) && CUDAPP_CUDA_VERSION < 11080
10341036
.value("COMPUTE_20", CU_TARGET_COMPUTE_20)
10351037
#endif
1036-
#if (CUDAPP_CUDA_VERSION >= 3020) && CUDAPP_CUDA_VERSION < 12000
1038+
#if (CUDAPP_CUDA_VERSION >= 3020) && CUDAPP_CUDA_VERSION < 11080
10371039
.value("COMPUTE_21", CU_TARGET_COMPUTE_21)
10381040
#endif
10391041
#if (CUDAPP_CUDA_VERSION >= 11000)

0 commit comments

Comments
 (0)