Carol Zheng fba02e3bd1

[Bugfix][TPU] Fix tpu model runner testcase failure (#18810) ...

Signed-off-by: Carol Zheng <cazheng@google.com>

2025-05-30 18:04:03 +08:00

..

worker

[Bugfix][TPU] Fix tpu model runner testcase failure (#18810)

2025-05-30 18:04:03 +08:00

__init__.py

[V1] TPU - Add tensor parallel support via Ray (#13618)

2025-03-08 08:19:38 -05:00

test_basic.py

[TPU] Enable gemma3-27b with TP>1 on multi-chips. (#17335)

2025-05-05 14:19:58 -07:00

test_mha_attn.py

[TPU][V1] MHA Pallas backend (#15288)

2025-03-21 08:50:39 -07:00

test_multimodal.py

[Frontend] Add missing chat templates for various MLLMs (#17758)

2025-05-07 00:10:01 -07:00

test_pallas.py

[TPU] Update PyTorch/XLA (#16288)

2025-04-09 14:46:32 +08:00

test_perf.py

[TPU][V1][CI] Update regression test baseline for v6 CI (#17064)

2025-04-30 04:03:22 -07:00

test_sampler.py

[TPU] Fix the test_sampler (#17820)

2025-05-08 05:51:33 -04:00

test_topk_topp_sampler.py

[V1][TPU] Enable Top K (#15489)

2025-04-17 18:18:11 +00:00