[PyTorch] Finalize quantization updates at logical backward boundaries - #34

Draft

pggPL wants to merge 9 commits into

mainfrom

bwd_state_update_refactor

Commits on Aug 17, 2026

[PyTorch] Replace first-module ownership with deferred backward quantization state update
pggPL
committed
5a53170
Adapt existing tests to deferred backward update timing
pggPL
committed
eefc761
Merge remote-tracking branch 'upstream/main' into bwd_state_update_refactor
pggPL
committed
14db0f5
Rename schedule_backward_quantization_update to request_backward_quantization_update
pggPL
committed
624cdc6
Move update-scheduling tests into test_recipe.py
pggPL
committed
cfbf8c0
Credit original investigation of the duplicate FP8 update bug

pggPL
and
AlbertYang514
committed
38a6abf
Make quantization state updates per-recipe via RecipeState
pggPL
committed
d64dcad
Restore NVTX range around the backward quantization state update
pggPL
committed
ef58889

Commits on Aug 18, 2026

[PyTorch] Finalize quantization updates at backward boundaries
pggPL
committed
90126ae