[Lightning Attention] Add Hopper SM90 CuTe DSL prefill - #111

Merged

icavan merged 8 commits into

inclusionAI:mainfrom

Aharrypotter:lightning-sm90-cutedsl-p8-pr-ready

Aug 3, 2026

Commits on Jul 27, 2026

feat: add SM90 CuTe DSL Lightning Attention prefill
Aharrypotter
committed
96282a9
perf(lightning): use TMA for packed SM90 tail stores
Aharrypotter
committed
c792bbe

Commits on Jul 28, 2026

perf(lightning): reduce SM90 QK handoff synchronization
Aharrypotter
committed
b83f86e
perf(lightning): default SM90 packed prefill to nonpersistent
Aharrypotter
committed
eebd4a6
perf(lightning): refine SM90 output and decay paths
Aharrypotter
committed
93da019

Commits on Jul 29, 2026

fix(lightning): use stable TensorMap workspace slots
Aharrypotter
committed
bb69d07
fix(lightning): synchronize decay LUT scan
Aharrypotter
committed
247941e

Commits on Jul 30, 2026

fix(lightning): support CuTe DSL 4.4 through 4.6
Aharrypotter
committed
ffd1291