Fix CI for sm90

NVIDIA · Mar 15, 2023 · 9b00558 · 9b00558
1 parent e6c34e0
commit 9b00558
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/cmake/ThrustCudaConfig.cmake b/cmake/ThrustCudaConfig.cmake
@@ -1,6 +1,17 @@
 enable_language(CUDA)
 
-set(THRUST_KNOWN_COMPUTE_ARCHS 35 37 50 52 53 60 61 62 70 72 75 80 86 90)
+set(THRUST_KNOWN_COMPUTE_ARCHS 50 52 53 60 61 62 70 72 75 80 86)
+
+if (NVIDIA STREQUAL "${CMAKE_CUDA_COMPILER_ID}")
+  if (CMAKE_CUDA_COMPILER_VERSION VERSION_GREATER 11.7)
+    list(APPEND THRUST_KNOWN_COMPUTE_ARCHS 90)
+  endif()
+  if (CMAKE_CUDA_COMPILER_VERSION VERSION_LESS 12.0)
+    list(APPEND THRUST_KNOWN_COMPUTE_ARCHS 35 37)
+  endif()
+else()
+  list(APPEND THRUST_KNOWN_COMPUTE_ARCHS 35 37 90)
+endif()
 
 # Split CUDA_FLAGS into 3 parts:
 #