[WIP] Blockscaled SM90#141

Open

KareemMusleh wants to merge 8 commits into

Dao-AILab:mainfrom

KareemMusleh:sm90-blockscaled-support

Commits on May 17, 2026

tests pass
Карим
committed

Commits on May 19, 2026

A bunch of bug fixed. ~1100 TFLOPS. works with multiple math wg
Карим
committed

Commits on May 23, 2026

add support for tile_m == 256. 1200 TFLOPS. Reuse acc_slow to minimize register usage
Карим
committed
lint
Карим
committed
more speed ups
Карим
committed
unroll = 8, added multicast for SFA, SFB is not is smem. Perf is comparable to DG 1d2d
Карим
committed
refactoring
Карим
committed

Commits on May 24, 2026

benchmark against DG
Карим
committed