Skip to content

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

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
coderfeli wants to merge 20 commits into
base: main
Choose a base branch
from
Open

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

Show file tree
Hide file tree
Changes from 9 commits
Commits
Show all changes
20 commits
Select commit Hold shift + click to select a range
61bf4b3
[MoE 2-stage port] Stage 1: land layout-API helper layer
Jul 27, 2026
c469794
[MoE 2-stage port] Stage 2b (part 1): native-fp8 gate-up building blocks
Jul 27, 2026
b2e56f3
[MoE 2-stage port] Stage A1 WIP checkpoint (pre-implementation)
Jul 30, 2026
3f9fa4e
[MoE 2-stage port] fp8 gemm1 correct+perf-aligned; bf16 gemm1; all ti…
Jul 31, 2026
8ed3496
[MoE 2-stage port] Port fp8 gemm2 to new fx.* pipeline (atomic/reduce…
Jul 31, 2026
71a5277
[MoE 2-stage port] Extend new stage2 pipeline from fp8 to bf16
Jul 31, 2026
cc0bec3
[MoE 2-stage port] Make new fx.* pipeline the canonical fp8/bf16 path
Aug 1, 2026
56a09e2
[MoE 2-stage port] Route bf16 back to legacy; keep only fp8 on new pi…
Aug 1, 2026
aa92958
[MoE 2-stage port] Gate new fp8 pipeline to CDNA4 (gfx95*); gfx942 us…
Aug 1, 2026
4023f3c
[MoE 2-stage port] Clean up new fp8 pipeline: drop dead helpers, trim…
Aug 1, 2026
76c9f22
[MoE 2-stage port] Enable new fp8 pipeline on gfx942 (CDNA3) with bf1…
Aug 1, 2026
0538cc7
[MoE 2-stage port] Fix ~600x atomic-scatter slowdown on small-tile/de…
Aug 1, 2026
5e81f8f
[MoE 2-stage port] Document profiled root-cause of fp8 gemm2 decode gap
Aug 1, 2026
7989e6a
[MoE 2-stage port] Fit gemm2 LDS in CDNA3 64KB (single region for A p…
Aug 1, 2026
b49e84d
[MoE 2-stage port] Document verified root-cause of fp8 gemm2 atomic gap
Aug 1, 2026
e241a54
[MoE 2-stage port] Roll fp8 gemm2 k-tile loop to recover atomic occup…
Aug 1, 2026
e53fa13
[MoE 2-stage port] Trim gemm2 perf-note docstring (condense to essent…
Aug 1, 2026
64c267d
[MoE 2-stage port] Trim comments/docstrings to 1-2 lines (LoC cleanup)
Aug 1, 2026
96e4ab1
Merge origin/main into worktree-agent-moe-2stage-port
Aug 5, 2026
62e4cec
[MoE] Port reduce to layout API (fp8 + EP-gather); drop legacy reduct…
Aug 7, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Loading
Loading