tune: a8w8 gemm tuning for Qwen3.5 MXFP4-AttnFP8 model - #4396
Conversation
🏷️ CI GuideRuns automatically on every PR:
Extended tests (opt-in via labels):
|
There was a problem hiding this comment.
Pull request overview
This PR adds a large set of tuned a8w8_bpreshuffle GEMM config rows intended to accelerate FP8 GEMMs for the Qwen3.5-397B MXFP4-AttnFP8 model on gfx950.
Changes:
- Appends new
gfx950tuning rows covering multiple(N, K)groups (notably64×4096,4096×4096,8704×4096,10240×4096) across a wideMrange. - Introduces additional kernel selections spanning
ck,cktile, andflydslimplementations for those shapes.
💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.
|
model configs should be added to the specific csv in model_configs dir |
yzhou103
left a comment
There was a problem hiding this comment.
please move the configs to csv in model_configs
done. please review it. |
There is already a related csv in aiter. aiter/configs/model_configs/a8w8_bpreshuffle_tuned_gemm_qwen3_5_397b_a17b_mxfp4_attnfp8.csv(#4317) |
sorry. please review it. |
Motivation
Adds a8w8_bpreshuffle FP8 GEMM tuning entries for the Qwen3.5-397B-A17B-MXFP4-AttnFP8-V2
model
Technical Details
Test Plan
Test Result
Submission Checklist