[FlyDSL] gfx950 FP8 MQA logits indexer kernel - #4538

Open

vpietila-amd wants to merge 32 commits into

mainfrom

users/vpietila/gfx950-fp8-mqa-logits

Commits on Aug 3, 2026

[FlyDSL] fp8_mqa_logits: introduce MfmaAtom, generalise the MFMA builder
vpietila-amd
committed
ccf59bb
[FlyDSL] fp8_mqa_logits: add the gfx950 scaled MFMA atoms
vpietila-amd
committed
73d8fc8
[FlyDSL] fp8_mqa_logits: add the LDS-pipelined gfx950 builder
vpietila-amd
committed
099c11a
[FlyDSL] fp8_mqa_logits: arch-aware variant registry and auto-selection
vpietila-amd
committed
013159f
[FlyDSL] fp8_mqa_logits: run the correctness sweep on gfx950 too
vpietila-amd
committed
c4b1cb9
[FlyDSL] fp8_mqa_logits: satisfy black and ruff
vpietila-amd
committed
c046d27
[FlyDSL] fp8_mqa_logits: add the FlyDSL-vs-Triton benchmark and plotter
vpietila-amd
committed
0c11a6b
Optimize number of kv splits per architecture.
vpietila-amd
committed
45e5901
Merge branch 'main' of github.com:ROCm/aiter into users/vpietila/gfx950-fp8-mqa-logits
vpietila-amd
committed
524b649
Fix test execution on gfx950.
vpietila-amd
committed
25f602a

Commits on Aug 4, 2026

[FlyDSL] fp8_mqa_logits: fix s_waitcnt mis-encoding in the LDS pipeline

vpietila-amd
and
claude
committed
217bb74
Enable benchmarking all FlyDSL variants at once.
vpietila-amd
committed
638028a
Black formatting.
vpietila-amd
committed
d8030fd
Code clean-up.
vpietila-amd
committed
9b354ff
Fuse separate initialization kernel into the logits kernel.
vpietila-amd
committed
49ab748
Code clean-up. Make clean-logit path a compile time constant.
vpietila-amd
committed
0d8b015

Commits on Aug 5, 2026

Commits on Aug 6, 2026

Commits on Aug 13, 2026

Merge remote-tracking branch 'origin/main' into users/vpietila/gfx950-fp8-mqa-logits
vpietila-amd
committed
aab7e97

Commits on Aug 18, 2026

Merge branch 'main' into users/vpietila/gfx950-fp8-mqa-logits
vpietila-amd
authored
f1e274f

Commits on Aug 25, 2026

Commits on Sep 2, 2026

Address feedback from code review.
vpietila-amd
committed
afaf016