Chengji Yao 471fe65630 [TPU][V1] Implicitly adjust page size when there's SMEM OOM (#16871) ...

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

2025-04-21 15:43:13 -06:00

..

worker

[TPU][V1] Fix padding recompilation when max-num-batched-tokens is not even (#16726)

2025-04-17 18:09:57 +00:00

__init__.py

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

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

test_basic.py

[TPU][V1] Implicitly adjust page size when there's SMEM OOM (#16871)

2025-04-21 15:43:13 -06:00

test_mha_attn.py

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

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

test_pallas.py

[TPU] Update PyTorch/XLA (#16288)

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

test_perf.py

[V1] TPU CI - Add basic perf regression test (#15414)

2025-03-31 13:25:20 -04:00

test_sampler.py

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

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

test_topk_topp_sampler.py

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

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