pytorch
521e1e83 - [Gradient Compression] Remove some low-level methods of GradBucket class (#53098)

Commit
3 years ago
[Gradient Compression] Remove some low-level methods of GradBucket class (#53098) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/53098 Remove some low-level methods that are no longer needed since `get_per_parameter_tensors` method is added to `GradBucket` class. Avoid unnecessary exposure to the internals before publishing GradBucket APIs. ghstack-source-id: 122723683 Test Plan: buck test mode/dev-nosan caffe2/test/distributed:c10d -- test_powerSGD_ddp_comm_hook_nccl Reviewed By: rohan-varma Differential Revision: D26720919 fbshipit-source-id: 46fb6423008792e72d7a1dd68930a31e0724c92c
Author
Yi Wang
Parents
Loading