Fix 10 bugs in BF16 XQA MLA kernel for SM120/SM121#2689

Open

blake-snc wants to merge 10 commits intoflashinfer-ai:mainfrom

blake-snc:feat-sm120-bf16-mla

Commits on Apr 16, 2026

Support BF16 MLA on SM120 with shared-mem fallback

maomao123321
authored and
blake-snc
committed
Fix 10 bugs in BF16 XQA MLA kernel for SM120/SM121

blake-snc
and
claude
committed
Address review feedback: simplify ternaries, deduplicate smem config, fix ruff format

blake-snc
and
claude
committed
Address review comments: SM12x guard, remove duplicate init, add swizzle assert

blake-snc
and
claude
committed
fix hardcoded device 0 in configureKernel()

blake-snc
and
claude
committed
use is_sm12x_supported() for SM12x capability checks

blake-snc
and
claude
committed
include CUDA version requirements in SM12x error messages

blake-snc
and
claude
committed
ci: retrigger CI after infrastructure failures

blake-snc
and
claude
committed

Commits on Apr 21, 2026