Repository navigation
[FlyDSL] optimize dsv4.1, k3, m3 moe perf for gfx1250 - #7296
Conversation
🏷️ CI GuideRuns automatically on every PR:
Extended tests (opt-in via labels):
One backend per PR: PR title tags & labels: |
NOT READY FOR HUMAN REVIEWPR Self-ReviewRevision Waiting for the PR author. The readiness report will appear after the self-review answers are complete for this revision. Questions to answer in the PR description
Edit the Missing or invalid answers
Shadow mode reports what would block but keeps the status successful. Set |
…pt_1031 # Conflicts: # aiter/ops/flydsl/grouped_gemm_mxfp4.py # aiter/ops/flydsl/kernels/mxfp4_preshuffle_gfx1250_tdm.py
Resolve the conflict in mxfp4_preshuffle_gfx1250_tdm.py: main moved the TDM output stores onto the shared tiled_store() helper. Keep that refactor and make store_tile() a k_waves guard around it (only k_group 0 writes the tile), routing both epilogue stores -- including the a_preshuffled one that auto-merged onto tiled_store() -- through the guard. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
Resolve conflicts with #7296: - mxfp4_preshuffle_gfx1250_tdm.py: keep the early quit for padded tiles (has_work = blk_m < tile_map[E-1]) in front of the one-pass psum lookup. - grouped_moe_gfx1250.py: take main's comment (same code). - tuned_grouped_fmoe.csv: keep this PR's dsv4 flash rows. split_k1/split_k2 now mean the in-workgroup K split, so retune the a8w4 T=1 row (was the old global split-K 4/2): 16x64x512 w1 / 16x128x256 w2, split_k 2/2.
Summary
optimize dsv4.1, k3, m3 moe perf for gfx1250
Test Result