# Find the NCCL libraries
#
# The following variables are optionally searched for defaults
#  NCCL_ROOT_DIR:    Base directory where all NCCL components are found
#
# The following are set after configuration is done:
#  NCCL_FOUND
#  NCCL_INCLUDE_DIR
#  NCCL_LIBRARY

find_path(NCCL_INCLUDE_DIR NAMES nccl.h
        PATHS ${NCCL_ROOT_DIR}/include
)

find_library(NCCL_LIBRARY NAMES nccl
        PATHS ${NCCL_ROOT_DIR}/lib ${NCCL_ROOT_DIR}/lib64)

include(FindPackageHandleStandardArgs)
find_package_handle_standard_args(NCCL DEFAULT_MSG NCCL_INCLUDE_DIR
        NCCL_LIBRARY)

if (NCCL_FOUND)
    message(STATUS "Found NCCL (include: ${NCCL_INCLUDE_DIR}, library:
  ${NCCL_LIBRARY})")
    mark_as_advanced(NCCL_INCLUDE_DIR NCCL_LIBRARY)
    set(NCCL_VERSION "${NCCL_MAJOR}.${NCCL_MINOR}.${NCCL_PATCH}")

endif ()