Skip to content

Commit 85d4b0f

Browse files
committed
Move dependency discovery files to cmake/modules
1 parent 0da4186 commit 85d4b0f

19 files changed

Lines changed: 12 additions & 6 deletions

CMakeLists.txt

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -208,7 +208,9 @@ if (USE_CUDA_MPI)
208208
abacus_add_feature_definitions(__CUDA_MPI)
209209
endif()
210210

211-
list(APPEND CMAKE_MODULE_PATH ${CMAKE_CURRENT_SOURCE_DIR}/cmake)
211+
list(APPEND CMAKE_MODULE_PATH
212+
"${CMAKE_CURRENT_SOURCE_DIR}/cmake"
213+
"${PROJECT_SOURCE_DIR}/cmake/modules")
212214

213215
if(ENABLE_COVERAGE)
214216
find_package(codecov)
@@ -441,12 +443,12 @@ if(USE_CUDA)
441443
"ENABLE_NCCL_PARALLEL_DEVICE requires ENABLE_MPI=ON.")
442444
endif()
443445
abacus_add_feature_definitions(__NCCL_PARALLEL_DEVICE)
444-
include(cmake/SetupNccl.cmake)
446+
include(cmake/models/SetupNccl.cmake)
445447
abacus_setup_nccl()
446448
endif()
447449
if (ENABLE_CUSOLVERMP)
448450
# Keep cuSOLVERMp discovery/linking logic in a dedicated module.
449-
include(cmake/SetupCuSolverMp.cmake)
451+
include(cmake/modules/SetupCuSolverMp.cmake)
450452
abacus_setup_cusolvermp()
451453
endif()
452454
if (ENABLE_CUBLASMP)
@@ -464,7 +466,7 @@ if(USE_CUDA)
464466
"cuBLASMp 0.8.0+ requires NCCL Symmetric Memory, but cuSOLVERMp is using CAL backend."
465467
"Please upgrade cuSOLVERMp to >= 0.7.0 to use NCCL for both.")
466468
endif()
467-
include(cmake/SetupCuBlasMp.cmake)
469+
include(cmake/modules/SetupCuBlasMp.cmake)
468470
abacus_setup_cublasmp()
469471
endif()
470472
endif()

0 commit comments

Comments
 (0)