diff --git a/cmake/nccl.cmake b/cmake/nccl.cmake index be84c54fd2fa1b1b16153000715ea453a10aeeef..9124fec0b856a6c46001da3c735454f9aff5493f 100644 --- a/cmake/nccl.cmake +++ b/cmake/nccl.cmake @@ -7,14 +7,14 @@ if(WIN32) return() endif() -set(NCCL_ROOT "/usr" CACHE PATH "NCCL ROOT") -find_path(NCCL_INCLUDE_DIR nccl.h - PATHS ${NCCL_ROOT} ${NCCL_ROOT}/include ${NCCL_ROOT}/local/include - $ENV{NCCL_ROOT} $ENV{NCCL_ROOT}/include $ENV{NCCL_ROOT}/local/include - NO_DEFAULT_PATH -) - if(WITH_NCCL) + set(NCCL_ROOT "/usr" CACHE PATH "NCCL ROOT") + find_path(NCCL_INCLUDE_DIR nccl.h + PATHS ${NCCL_ROOT} ${NCCL_ROOT}/include ${NCCL_ROOT}/local/include + $ENV{NCCL_ROOT} $ENV{NCCL_ROOT}/include $ENV{NCCL_ROOT}/local/include + NO_DEFAULT_PATH + ) + file(READ ${NCCL_INCLUDE_DIR}/nccl.h NCCL_VERSION_FILE_CONTENTS) string(REGEX MATCH "define NCCL_VERSION_CODE +([0-9]+)"