[MoE] Port moe_gemm_2stage (stage1+stage2) to the new fx.* pipeline — fp8 (gfx942 + gfx950) - #947

Open

coderfeli wants to merge 20 commits into

mainfrom

worktree-agent-moe-2stage-port

Commits on Jul 27, 2026

[MoE 2-stage port] Stage 1: land layout-API helper layer

Felix Li
and
claude
committed
61bf4b3
[MoE 2-stage port] Stage 2b (part 1): native-fp8 gate-up building blocks

Felix Li
and
claude
committed
c469794

Commits on Jul 30, 2026

[MoE 2-stage port] Stage A1 WIP checkpoint (pre-implementation)

Felix Li
and
claude
committed
b2e56f3

Commits on Jul 31, 2026

[MoE 2-stage port] fp8 gemm1 correct+perf-aligned; bf16 gemm1; all tile configs

Felix Li
and
claude
committed
3f9fa4e
[MoE 2-stage port] Port fp8 gemm2 to new fx.* pipeline (atomic/reduce/f32)

Felix Li
and
claude
committed
8ed3496
[MoE 2-stage port] Extend new stage2 pipeline from fp8 to bf16

Felix Li
and
claude
committed
71a5277

Commits on Aug 1, 2026

[MoE 2-stage port] Make new fx.* pipeline the canonical fp8/bf16 path

Felix Li
and
claude
committed
cc0bec3
[MoE 2-stage port] Route bf16 back to legacy; keep only fp8 on new pipeline

Felix Li
and
claude
committed
56a09e2
[MoE 2-stage port] Gate new fp8 pipeline to CDNA4 (gfx95*); gfx942 uses legacy

Felix Li
and
claude
committed
aa92958
[MoE 2-stage port] Clean up new fp8 pipeline: drop dead helpers, trim LOC

Felix Li
and
claude
committed
4023f3c
[MoE 2-stage port] Enable new fp8 pipeline on gfx942 (CDNA3) with bf16-atomic carve-out

Felix Li
and
claude
committed
76c9f22
[MoE 2-stage port] Fix ~600x atomic-scatter slowdown on small-tile/decode shapes

Felix Li
and
claude
committed
0538cc7
[MoE 2-stage port] Document profiled root-cause of fp8 gemm2 decode gap

Felix Li
and
claude
committed
5e81f8f
[MoE 2-stage port] Fit gemm2 LDS in CDNA3 64KB (single region for A ping/pong + CShuffle)

Felix Li
and
claude
committed
7989e6a
[MoE 2-stage port] Document verified root-cause of fp8 gemm2 atomic gap

Felix Li
and
claude
committed
b49e84d
[MoE 2-stage port] Roll fp8 gemm2 k-tile loop to recover atomic occupancy

Felix Li
and
claude
committed
e241a54
[MoE 2-stage port] Trim gemm2 perf-note docstring (condense to essential why)

Felix Li
and
claude
committed
e53fa13
[MoE 2-stage port] Trim comments/docstrings to 1-2 lines (LoC cleanup)

Felix Li
and
claude
committed
64c267d

Commits on Aug 5, 2026

Merge origin/main into worktree-agent-moe-2stage-port

Felix Li
and
claude
committed
96e4ab1

Commits on Aug 7, 2026

[MoE] Port reduce to layout API (fp8 + EP-gather); drop legacy reduction.py

Felix Li
and
claude
committed
62e4cec