Add ada to cuda_arch_helpers.cmake (#944)

* Add ada to cuda-arch_helpers.cmake * Also define ADA_ARCH --------- Co-authored-by: Manolis Papadakis <[email protected]>
nv-legate · May 2, 2024 · 90944d7 · 90944d7
1 parent 8f7bff6
commit 90944d7
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/cmake/Modules/cuda_arch_helpers.cmake b/cmake/Modules/cuda_arch_helpers.cmake
@@ -44,6 +44,9 @@ function(set_cuda_arch_from_names)
   if(Legion_CUDA_ARCH MATCHES "ampere")
     list(APPEND cuda_archs 80)
   endif()
+  if(Legion_CUDA_ARCH MATCHES "ada")
+    list(APPEND cuda_archs 89)
+  endif()
   if(Legion_CUDA_ARCH MATCHES "hopper")
     list(APPEND cuda_archs 90)
   endif()
@@ -91,6 +94,7 @@ function(add_cuda_architecture_defines)
   add_def_if_arch_enabled("70" "VOLTA_ARCH")
   add_def_if_arch_enabled("75" "TURING_ARCH")
   add_def_if_arch_enabled("80" "AMPERE_ARCH")
+  add_def_if_arch_enabled("89" "ADA_ARCH")
   add_def_if_arch_enabled("90" "HOPPER_ARCH")
 
   set(${cuda_DEFS} ${_defs} PARENT_SCOPE)