mirror of
https://github.com/ml-explore/mlx.git
synced 2025-06-28 12:13:21 +08:00
50dfb664db
* Start the communications branch using MPI * Add ops and primitives * Add python bindings for distributed
17 lines
311 B
CMake
17 lines
311 B
CMake
target_sources(
|
|
mlx
|
|
PRIVATE
|
|
${CMAKE_CURRENT_SOURCE_DIR}/primitives.cpp
|
|
${CMAKE_CURRENT_SOURCE_DIR}/ops.cpp
|
|
)
|
|
|
|
if (MPI_FOUND AND MLX_BUILD_CPU)
|
|
add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/mpi)
|
|
else()
|
|
target_sources(
|
|
mlx
|
|
PRIVATE
|
|
${CMAKE_CURRENT_SOURCE_DIR}/no_distributed.cpp
|
|
)
|
|
endif()
|