From ea188e5ad4789a780ab398111eff55ac7d72e5f4 Mon Sep 17 00:00:00 2001 From: kradchen Date: Wed, 18 Dec 2024 13:16:53 +0800 Subject: [PATCH] feat: Add new CUDA find logic to CMAKE file. --- CMakeLists.txt | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index b6469cb..4d95219 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -8,9 +8,10 @@ set(Aurora_USE_CUDA ON) if (Aurora_USE_CUDA) set(CMAKE_CUDA_COMPILER /usr/local/cuda/bin/nvcc) - +set(CUDA_TOOLKIT_ROOT /usr/local/cuda) enable_language(CUDA) +find_package(CUDA) find_package(CUDAToolkit REQUIRED) add_definitions(-DUSE_CUDA) endif(Aurora_USE_CUDA) @@ -43,12 +44,12 @@ target_link_libraries(Aurora PUBLIC $) target_link_libraries(Aurora PUBLIC OpenMP::OpenMP_CXX) target_link_libraries(Aurora PUBLIC matio) if (Aurora_USE_CUDA) -target_include_directories(Aurora PRIVATE ./src /usr/local/cuda/include) +target_include_directories(Aurora PRIVATE ./src ${CUDA_INCLUDE_DIRS}) set_target_properties(Aurora PROPERTIES CUDA_SEPARABLE_COMPILATION ON) target_compile_options(Aurora PRIVATE $<$: -arch=sm_75 --expt-extended-lambda >) -target_link_libraries(Aurora PRIVATE ${CUDA_RUNTIME_LIBRARY} CUDA::cufft CUDA::cudart) +target_link_libraries(Aurora PRIVATE ${CUDA_RUNTIME_LIBRARY} ${CUDA_cufft_LIBRARY}) target_link_libraries(Aurora PRIVATE ${CUDA_cublas_LIBRARY}) target_link_libraries(Aurora PRIVATE ${CUDA_cusolver_LIBRARY}) endif(Aurora_USE_CUDA)