Yuan Luo 7bb41989fa [1/N] Optimize All Reduce - Benchmark different AR operations (#13797) ...

Co-authored-by: luoyuan.luo <luoyuan.luo@antgroup.com>

2026-01-26 22:44:13 +08:00

..

all_reduce_utils.py

Add prefix for torch symm mem (#12506)

2025-11-02 11:23:05 -08:00

cuda_wrapper.py

…

custom_all_reduce_ops.py

[amd] Add deterministic all-reduce kernel for AMD (ROCm) (#15340)

2025-12-18 23:36:03 -08:00

custom_all_reduce_utils.py

[Feature] Integrate quick allreduce and select the best allreduce implementation (#6619)

2025-07-24 20:48:42 -07:00

custom_all_reduce.py

[1/N] Optimize All Reduce - Benchmark different AR operations (#13797)

2026-01-26 22:44:13 +08:00

hpu_communicator.py

…

npu_communicator.py

[Feature] Support for Ascend NPU backend (#3853)

2025-05-06 20:32:53 -07:00

pymscclpp.py

Move custom_ops under layers; move _custom_ops.py → custom_all_reduce_ops.py (#14326)

2025-12-03 10:33:37 -08:00

pynccl_allocator.py

Add clear error message when OOM with symmetric memory (#17038)

2026-01-16 20:44:37 +08:00

pynccl_wrapper.py

[Auto Sync] Update pynccl_wrapper.py, environ.py, registry.... (20251111) (#13097)

2025-11-13 15:58:47 -08:00

pynccl.py

(1/n)support context parallel with deepseekv3.2-DSA (#12065)

2025-11-16 20:12:25 -08:00

quick_all_reduce.py

Move custom_ops under layers; move _custom_ops.py → custom_all_reduce_ops.py (#14326)

2025-12-03 10:33:37 -08:00

shm_broadcast.py

fix(tcp-port): replace bind_server_socket to get_zmq_socket(Port conflict) (#11961)

2025-11-12 18:05:16 +08:00

torch_symm_mem.py

Add prefix for torch symm mem (#12506)

2025-11-02 11:23:05 -08:00

xpu_communicator.py

…