Siyuan Liu 48ac2bed5b

[Hardware][TPU] Optionally import for TPU backend (#18269) ...

Signed-off-by: Siyuan Liu <lsiyuan@google.com>
Signed-off-by: Jade Zheng <zheng.shoujian@outlook.com>
Co-authored-by: Carol Zheng <cazheng@google.com>
Co-authored-by: Jade Zheng <zheng.shoujian@outlook.com>
Co-authored-by: Hongmin Fan <fanhongmin@google.com>

2025-05-17 15:23:12 +08:00

..

__init__.py

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

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

all2all.py

[core][distributed] add ep group and all2all interface (#18077)

2025-05-14 10:46:49 +08:00

base_device_communicator.py

Update deprecated type hinting in vllm/device_allocator and vllm/distributed (#18126)

2025-05-14 04:07:57 -07:00

cpu_communicator.py

Update deprecated type hinting in vllm/device_allocator and vllm/distributed (#18126)

2025-05-14 04:07:57 -07:00

cuda_communicator.py

Update deprecated type hinting in vllm/device_allocator and vllm/distributed (#18126)

2025-05-14 04:07:57 -07:00

cuda_wrapper.py

Update deprecated type hinting in vllm/device_allocator and vllm/distributed (#18126)

2025-05-14 04:07:57 -07:00

custom_all_reduce_utils.py

Update deprecated type hinting in vllm/device_allocator and vllm/distributed (#18126)

2025-05-14 04:07:57 -07:00

custom_all_reduce.py

[Feature] Support Pipeline Parallism in torchrun SPMD offline inference for V1 (#17827)

2025-05-15 22:28:27 -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

Update deprecated type hinting in vllm/device_allocator and vllm/distributed (#18126)

2025-05-14 04:07:57 -07: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

Update deprecated type hinting in vllm/device_allocator and vllm/distributed (#18126)

2025-05-14 04:07:57 -07:00

tpu_communicator.py

[Hardware][TPU] Optionally import for TPU backend (#18269)

2025-05-17 15:23:12 +08:00

xpu_communicator.py

[Bugfix] fix xpu communicator (#13368)

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