Chengji Yao 01b6113659 [TPU] optimize the all-reduce performance (#15903) ...

Signed-off-by: Chengji Yao <chengjiyao@google.com>

2025-04-03 00:25:14 +00: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

[CPU][Bugfix] Using custom allreduce for CPU backend (#15934)

2025-04-02 07:46:47 -07: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

[TPU] optimize the all-reduce performance (#15903)

2025-04-03 00:25:14 +00:00

xpu_communicator.py

[Bugfix] fix xpu communicator (#13368)

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