diff --git a/cmake/Modules/ConfigureCuda.cmake b/cmake/Modules/ConfigureCuda.cmake index 32013a0735da..0d650f05a1ab 100644 --- a/cmake/Modules/ConfigureCuda.cmake +++ b/cmake/Modules/ConfigureCuda.cmake @@ -32,7 +32,13 @@ IF(NOT CUDAToolkit_FOUND AND NOT CUDA_FOUND) ENDIF() IF(NOT DEFINED CMAKE_CUDA_ARCHITECTURES) - set(CMAKE_CUDA_ARCHITECTURES 52-real 60-real 61-real 70-real 75) + IF(CUDAToolkit_VERSION VERSION_GREATER_EQUAL 13) + set(CMAKE_CUDA_ARCHITECTURES 75-real 86-real 89-real 120) + ELSEIF(CUDAToolkit_VERSION VERSION_GREATER_EQUAL 12) + set(CMAKE_CUDA_ARCHITECTURES 52-real 60-real 61-real 70-real 75-real 86-real 89) + ELSE() + set(CMAKE_CUDA_ARCHITECTURES 52-real 60-real 61-real 70-real 75) + ENDIF() ENDIF() set(CUDART_LIBRARY CUDA::cudart_static)