pytorch
57ba6156 - Back out "[shard] use scatter in shard_parameter API" (#75295)

Commit
3 years ago
Back out "[shard] use scatter in shard_parameter API" (#75295) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/75295 Original commit changeset: c823c5d0066a Original Phabricator Diff: D33933419 (https://github.com/pytorch/pytorch/commit/288de548fd0e270b527d941f8d08302006ad9fa5) https://github.com/pytorch/pytorch/pull/72160 broke some sharded tensor tests. Let's revert it first and reland it again once the test failure has been fixed. ghstack-source-id: 153135278 (Note: this ignores all push blocking failures!) Test Plan: CI Reviewed By: wanchaol Differential Revision: D35418031 fbshipit-source-id: a4435a62d4487a927d0cf79624afe4f0951809c8 (cherry picked from commit a20d1190fa09fee08534aa91a52ffab324edfdd9)
Author
Committer
Parents
Loading