iefgnoix b6be6f8d1e [TPU] Support sliding window and logit soft capping in the paged attention kernel for TPU. (#15732) ...

Signed-off-by: Xiongfei Wei <isaacwxf23@gmail.com>

2025-04-03 14:23:28 -07:00

..

worker

[TPU][CI] Fix TPUModelRunner Test (#15667)

2025-03-28 00:01:26 -07:00

__init__.py

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

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

test_basic.py

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

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

test_mha_attn.py

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

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

test_pallas.py

[TPU] Support sliding window and logit soft capping in the paged attention kernel for TPU. (#15732)

2025-04-03 14:23:28 -07:00

test_perf.py

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

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

test_sampler.py

[Bugfix][TPU][V1] Fix recompilation (#15553)

2025-03-27 19:15:06 +00:00

test_topk_topp_sampler.py

[V1][TPU] TPU-optimized top-p implementation (avoids scattering). (#15736)

2025-04-02 17:18:08 -07:00