Chengji YaoandChengji Yao e9d6a3db69 [TPU] make ptxla not imported when using tpu_commons (#23081) ...

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

2025-08-19 11:46:42 +08:00

..

__init__.py

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

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

all2all.py

[Refactor] Create a function util and cache the results for has_deepgemm, has_deepep, has_pplx (#20187)

2025-06-28 22:06:38 +00:00

base_device_communicator.py

[Kernels] Clean up FusedMoeMethodBase and modular kernel setup. Remove extra arguments from modular kernel methods. (#22035)

2025-08-15 14:46:00 -04:00

cpu_communicator.py

[CPU] Enable shared-memory based pipeline parallel for CPU backend (#21289)

2025-07-21 09:07:08 -07:00

cuda_communicator.py

Fix Flashinfer CUTLASS MOE Allgather (#21963)

2025-08-07 19:18:25 -07:00

cuda_wrapper.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

custom_all_reduce_utils.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

custom_all_reduce.py

[CI] Fix tests/distributed/test_ca_buffer_sharing.py (#22849)

2025-08-13 20:09:30 -07:00

neuron_communicator.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

pynccl_wrapper.py

Add pynccl all-gatherv and reducescatterv (#20154)

2025-07-11 18:59:23 -07:00

pynccl.py

Add pynccl all-gatherv and reducescatterv (#20154)

2025-07-11 18:59:23 -07:00

quick_all_reduce.py

[Feature] add quick all reduce (#19744)

2025-06-26 20:54:24 -07:00

ray_communicator.py

[Misc] correct static type check for GroupCoordinator (#21946)

2025-08-05 03:17:46 -07:00

shm_broadcast.py

[MISC][Bugfix] Use less CPU when message queue has been empty for some time (#16226)

2025-06-05 16:53:08 +00:00

tpu_communicator.py

[TPU] make ptxla not imported when using tpu_commons (#23081)

2025-08-19 11:46:42 +08:00

xpu_communicator.py

[XPU][CI] enhance xpu test support (#20652)

2025-07-09 16:53:09 +00:00