Procházet zdrojové kódy

Merge topic 'cuda_add_12.8_new_sm_support'

a745b6869e CUDA/Clang: Update architectures supported by CUDA 12.8

Acked-by: Kitware Robot <[email protected]>
Merge-request: !10289
Brad King před 10 měsíci
rodič
revize
bf47158d50

+ 9 - 0
Modules/Internal/CMakeCUDAArchitecturesAll.cmake

@@ -71,6 +71,15 @@ function(cmake_cuda_architectures_all lang lang_var_)
     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)
     list(REMOVE_ITEM CMAKE_CUDA_ARCHITECTURES_ALL 35 37)
     list(REMOVE_ITEM CMAKE_CUDA_ARCHITECTURES_ALL_MAJOR 35)