pytorch
3e88acbf - [Gradient Compression] Do not skip the comm hook tests on Gloo backend (#58784)

Commit
3 years ago
[Gradient Compression] Do not skip the comm hook tests on Gloo backend (#58784) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/58784 DDP communication hooks are already supported on Gloo backend. No longer need to skip these tests on Gloo. Original PR issue: https://github.com/pytorch/pytorch/issues/58467 ghstack-source-id: 129635828 Test Plan: buck test mode/dev-nosan caffe2/test/distributed:distributed_gloo_fork -- test_ddp_comm_hook_logging buck test mode/dev-nosan caffe2/test/distributed:distributed_gloo_fork -- test_ddp_hook_parity_allreduce buck test mode/dev-nosan caffe2/test/distributed:distributed_gloo_fork -- test_ddp_hook_parity_allreduce_process_group buck test mode/dev-nosan caffe2/test/distributed:distributed_gloo_fork -- test_ddp_hook_parity_powerSGD Reviewed By: rohan-varma Differential Revision: D28617214 fbshipit-source-id: 3bafb0c837a15ad203a8570f90750bc5177d5207
Author
Yi Wang
Parents
Loading