Optionally pad QK/V for better kernel selection - #3339

Open

janEbert wants to merge 6 commits into

NVIDIA:mainfrom

janEbert:optional-mla-pad

Commits on Aug 10, 2026

Automate MLA padding-related kernel selection
janEbert
committed
438ecfe
Add padding-related tests
janEbert
committed
573edd8
Fix tests
janEbert
committed
50dbfef
Add memoization
janEbert
committed
d24be6d
Add test to CI
janEbert
committed
bdc487e
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot]
committed
836f919