From cc156214dbe5537d0f2a0e0c6e6b0aba233907b3 Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Mon, 6 Nov 2023 18:19:21 -0500 Subject: [PATCH] respect user defined CUDAARCHS Although we set it to `all` by default, one may manually set it via [CMAKE_CUDA_ARCHITECTURES](https://cmake.org/cmake/help/latest/variable/CMAKE_CUDA_ARCHITECTURES.html) or [CUDAARCHS](https://cmake.org/cmake/help/latest/envvar/CUDAARCHS.html#envvar:CUDAARCHS). Signed-off-by: Jinzhe Zeng --- source/lib/src/gpu/CMakeLists.txt | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/source/lib/src/gpu/CMakeLists.txt b/source/lib/src/gpu/CMakeLists.txt index c78da978a2..5eb833e24d 100644 --- a/source/lib/src/gpu/CMakeLists.txt +++ b/source/lib/src/gpu/CMakeLists.txt @@ -5,7 +5,9 @@ if(USE_CUDA_TOOLKIT) project(deepmd_op_cuda) set(GPU_LIB_NAME deepmd_op_cuda) - set(CMAKE_CUDA_ARCHITECTURES all) + if(NOT DEFINED CMAKE_CUDA_ARCHITECTURES) + set(CMAKE_CUDA_ARCHITECTURES all) + endif() enable_language(CUDA) set(CMAKE_CUDA_STANDARD 11) add_compile_definitions(