echodict/llama.cpp

1# cmake/FindNCCL.cmake2 3# NVIDIA does not distribute CMake files with NCCl, therefore use this file to find it instead.4 5find_path(NCCL_INCLUDE_DIR6    NAMES nccl.h7    HINTS ${NCCL_ROOT} $ENV{NCCL_ROOT} $ENV{CUDA_HOME} /usr/local/cuda8    PATH_SUFFIXES include9)10 11find_library(NCCL_LIBRARY12    NAMES nccl13    HINTS ${NCCL_ROOT} $ENV{NCCL_ROOT} $ENV{CUDA_HOME} /usr/local/cuda14    PATH_SUFFIXES lib lib6415)16 17include(FindPackageHandleStandardArgs)18find_package_handle_standard_args(NCCL19    DEFAULT_MSG20    NCCL_LIBRARY NCCL_INCLUDE_DIR21)22 23if(NCCL_FOUND)24    set(NCCL_LIBRARIES ${NCCL_LIBRARY})25    set(NCCL_INCLUDE_DIRS ${NCCL_INCLUDE_DIR})26 27    if(NOT TARGET NCCL::NCCL)28        add_library(NCCL::NCCL UNKNOWN IMPORTED)29        set_target_properties(NCCL::NCCL PROPERTIES30            IMPORTED_LOCATION "${NCCL_LIBRARY}"31            INTERFACE_INCLUDE_DIRECTORIES "${NCCL_INCLUDE_DIR}"32        )33    endif()34endif()35 36mark_as_advanced(NCCL_INCLUDE_DIR NCCL_LIBRARY)37