From b3fc6a859ca43ef3eb09f6edb7685fe4b521e9bd Mon Sep 17 00:00:00 2001 From: Robert Maynard Date: Thu, 23 Jan 2025 10:37:03 -0500 Subject: [PATCH] Add support for new architectures in CUDA 12.8 --- rapids-cmake/cuda/set_architectures.cmake | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/rapids-cmake/cuda/set_architectures.cmake b/rapids-cmake/cuda/set_architectures.cmake index e414c5a8..bc8ee253 100644 --- a/rapids-cmake/cuda/set_architectures.cmake +++ b/rapids-cmake/cuda/set_architectures.cmake @@ -1,5 +1,5 @@ #============================================================================= -# Copyright (c) 2021-2024, NVIDIA CORPORATION. +# Copyright (c) 2021-2025, NVIDIA CORPORATION. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. @@ -56,7 +56,7 @@ Result Variables function(rapids_cuda_set_architectures mode) list(APPEND CMAKE_MESSAGE_CONTEXT "rapids.cuda.set_architectures") - set(supported_archs "70" "75" "80" "86" "90") + set(supported_archs "70" "75" "80" "86" "90" "100" "120") if(CMAKE_CUDA_COMPILER_ID STREQUAL "NVIDIA" AND CMAKE_CUDA_COMPILER_VERSION VERSION_LESS 11.1.0) list(REMOVE_ITEM supported_archs "86") @@ -64,6 +64,10 @@ function(rapids_cuda_set_architectures mode) if(CMAKE_CUDA_COMPILER_ID STREQUAL "NVIDIA" AND CMAKE_CUDA_COMPILER_VERSION VERSION_LESS 11.8.0) list(REMOVE_ITEM supported_archs "90") endif() + if(CMAKE_CUDA_COMPILER_ID STREQUAL "NVIDIA" AND CMAKE_CUDA_COMPILER_VERSION VERSION_LESS 12.8.0) + list(REMOVE_ITEM supported_archs "120") + list(REMOVE_ITEM supported_archs "100") + endif() if(${mode} STREQUAL "RAPIDS")