Chengji Yao 7da296be04 [TPU] kv cache update kernel supports dynamic grid (#20235) ...

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

2025-07-02 06:33:37 +00:00

..

blocksparse_attention

[Misc] Add SPDX-FileCopyrightText (#19100)

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

__init__.py

[Core] Refactor Attention Take 2 (#3462)

2024-03-25 04:39:33 +00:00

chunked_prefill_paged_decode.py

[Misc] Add SPDX-FileCopyrightText (#19100)

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

flashmla.py

[Misc] Add SPDX-FileCopyrightText (#19100)

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

hpu_paged_attn.py

[Misc] Add SPDX-FileCopyrightText (#19100)

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

ipex_attn.py

[Kernel] correct cpu worker function parameter type (#19745)

2025-06-20 10:50:13 +00:00

merge_attn_states.py

[Misc] Add SPDX-FileCopyrightText (#19100)

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

nki_flash_attn.py

[Refactor] Remove duplicate ceil_div (#20023)

2025-06-25 05:19:09 +00:00

paged_attn.py

[Misc] Add SPDX-FileCopyrightText (#19100)

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

pallas_kv_cache_update.py

[TPU] kv cache update kernel supports dynamic grid (#20235)

2025-07-02 06:33:37 +00:00

prefix_prefill.py

[Misc] Add SPDX-FileCopyrightText (#19100)

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

rocm_aiter_mla.py

[Misc] Add SPDX-FileCopyrightText (#19100)

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

rocm_aiter_paged_attn.py

[Misc] Add SPDX-FileCopyrightText (#19100)

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

triton_decode_attention.py

[Misc] Add SPDX-FileCopyrightText (#19100)

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

triton_flash_attention.py

[Misc] Fix misleading ROCm warning (#19486)

2025-06-12 00:12:10 +08:00

triton_merge_attn_states.py

[Misc] Add SPDX-FileCopyrightText (#19100)

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

triton_unified_attention.py

[Kernel] Add Split-KV Support to Unified Triton Attention Kernel (#19152)

2025-06-17 10:45:07 +00:00