pytorch

c523d7d8 - Add a new hook (#99854)

Add a new hook (#99854) Differential Revision: D45220984 Pull Request resolved: https://github.com/pytorch/pytorch/pull/99854 Approved by: https://github.com/albanD