marian-nmt/marian
1
1
Fork 0
mirror of https://github.com/marian-nmt/marian.git synced 2024-09-11 06:15:56 +03:00
Code Issues Packages Projects Releases Wiki Activity

find static NCCL

Browse Source
This commit is contained in:
Marcin Junczys-Dowmunt 2018-08-05 22:08:44 -07:00
parent fedab0dc90
commit d96c03f7cb
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
cmake/FindNCCL.cmake
View File

@ -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)