Jelajahi Sumber

Merge topic 'cuda_ensure_arch_order' into release-4.0

8beb309b79 CUDA: Make sure CUDA architectures are in order

Acked-by: Kitware Robot <[email protected]>
Merge-request: !11014
Brad King 7 bulan lalu
induk
melakukan
1ff9ebe222
1 mengubah file dengan 3 tambahan dan 2 penghapusan
  1. 3 2
      Modules/Internal/CMakeCUDAArchitecturesAll.cmake

+ 3 - 2
Modules/Internal/CMakeCUDAArchitecturesAll.cmake

@@ -76,7 +76,7 @@ function(cmake_cuda_architectures_all lang lang_var_)
     list(REMOVE_ITEM CMAKE_CUDA_ARCHITECTURES_ALL_MAJOR 35)
   endif()
 
-  if(${lang_var_}TOOLKIT_VERSION VERSION_GREATER_EQUAL 12.8)
+  if(${lang_var_}TOOLKIT_VERSION VERSION_GREATER_EQUAL 12.8 AND ${lang_var_}TOOLKIT_VERSION VERSION_LESS 12.9)
     if(CMAKE_${lang}_COMPILER_ID STREQUAL "NVIDIA"
         OR (CMAKE_${lang}_COMPILER_ID STREQUAL "Clang" AND CMAKE_${lang}_COMPILER_VERSION VERSION_GREATER_EQUAL 20.1)
         )
@@ -89,7 +89,8 @@ function(cmake_cuda_architectures_all lang lang_var_)
     if(CMAKE_${lang}_COMPILER_ID STREQUAL "NVIDIA"
         OR (CMAKE_${lang}_COMPILER_ID STREQUAL "Clang" AND CMAKE_${lang}_COMPILER_VERSION VERSION_GREATER_EQUAL 21.1)
         )
-      list(APPEND CMAKE_CUDA_ARCHITECTURES_ALL 103 121)
+      list(APPEND CMAKE_CUDA_ARCHITECTURES_ALL 100 101 103 120 121)
+      list(APPEND CMAKE_CUDA_ARCHITECTURES_ALL_MAJOR 100 120)
     endif()
   endif()