mirror of
https://github.com/marian-nmt/marian.git
synced 2024-09-11 06:15:56 +03:00
find static NCCL
This commit is contained in:
parent
fedab0dc90
commit
d96c03f7cb
@ -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)
|
||||
|
Loading…
Reference in New Issue
Block a user