Ilya Markovandilmarkov b7b7676d67 [Distributed] Add custom allreduce support for ROCM (#14125) ...

Signed-off-by: ilmarkov <imarkov@redhat.com>
Co-authored-by: ilmarkov <imarkov@redhat.com>

2025-03-31 22:49:12 -07:00

..

__init__.py

[Core][Refactor] move parallel_utils into vllm/distributed (#3950)

2024-04-10 15:33:30 -07:00

base_device_communicator.py

[platform] add base class for communicators (#13208)

2025-02-16 22:14:22 +08:00

cpu_communicator.py

[Bugfix][CPU] Fix cpu all-reduce using native pytorch implementation (#13586)

2025-02-20 22:24:17 -08:00

cuda_communicator.py

[core] set up data parallel communication (#13591)

2025-02-22 19:28:59 +08:00

cuda_wrapper.py

Simplify logic of locating CUDART so file path (#13203)

2025-02-13 13:52:41 +08:00

custom_all_reduce_utils.py

[Misc] Add SPDX-License-Identifier headers to python source files (#12628)

2025-02-02 11:58:18 -08:00

custom_all_reduce.py

[Distributed] Add custom allreduce support for ROCM (#14125)

2025-03-31 22:49:12 -07:00

hpu_communicator.py

[platform] add base class for communicators (#13208)

2025-02-16 22:14:22 +08:00

neuron_communicator.py

[Neuron] Add Neuron device communicator for vLLM v1 (#14085)

2025-03-10 18:37:04 -07:00

pynccl_wrapper.py

[Misc] set single whitespace between log sentences (#13771)

2025-02-25 10:26:12 +08:00

pynccl.py

[Misc] Add SPDX-License-Identifier headers to python source files (#12628)

2025-02-02 11:58:18 -08:00

shm_broadcast.py

[Bugfix] Fix failure to launch in Tensor Parallel TP mode on macOS. (#14948)

2025-03-28 10:13:41 +08:00

tpu_communicator.py

[V1] TPU - Tensor parallel MP support (#15059)

2025-03-20 00:55:18 +00:00

xpu_communicator.py

[Bugfix] fix xpu communicator (#13368)

2025-02-17 20:59:18 +08:00