fix: checkpoint saving with distributed optimizer + overlap param gather#949

Merged

terrykong merged 1 commit intoNVIDIA-NeMo:mainfrom

ananthsub:ananthsub/fix-ckpt-save-dist-opt-overlap-param-gather

Aug 21, 2025

Commits on Aug 20, 2025

handle forward pre hooks if dist opt + param gather are used together for checkpoint saving
ananthsub
committed