diff --git a/src/wrapper/wrap_cudadrv.cpp b/src/wrapper/wrap_cudadrv.cpp index 58581f548076a40bbfb255399723760c70b2de8d..3758689b66d1e4b5ead8d5e9d79b0e714b64ca62 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)