mlx/mlx/distributed
Awni Hannun d2f540f4e0
Use nccl header only when nccl is not present (#2539)
* use nccl header only when nccl is not present

* larger machine for cuda build
2025-08-25 14:17:25 -07:00
..
mpi NCCL backend (#2476) 2025-08-21 11:56:15 -07:00
nccl Use nccl header only when nccl is not present (#2539) 2025-08-25 14:17:25 -07:00
ring NCCL backend (#2476) 2025-08-21 11:56:15 -07:00
CMakeLists.txt NCCL backend (#2476) 2025-08-21 11:56:15 -07:00
distributed_impl.h NCCL backend (#2476) 2025-08-21 11:56:15 -07:00
distributed.cpp nccl default for backend=any (#2528) 2025-08-22 12:24:27 -07:00
distributed.h NCCL backend (#2476) 2025-08-21 11:56:15 -07:00
ops.cpp NCCL backend (#2476) 2025-08-21 11:56:15 -07:00
ops.h Min / max reductions (#2041) 2025-04-09 23:22:20 -07:00
primitives.cpp Min / max reductions (#2041) 2025-04-09 23:22:20 -07:00
primitives.h Add Primitive::name and remove Primitive::print (#2365) 2025-07-14 14:06:35 -07:00