feat: support mxfp4 × mxfp4 gemm on sm100#348

Open

z52527 wants to merge 12 commits into

deepseek-ai:nv_devfrom

z52527:fea-fp4-synced

Commits on May 29, 2026

FP4: port MXF4 kernel + wrappers onto main's refactored APIs

z52527
and
claude
committed
FP4: progress port towards runtime — IMA fixed, smem cap, multicast clamp

z52527
and
claude
committed
FP4: layout override to fea-fp4 tested config + test int8 packing

z52527
and
claude
committed

Commits on Jun 1, 2026

FP4: tests/test_fp4.py — use FP32 output (kernel's only working epilogue path)

z52527
and
claude
committed
FP4: tests/test_fp4.py — full suite passes on synced branch (10/10)

z52527
and
claude
committed
FP4: port fea-fp4's full get_best_fp4_config heuristic — perf parity

z52527
and
claude
committed
FP4: rewrite tests/test_fp4.py to mirror test_fp8_fp4.py style

z52527
and
claude
committed
FP4: align .cuh comments with sm100_fp8_fp4_gemm_1d1d.cuh style

z52527
and
claude
committed
FP4: address codex review (PDL sync, strict FP32 D, prune unused fields)

z52527
and
claude
committed
FP4: remove .item() host sync in m-grouped contiguous swap_ab gating

z52527
and
claude
committed
FP4: address codex re-audit (K-major guard, k%8 assert, dedup C copy, style)

z52527
and
claude
committed
FP4: clarify swap_ab gating comment in contiguous wrapper

z52527
and
claude
committed