Repository navigation
[DSV4] Add E=49 shared-expert-fusion fp8/fp4 fused-MoE tuned configs (gfx950) - #1
karverma-amd wants to merge 1 commit into
Conversation
…s (gfx950) EP8 with shared-expert-fusion produces w1.shape[0]=49 (48 routed + 1 shared) per rank; no tuned dsv4_fp8fp4 rows existed for E=49, so the routed MoE GEMM fell back to the heuristic FlyDSL tile picker. Tuned the full M-tier ladder (4..32768: decode conc 4-512 + prefill to 32k) at model_dim=7168, inter_dim=3072, topk=6, per_1x32 a8(fp8)/w4(fp4). All 14 shapes verified 0.0% err; tuner-vs- heuristic tile: +26% @m=512, +37% @m=1024, +7-9% on large prefill, tie elsewhere.
🏷️ CI GuideRuns automatically on every PR:
Extended tests (opt-in via labels):
PR title tags & labels: |
GSM8K validation (new image, tuned rows exercised)Image: Tuned rows confirmed activeServer boot + full eval produced 0 GSM8K — 1319 questions, parallel 13195-shot, temperature 0, max-new-tokens 512,
Reference: bf16-dispatch baseline on this workload is 0.945; MXFP8 arm 0.951. This run is |
[DSV4] E=49 shared-expert-fusion fp8/fp4 fused-MoE tuned configs (gfx950)
Draft — attaching GSM8K validation below before marking ready.
What
Adds 14 tuned FlyDSL rows to
aiter/configs/model_configs/dsv4_fp8fp4_tuned_fmoe.csvfor the E=49 shape (48 routed + 1 shared expert per rank), which is what
DeepSeek-V4-Pro produces under EP8 with shared-expert fusion (
w1.shape[0] == 49).Why
The runtime fused-MoE config lookup keys on
expert(the materialized expert-weightcount), so shared-fusion EP8 queries
expert=49. Nodsv4_fp8fp4rows existed forE=49, so the routed a8(fp8)×w4(fp4) GEMM fell back to the heuristic FlyDSL tile picker
(
[aiter][fused_moe] no tuned FlyDSL config).Coverage
Full padded-M ladder — decode conc 4–512 and prefill to 32k:
M ∈ {4,8,16,32,64,128,256,512,1024,2048,4096,8192,16384,32768}atmodel_dim=7168, inter_dim=3072, topk=6, per_1x32,a=fp8_e4m3fn,w=fp4_e2m1fn_x2.All 14 shapes verified 0.0% err on both stages (
flydsl_moe1/2_..._gui_fp8).Tuner vs heuristic tile
Validation
GSM8K (1319 questions, 5-shot, temperature 0, parallel 1319) — results below.