record Torch DP and DDP modules forward (#55578)
Summary:
Fixes #{issue number}
Pull Request resolved: https://github.com/pytorch/pytorch/pull/55578
Reviewed By: gdankel
Differential Revision: D27862392
Pulled By: ilia-cher
fbshipit-source-id: 18545d23e35a97c8f760707fecb696a24d47dc0a
