From 939675adc16742861a533249869d7ab93bc305c8 Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Mon, 19 Jun 2023 14:00:00 -0500 Subject: [PATCH] Drop compute target 2.x at CUDA 11.8 for AArch64 compatibility (closes gh-415) --- src/wrapper/wrap_cudadrv.cpp | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/wrapper/wrap_cudadrv.cpp b/src/wrapper/wrap_cudadrv.cpp index 58581f54..3758689b 100644 --- a/src/wrapper/wrap_cudadrv.cpp +++ b/src/wrapper/wrap_cudadrv.cpp @@ -1030,10 +1030,12 @@ BOOST_PYTHON_MODULE(_driver) .value("COMPUTE_12", CU_TARGET_COMPUTE_12) .value("COMPUTE_13", CU_TARGET_COMPUTE_13) #endif -#if (CUDAPP_CUDA_VERSION >= 3000) && CUDAPP_CUDA_VERSION < 12000 +// Aarch64 CUDA seems to have dropped these at CUDA 11.8 already. +// https://github.com/inducer/pycuda/issues/415 +#if (CUDAPP_CUDA_VERSION >= 3000) && CUDAPP_CUDA_VERSION < 11080 .value("COMPUTE_20", CU_TARGET_COMPUTE_20) #endif -#if (CUDAPP_CUDA_VERSION >= 3020) && CUDAPP_CUDA_VERSION < 12000 +#if (CUDAPP_CUDA_VERSION >= 3020) && CUDAPP_CUDA_VERSION < 11080 .value("COMPUTE_21", CU_TARGET_COMPUTE_21) #endif #if (CUDAPP_CUDA_VERSION >= 11000) -- GitLab