FlashMLA ported to consumer Blackwell sm_120, 2-3x faster MLA attention vs SDPA
smashedshanky · reddit · 2026-08-30
While building an open-source LLM training repo supporting arbitrary architectures (GQA/MLA/Dense), a developer found DeepSeek's FlashMLA kernel only compiled for data-center sm100/sm90, so they ported and open-sourced a Windows/Linux build for consumer Blackwell sm120.
Inference vs PyTorch SDPA: sparse fp8 decode 2.62x faster; sparse serving 5x; sparse prefill (sq=512, skv=8192) 2.61x; FP8 KV cache gives -8% latency and 1.84x less cache memory (SDPA: +1.8%). Model-level bf16 decode roughly at parity.
Training (fwd+bwd): Dense seq 1024/4096/8192 at 3.29x / 2.40x / 3.04x, sparse prefill 3.01x.
More from Infra
- 40nm Neural-Dynamics Chip Uses Conductance Drift for 2.12ms Iteration Latency — maier_ak · 2026-09-01
- Qwen3.8 Flash hits 415 tok/s on dual DGX Sparks — NVIDIAAI · 2026-09-01
- OpenAI's 'Jalapeno' Chip Revealed: 1500 Tokens/s Throughput — firstadopter · 2026-09-01
- TensorSharp vs llama.cpp: Qwen 3.8 Flash Next Benchmarks — fuzhongkai · 2026-09-01
- Tencent Hunyuan AngelSlim: Compressing Hy4 Model to 214GB with Heterogeneous Inference — 腾讯混元 · 2026-09-01
- Samsung shifts to 8-layer HBM4E for Nvidia with ~20% higher speed spec — 创业邦 · 2026-09-01