diff --git a/cmake/Modules/CompilerHelpers.cmake b/cmake/Modules/CompilerHelpers.cmake index 876778007..755fcdb3e 100644 --- a/cmake/Modules/CompilerHelpers.cmake +++ b/cmake/Modules/CompilerHelpers.cmake @@ -47,6 +47,7 @@ function(cuda_archs_to_gpu_list archs out_var) set(_gpus) foreach(_arch IN LISTS archs) string(REGEX REPLACE "-(real|virtual)$" "" _arch "${_arch}") + string(REGEX REPLACE "^cc" "" _arch "${_arch}") string(PREPEND _arch "cc") list(APPEND _gpus "${_arch}") endforeach()