From 3b193321dd417cda8757a613e0b08e5d8d0b7759 Mon Sep 17 00:00:00 2001 From: Philip Hyunsu Cho Date: Fri, 31 Jan 2025 12:07:20 -0800 Subject: [PATCH] [backport] When building with CUDA 12.8+ enable blackwell support (#11187) (#11191) Co-authored-by: Robert Maynard --- cmake/Utils.cmake | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/cmake/Utils.cmake b/cmake/Utils.cmake index b12302a166c0..4e66c15186eb 100644 --- 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)