Files
mlx/mlx/distributed/nccl/CMakeLists.txt
Anastasiia Filippova 9392fc3f88 NCCL backend (#2476)
2025-08-21 11:56:15 -07:00

9 lines
319 B
CMake

if(MLX_BUILD_CUDA)
target_sources(mlx PRIVATE ${CMAKE_CURRENT_SOURCE_DIR}/nccl.cpp)
find_package(NCCL REQUIRED)
target_link_libraries(mlx PRIVATE ${NCCL_LIBRARIES})
target_include_directories(mlx PRIVATE ${NCCL_INCLUDE_DIRS})
else()
target_sources(mlx PRIVATE ${CMAKE_CURRENT_SOURCE_DIR}/no_nccl.cpp)
endif()