Chengji Yao 190c45a6af

[TPU][Bugfix] fix the missing apply_model in tpu worker (#25526) ...

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

2025-09-24 05:18:08 +00:00

..

worker

[Core] Use CpuGpuBuffer for block table tensors (#24795)

2025-09-16 19:18:06 -07:00

__init__.py

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

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

test_basic.py

[TPU][Test] Rollback PR-21550. (#21619)

2025-07-25 13:22:01 -07:00

test_kv_cache_update_kernel.py

[TPU] kv cache update kernel doesn't need to be padded slices to multiple of num_slices_per_block (#22394)

2025-08-09 20:49:04 -07:00

test_mha_attn.py

[V0 Deprecation][TPU] Remove V1 flag check from tests (#22248)

2025-08-05 06:53:23 -07:00

test_multimodal.py

[CI/Build] Serve images used by multimodal tests through local HTTP Server (#23907)

2025-09-03 16:13:11 +08:00

test_pallas.py

[Attention][FlashInfer] Enable FP8 FlashInfer (TRTLLM) MLA decode (#24705)

2025-09-12 15:45:53 -06:00

test_perf.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

test_sampler.py

[V0 Deprecation][TPU] Remove V1 flag check from tests (#22248)

2025-08-05 06:53:23 -07:00

test_spmd_model_weight_loading.py

[Misc] Add SPDX-FileCopyrightText (#20428)

2025-07-04 07:40:42 +00:00

test_topk_topp_sampler.py

[TPU] Deprecate xm.mark_step in favor of `torch_xla.sync (#25254)

2025-09-22 10:12:57 +00:00

test_tpu_int8.py

[TPU][Bugfix] fix the missing apply_model in tpu worker (#25526)

2025-09-24 05:18:08 +00:00

test_tpu_qkv_linear.py

[Misc] Add SPDX-FileCopyrightText (#20428)

2025-07-04 07:40:42 +00:00