Add 8-wave warp-pipeline GEMM kernels (a16w16 / a8w8 / a4w4)#44

Open

zhanglx13 wants to merge 71 commits into

mainfrom

8wave-warp-pipeline

Commits on Jul 10, 2026

Add a16w16-8wave: 8-wave warp-pipeline FP16 GEMM (gfx950)

zhanglx13
and
claude
committed
Fix cross-wave GR-completion race in 8-wave warp-pipeline (u3)

zhanglx13
and
claude
committed
Restore warp-pipeline: hoist step-0 wait_group above loop-index arith

zhanglx13
and
claude
committed
8wave u3: non-relaxed smem.load + inline buffer_load_to_shared (v9 style)

zhanglx13
and
claude
committed
8wave u3: v9 pointer-walk base/offset for buffer_load_to_shared

zhanglx13
and
claude
committed
8wave: relaxed loads (drop redundant membar barrier) + version the kernels

zhanglx13
and
claude
committed
8wave v1: implement sliceMN BLOCK_K=64 2-buffer warp-pipeline
zhanglx13
committed
8wave v1: kill the 67 epilogue spills (store pointer-walk + de-interleave)
zhanglx13
committed
8wave: add per-version READMEs for v0 and v1
zhanglx13
committed
8wave: rewrite the dir README; add K=32768 to the bench sweep
zhanglx13
committed
8wave: turn the v0 fix list into a table with an as-ported baseline
zhanglx13
committed
a16w16-8wave README: add MI355X perf table; mark §2 numbers as MI350X
zhanglx13
committed
a16w16-8wave README: refresh v0 MI355X row to 5-run medians
zhanglx13
committed
a16w16-8wave README: add MI355X v0/v1/v9 ATT trace image to §4
zhanglx13
committed
a16w16-8wave README: include full repo name (AMD-Triton/gluon-kernels) for the v0 port
zhanglx13
committed
v1_sliceMN_BK64_nS2 README: add ping-pong loop design figure (side-by-side with text)
zhanglx13
committed
v1 README: enlarge ping-pong design figure (300 -> 480px)
zhanglx13
committed
v1 README: show ping-pong figure full-size (640px, centered) instead of side-by-side
zhanglx13
committed
a16w16-8wave: bake no-AGPR into kernels via llvm_fn_attrs
zhanglx13
committed
docs: add warp-pipelining theory reference
zhanglx13
committed
Add 8-wave warp-pipeline a8w8 (BF8) and a4w4 (MXFP4) GEMM kernels
zhanglx13
committed
docs(8wave): add 8-wave section to gemm README; refresh perf/VGPR to current build

zhanglx13
and
claude
committed

Commits on Jul 11, 2026

Commits on Jul 12, 2026

Commits on Jul 13, 2026