grimulkan a1257fd1ea [Kernel] Add FP8 KV cache support to Triton MLA decode attention (#34597) ...

Signed-off-by: grimulkan <grimulkan@gmail.com>

2026-03-12 08:32:34 -07:00

..

__init__.py

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

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

chunked_prefill_paged_decode.py

Fix IndexError with encoder-decoder models when using Custom Paged Attention (#33112)

2026-01-27 10:33:37 +08:00

common.py

[Perf] Optimize dcp allocate tensor (#33102)

2026-01-27 17:24:41 -05:00

dcp_alltoall.py

[Core] Add All-to-All communication backend for DCP (#34883)

2026-03-04 10:01:57 -05:00

flashmla.py

Fix spelling errors (#33978)

2026-02-06 23:58:50 -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

[XPU][9/N] clean up existing ipex code/doc (#34111)

2026-02-11 00:27:15 -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

[ROCm][Perf] Enable sparse_mla's cudagraph on ROCm platform (#35719)

2026-03-10 09:14:35 -07:00

triton_decode_attention.py

[Kernel] Add FP8 KV cache support to Triton MLA decode attention (#34597)

2026-03-12 08:32:34 -07: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

[Performance] Improve Triton prefill attention kernel's performance (#32403)

2026-01-17 20:19:59 -08: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

[Feature] Enable TRITON_ATTN for Batch Invariance (#33688)

2026-02-04 13:27:34 +08:00

vit_attn_wrappers.py

Flashinfer cuDNN backend for Qwen3 VL ViT attention (#34580)

2026-02-27 20:20:23 +08:00

xpu_mla_sparse.py

Add XPU MLA Sparse backend for DeepSeek v3.2 (#33230)

2026-03-11 19:19:15 +08:00