Anastasiia Filippova 27778156dc Nccl reduce scatter, all gather (#2727) ...

* Added reduce scatter and all gather for nccl

* fix unused import, delete unused file

* small fix

* deleted useless condition

* fixed comments

* fix bug in eval_gpu, renamed to sum_scatter, fix docs

* final fix docs

* remove and

* Update mlx/distributed/mpi/mpi.cpp

Co-authored-by: Awni Hannun <awni.hannun@gmail.com>

* fix broken set input output

* fixes set output

* typo

* fix typo

* no cpu, no gpu for reduce scatter

---------

Co-authored-by: Awni Hannun <awni.hannun@gmail.com>

2025-11-05 08:21:11 -08:00

..

mlx

only build for macos 14 and up (#2731)

2025-11-04 09:44:15 -08:00

scripts

nccl dep + default for cuda (#2526)

2025-08-21 17:57:49 -07:00

src

Nccl reduce scatter, all gather (#2727)

2025-11-05 08:21:11 -08:00

tests

Nccl reduce scatter, all gather (#2727)

2025-11-05 08:21:11 -08:00