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

Co-authored-by: Thomas Wang <1am9trash@gmail.com>

2025-12-18 23:36:03 -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

…

custom_all_reduce.py

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

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

hpu_communicator.py

…

npu_communicator.py

…

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

Revert PR #14044: Restore separate memory pool for piecewise CUDA graph (#14278)

2025-12-02 09:53:16 -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

…