Skip to content

Revert "Refactor preshuffle GEMM indexing with layout algebra (#1131)" - #1189

Merged
coderfeli merged 1 commit into
mainfrom
revert-1131-preshuffle-layout
Sep 24, 2026
Merged

coderfeli merged 1 commit into
mainfrom
revert-1131-preshuffle-layout

Conversation

@coderfeli

@coderfeli coderfeli commented Sep 24, 2026 •

Copy link
Copy Markdown
Collaborator

Summary

Reverts #1131 (e70bf79), which broke main CI on MI35x (run):

FAILED tests/kernels/test_preshuffle_gemm.py::test_mfma_a8_flyc_preshuffle[fp16-M16-tile16x32x1024-async-eager]
FAILED tests/kernels/test_preshuffle_gemm.py::test_mfma_a8_flyc_preshuffle[fp16-M33-tile32x32x1024-async-eager]
logits_diff ~0.95

This reverts commit e70bf79.

The new DMA layout computes rows_per_wave = 64 // (tile_k_bytes // 16),
which is 0 when an A row exceeds 1024 bytes (fp16/bf16 with tile_k=1024,
exercised by the 2-wave tiles added in #1183), producing wrong results on
gfx950 (test_mfma_a8_flyc_preshuffle[fp16-*-tile*x32x1024-async-eager]).

Co-Authored-By: Claude Opus 5.5 (1M context) <noreply@anthropic.com>
@coderfeli
coderfeli merged commit a90d906 into main Sep 24, 2026
12 checks passed
@coderfeli
coderfeli deleted the revert-1131-preshuffle-layout branch September 24, 2026 03:08
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant