Add MXFP8 grouped GEMM (E4M3) for routed-expert MoE training#31

Merged

hann-wang merged 18 commits into

mainfrom

yue/mxfp8-grouped-gemm

Jun 26, 2026

Commits on Jun 1, 2026

mxfp8: implement forward grouped GEMM kernel (Step 2)
ysa2215
committed

Commits on Jun 2, 2026

Add MXFP8 grouped GEMM forward guards and coverage
ysa2215
committed

Commits on Jun 5, 2026

mxfp8: implement grouped GEMM backward path
ysa2215
committed

Commits on Jun 10, 2026

Commits on Jun 15, 2026

mxfp8: add offsets entry + padded buffer for grouped GEMM, wire dispatch
ysa2215
committed

Commits on Jun 16, 2026

refactor: merge and simplify mxfp8 grouped gemm related tests, update design and feature doc
ysa2215
committed

Commits on Jun 23, 2026

Merge remote-tracking branch 'origin/main' into yue/mxfp8-grouped-gemm
ysa2215
committed

Commits on Jun 25, 2026

Commits on Jun 26, 2026