Triton MLA perf fixes#33529

Merged

MatthewBonanni merged 17 commits intovllm-project:mainfrom

koush:triton-mla-perf

Apr 2, 2026

Commits on Feb 2, 2026

[Bugfix] implement dynamic splits in triton mla

root
authored and
koush
committed
[Bugfix] Fix triton gqa load access patterns

root
authored and
koush
committed
[Bugfix]: Use tl.range for loop and explciit cache_modifiers

root
authored and
koush
committed
Fix pre-commit issues

root
authored and
koush
committed
Fix pre-commit issues

root
authored and
koush
committed
Remove transpose, was mistaken about PTX improvement
koush
committed
get sm count outside of forward path
koush
committed
Fix accidental load reordering causing SMEM usage increase
koush
committed

Commits on Feb 3, 2026

Commits on Feb 9, 2026

Commits on Feb 20, 2026

Merge branch 'main' into triton-mla-perf
DarkLight1337
authored

Commits on Mar 27, 2026

Commits on Mar 29, 2026

missing import
koush
committed

Commits on Apr 1, 2026

Merge branch 'main' into triton-mla-perf
koush
authored