[backport] When building with CUDA 12.8+ enable blackwell support (#1…

…1187) (#11191) Co-authored-by: Robert Maynard <[email protected]>
dmlc · Jan 31, 2025 · 3b19332 · 3b19332
1 parent 600be4d
commit 3b19332
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/cmake/Utils.cmake b/cmake/Utils.cmake
@@ -55,7 +55,9 @@ function(compute_cmake_cuda_archs archs)
 
   # Set up defaults based on CUDA varsion
   if(NOT CMAKE_CUDA_ARCHITECTURES)
-    if(CUDA_VERSION VERSION_GREATER_EQUAL "11.8")
+    if(CUDA_VERSION VERSION_GREATER_EQUAL "12.8")
+      set(CMAKE_CUDA_ARCHITECTURES 50 60 70 80 90 100 120)
+    elseif(CUDA_VERSION VERSION_GREATER_EQUAL "11.8")
       set(CMAKE_CUDA_ARCHITECTURES 50 60 70 80 90)
     elseif(CUDA_VERSION VERSION_GREATER_EQUAL "11.0")
       set(CMAKE_CUDA_ARCHITECTURES 50 60 70 80)