Skip to content

Commit 10caa72

Browse files
committed
Make sm80 the default when compiling tests for Nvidia
1 parent 99cfff4 commit 10caa72

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

tests/CMakeLists.txt

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,13 @@ if(${KERNEL_FLOAT_LANGUAGE_CUDA})
1616

1717
target_compile_options(kernel_float_tests PRIVATE "-ftime-report -ftime-report-details")
1818
target_compile_options(kernel_float_tests PRIVATE "--extended-lambda")
19-
set_target_properties(kernel_float_tests PROPERTIES CUDA_ARCHITECTURES "all-major")
19+
20+
# Default to 80 if user hasn't specified
21+
if(NOT DEFINED CMAKE_CUDA_ARCHITECTURES)
22+
set(CMAKE_CUDA_ARCHITECTURES "80")
23+
endif()
24+
25+
set_target_properties(kernel_float_tests PROPERTIES CUDA_ARCHITECTURES "${CMAKE_CUDA_ARCHITECTURES}")
2026
endif()
2127

2228
if(${KERNEL_FLOAT_LANGUAGE_HIP})

0 commit comments

Comments
 (0)