tests/v1/tpu/test_tpu_qkv_linear.py · cc7f22a8b18a3a5f30a6ea0a42bd042ad91d64cc · 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

test_tpu_qkv_linear.py 2.71 KB

Replace test_tpu_qkv_linear.py