Skip to content

Commit 34bcd7f

Browse files
When building with CUDA 12.8+ enable blackwell support (dmlc#11187)
1 parent 231977f commit 34bcd7f

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

cmake/Utils.cmake

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,9 @@ function(compute_cmake_cuda_archs archs)
5555

5656
# Set up defaults based on CUDA varsion
5757
if(NOT CMAKE_CUDA_ARCHITECTURES)
58-
if(CUDA_VERSION VERSION_GREATER_EQUAL "11.8")
58+
if(CUDA_VERSION VERSION_GREATER_EQUAL "12.8")
59+
set(CMAKE_CUDA_ARCHITECTURES 50 60 70 80 90 100 120)
60+
elseif(CUDA_VERSION VERSION_GREATER_EQUAL "11.8")
5961
set(CMAKE_CUDA_ARCHITECTURES 50 60 70 80 90)
6062
elseif(CUDA_VERSION VERSION_GREATER_EQUAL "11.0")
6163
set(CMAKE_CUDA_ARCHITECTURES 50 60 70 80)

0 commit comments

Comments
 (0)