pytorch
c22fc448 - [Gradient Compression] Remove cuda.syncrhonize in batched powerSGD (#54482)

Commit
3 years ago
[Gradient Compression] Remove cuda.syncrhonize in batched powerSGD (#54482) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/54482 `cuda.synchronize` is unnecessary for `batched_powerSGD_hook`. ghstack-source-id: 124607761 Test Plan: f259607860 f259563921 Reviewed By: rohan-varma Differential Revision: D27254314 fbshipit-source-id: 4744c07a6f0c8939e766ffa935ddbf3c47e85d18
Author
Yi Wang
Parents
Loading