| set(NCCL_INC_PATHS | |
| /usr/include | |
| /usr/local/include | |
| $ENV{NCCL_DIR}/include | |
| ) | |
| set(NCCL_LIB_PATHS | |
| /lib | |
| /lib64 | |
| /usr/lib | |
| /usr/lib64 | |
| /usr/local/lib | |
| /usr/local/lib64 | |
| $ENV{NCCL_DIR}/lib | |
| ) | |
| find_path(NCCL_INCLUDE_DIR NAMES nccl.h PATHS ${NCCL_INC_PATHS}) | |
| find_library(NCCL_LIBRARIES NAMES nccl PATHS ${NCCL_LIB_PATHS}) | |
| include(FindPackageHandleStandardArgs) | |
| find_package_handle_standard_args(NCCL DEFAULT_MSG NCCL_INCLUDE_DIR NCCL_LIBRARIES) | |
| if (NCCL_FOUND) | |
| message(STATUS "Found NCCL (include: ${NCCL_INCLUDE_DIR}, library: ${NCCL_LIBRARIES})") | |
| mark_as_advanced(NCCL_INCLUDE_DIR NCCL_LIBRARIES) | |
| endif () | |