Isotr0py 9dbe1fe960 [Bugfix] Fix missing scale passing for encoder Triton Attention implementation (#32149) ...

Signed-off-by: Isotr0py <mozf@mail2.sysu.edu.cn>

2026-01-12 11:13:41 +00:00

..

__init__.py

[1/N][Attention] Restructure attention: move files (#31916)

2026-01-09 13:10:24 -08:00

chunked_prefill_paged_decode.py

[1/N][Attention] Restructure attention: move files (#31916)

2026-01-09 13:10:24 -08:00

common.py

[1/N][Attention] Restructure attention: move files (#31916)

2026-01-09 13:10:24 -08:00

flashmla.py

[1/N][Attention] Restructure attention: move files (#31916)

2026-01-09 13:10:24 -08:00

merge_attn_states.py

[1/N][Attention] Restructure attention: move files (#31916)

2026-01-09 13:10:24 -08:00

paged_attn.py

[2/N][Attention] Fix pre-commit errors (#32052)

2026-01-10 00:27:15 +00:00

pallas_kv_cache_update.py

[1/N][Attention] Restructure attention: move files (#31916)

2026-01-09 13:10:24 -08:00

prefix_prefill.py

[1/N][Attention] Restructure attention: move files (#31916)

2026-01-09 13:10:24 -08:00

rocm_aiter_mla_sparse.py

[1/N][Attention] Restructure attention: move files (#31916)

2026-01-09 13:10:24 -08:00

triton_decode_attention.py

[1/N][Attention] Restructure attention: move files (#31916)

2026-01-09 13:10:24 -08:00

triton_merge_attn_states.py

[1/N][Attention] Restructure attention: move files (#31916)

2026-01-09 13:10:24 -08:00

triton_prefill_attention.py

[Bugfix] Fix missing scale passing for encoder Triton Attention implementation (#32149)

2026-01-12 11:13:41 +00:00

triton_reshape_and_cache_flash.py

[1/N][Attention] Restructure attention: move files (#31916)

2026-01-09 13:10:24 -08:00

triton_unified_attention.py

[Kernel] Optimize Sliding Window Attention in 3D Triton Kernel (#31984)

2026-01-10 18:13:44 +00:00

vit_attn_wrappers.py

[Misc][LLaMa4] Compile LLaMa Vision Encoder (#30709)

2026-01-09 22:01:38 -05:00