[PyTorch] Add optional caller-provided output/grad-input buffers to GroupedLinear and fused grouped MLP#3161

Merged

timmoon10 merged 11 commits into

NVIDIA:mainfrom

phu0ngng:pyt-gg-w-symm

Jul 17, 2026

Commits on Jul 2, 2026

Add optional caller-provided output/grad-input buffers to GroupedLinear module and fusible ops
phu0ngng
committed
[pre-commit.ci] auto fixes from pre-commit.com hooks

pre-commit-ci[bot]
authored and
phu0ngng
committed
Route per-op kwargs through Sequential via module-keyed op_kwargs mapping
phu0ngng
committed
Write fused grouped MLP MXFP8 output and dgrad directly into caller buffers, eliminating the D2D copy + cleanup
phu0ngng
committed
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot]
committed
Use 256-aligned splits in caller-buffer grouped MLP test
phu0ngng
committed

Commits on Jul 11, 2026

Merge branch 'main' into pyt-gg-w-symm
phu0ngng
authored

Commits on Jul 15, 2026

Commits on Jul 16, 2026