pytorch

4e37a015 - [FX] Fix _replicate_for_data_parallel (#63821)

[FX] Fix _replicate_for_data_parallel (#63821) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/63821 Test Plan: Imported from OSS Reviewed By: suo Differential Revision: D30502115 Pulled By: jamesr66a fbshipit-source-id: 0f004f95def6e1ba21ccbeab40cb0a739a0ad20c