vllm/distributed/tpu_distributed_utils.py · 783921d889f942d6ec121e84d8eaa4704e5e0f27 · OpenDAS / vllm_cscc

[Hardware][TPU] Initial support of model parallelism with single worker using SPMD (#18011) · 9112b443

Siyuan Liu authored Jun 02, 2025


Signed-off-by: Siyuan Liu <lsiyuan@google.com>
Co-authored-by: Hossein Sarshar <hossein.sarshar@gmail.com>
Co-authored-by: Chengji Yao <chengjiyao@google.com>

9112b443

tpu_distributed_utils.py 7.58 KB

Replace tpu_distributed_utils.py