From b2251e0deb26e782504841450840baee4b9cb1b8 Mon Sep 17 00:00:00 2001
From: Robert Maynard <rmaynard@nvidia.com>
Date: Tue, 4 Feb 2025 12:21:16 -0500
Subject: CUDA/Clang: Update architectures supported by CUDA 12.8

---
 Modules/Internal/CMakeCUDAArchitecturesAll.cmake | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/Modules/Internal/CMakeCUDAArchitecturesAll.cmake b/Modules/Internal/CMakeCUDAArchitecturesAll.cmake
index c588dfb..b97a202 100644
--- a/Modules/Internal/CMakeCUDAArchitecturesAll.cmake
+++ b/Modules/Internal/CMakeCUDAArchitecturesAll.cmake
@@ -71,6 +71,15 @@ function(cmake_cuda_architectures_all lang lang_var_)
     endif()
   endif()
 
+  if(${lang_var_}TOOLKIT_VERSION VERSION_GREATER_EQUAL 12.8)
+    if(CMAKE_${lang}_COMPILER_ID STREQUAL "NVIDIA"
+        OR (CMAKE_${lang}_COMPILER_ID STREQUAL "Clang" AND CMAKE_${lang}_COMPILER_VERSION VERSION_GREATER_EQUAL 16.0)
+        )
+      list(APPEND CMAKE_CUDA_ARCHITECTURES_ALL 100 101 120)
+      list(APPEND CMAKE_CUDA_ARCHITECTURES_ALL_MAJOR 100 120)
+    endif()
+  endif()
+
   if(${lang_var_}TOOLKIT_VERSION VERSION_GREATER_EQUAL 12.0)
     list(REMOVE_ITEM CMAKE_CUDA_ARCHITECTURES_ALL 35 37)
     list(REMOVE_ITEM CMAKE_CUDA_ARCHITECTURES_ALL_MAJOR 35)
-- 
cgit v0.12