pytorch
e66c498d - Log modules FSDP hooks fire for (#102508)

Commit
1 year ago
Log modules FSDP hooks fire for (#102508) Under torch_distributed_debug >= INFO and use_orig_params=True, log post backward hook firing to debug things like FSDP + AC integration. Differential Revision: [D46172916](https://our.internmc.facebook.com/intern/diff/D46172916/) Pull Request resolved: https://github.com/pytorch/pytorch/pull/102508 Approved by: https://github.com/awgu, https://github.com/fegin
Author
Committer
Parents
Loading