diff options
author | Marcin Junczys-Dowmunt <junczys@amu.edu.pl> | 2018-08-06 08:08:44 +0300 |
---|---|---|
committer | Marcin Junczys-Dowmunt <junczys@amu.edu.pl> | 2018-08-06 08:08:44 +0300 |
commit | d96c03f7cb599f4b1d7ccdfe05e13085242503c9 (patch) | |
tree | d1fee6af12abf033b1557b51a97df0765195649d | |
parent | fedab0dc906859806e528c649f61431d7d873ce1 (diff) |
find static NCCL
-rw-r--r-- | cmake/FindNCCL.cmake | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/cmake/FindNCCL.cmake b/cmake/FindNCCL.cmake index ab3c55a8..d6100e85 100644 --- a/cmake/FindNCCL.cmake +++ b/cmake/FindNCCL.cmake @@ -16,10 +16,19 @@ set(NCCL_LIB_PATHS /usr/local/cuda/lib64 $ENV{NCCL_DIR}/lib64 $ENV{CUDA_TOOLKIT_ROOT_DIR}/lib64 + /usr/local/cuda/lib + $ENV{NCCL_DIR}/lib + $ENV{CUDA_TOOLKIT_ROOT_DIR}/lib ) find_path(NCCL_INCLUDE_DIR NAMES nccl.h PATHS ${NCCL_INC_PATHS}) -find_library(NCCL_LIBRARIES NAMES nccl PATHS ${NCCL_LIB_PATHS}) + +if (USE_STATIC_LIBS) + message(STATUS "Trying to find static NCCL library") + find_library(NCCL_LIBRARIES NAMES libnccl_static.a PATHS ${NCCL_LIB_PATHS}) +else (USE_STATIC_LIBS) + find_library(NCCL_LIBRARIES NAMES nccl PATHS ${NCCL_LIB_PATHS}) +endif (USE_STATIC_LIBS) include(FindPackageHandleStandardArgs) find_package_handle_standard_args(NCCL DEFAULT_MSG NCCL_INCLUDE_DIR NCCL_LIBRARIES) |