[FX] Fix _replicate_for_data_parallel (#63821)
authorJames Reed <jamesreed@fb.com>
Tue, 24 Aug 2021 20:44:52 +0000 (13:44 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Tue, 24 Aug 2021 20:48:15 +0000 (13:48 -0700)
commit4e37a015c7ed7ac4a4f57057cb726f53b94e693b
tree9f80cdd070e323864388ab57db44b3a97b686142
parent5be17ec1fca4fcce5464cd679ee3f3dd6f102059
[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
test/test_fx.py
torch/fx/graph_module.py