|
@@ -71,6 +71,15 @@ function(cmake_cuda_architectures_all lang lang_var_)
|
|
|
endif()
|
|
endif()
|
|
|
endif()
|
|
endif()
|
|
|
|
|
|
|
|
|
|
+ if(${lang_var_}TOOLKIT_VERSION VERSION_GREATER_EQUAL 12.8)
|
|
|
|
|
+ if(CMAKE_${lang}_COMPILER_ID STREQUAL "NVIDIA"
|
|
|
|
|
+ OR (CMAKE_${lang}_COMPILER_ID STREQUAL "Clang" AND CMAKE_${lang}_COMPILER_VERSION VERSION_GREATER_EQUAL 16.0)
|
|
|
|
|
+ )
|
|
|
|
|
+ list(APPEND CMAKE_CUDA_ARCHITECTURES_ALL 100 101 120)
|
|
|
|
|
+ list(APPEND CMAKE_CUDA_ARCHITECTURES_ALL_MAJOR 100 120)
|
|
|
|
|
+ endif()
|
|
|
|
|
+ endif()
|
|
|
|
|
+
|
|
|
if(${lang_var_}TOOLKIT_VERSION VERSION_GREATER_EQUAL 12.0)
|
|
if(${lang_var_}TOOLKIT_VERSION VERSION_GREATER_EQUAL 12.0)
|
|
|
list(REMOVE_ITEM CMAKE_CUDA_ARCHITECTURES_ALL 35 37)
|
|
list(REMOVE_ITEM CMAKE_CUDA_ARCHITECTURES_ALL 35 37)
|
|
|
list(REMOVE_ITEM CMAKE_CUDA_ARCHITECTURES_ALL_MAJOR 35)
|
|
list(REMOVE_ITEM CMAKE_CUDA_ARCHITECTURES_ALL_MAJOR 35)
|