Fix distributed AutoQuantize scoring and share backward setup - #2231

Merged

realAsma merged 12 commits into

NVIDIA:mainfrom

joshua-hill:fix/autoquant-scoring-infrastructure

Sep 13, 2026

Commits on Aug 22, 2026

Refactor AutoQuantize backward scoring
joshua-hill
committed
cb93b2d

Commits on Aug 23, 2026

Restore AutoQuantize forward attributes exactly
joshua-hill
committed
d733339

Commits on Aug 24, 2026

Restore AutoQuantize backward hook state
joshua-hill
committed
2edb3a3

Commits on Aug 25, 2026

Fix gradient scoring for reused modules
joshua-hill
committed
70acb63
Guard gradient replay stacks by autograd participation
joshua-hill
committed
14b2af2
Bind replay scoring to output invocations
joshua-hill
committed
00aba57
Avoid scalar extraction in distributed scoring test
joshua-hill
committed
e34f072
Bound output hooks to autograd graph lifetime
joshua-hill
committed
d5a62bd
Merge branch 'main' into fix/autoquant-scoring-infrastructure
joshua-hill
authored
02cf55e

Commits on Sep 4, 2026

Merge branch 'main' into fix/autoquant-scoring-infrastructure
joshua-hill
authored
4a99034

Commits on Sep 5, 2026

Warn when inferring AutoQuantize parallel state

joshua-hill
and
cursoragent
committed
fbfe607

Commits on Sep 12, 2026

Merge branch 'main' into fix/autoquant-scoring-infrastructure
realAsma
authored
c96355c