Skip to content

Commit 90944d7

Browse files
gmarkallmanopapad
andauthored
Add ada to cuda_arch_helpers.cmake (#944)
* Add ada to cuda-arch_helpers.cmake * Also define ADA_ARCH --------- Co-authored-by: Manolis Papadakis <manopapad@gmail.com>
1 parent 8f7bff6 commit 90944d7

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

cmake/Modules/cuda_arch_helpers.cmake

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,9 @@ function(set_cuda_arch_from_names)
4444
if(Legion_CUDA_ARCH MATCHES "ampere")
4545
list(APPEND cuda_archs 80)
4646
endif()
47+
if(Legion_CUDA_ARCH MATCHES "ada")
48+
list(APPEND cuda_archs 89)
49+
endif()
4750
if(Legion_CUDA_ARCH MATCHES "hopper")
4851
list(APPEND cuda_archs 90)
4952
endif()
@@ -91,6 +94,7 @@ function(add_cuda_architecture_defines)
9194
add_def_if_arch_enabled("70" "VOLTA_ARCH")
9295
add_def_if_arch_enabled("75" "TURING_ARCH")
9396
add_def_if_arch_enabled("80" "AMPERE_ARCH")
97+
add_def_if_arch_enabled("89" "ADA_ARCH")
9498
add_def_if_arch_enabled("90" "HOPPER_ARCH")
9599

96100
set(${cuda_DEFS} ${_defs} PARENT_SCOPE)

0 commit comments

Comments
 (0)