pytorch
94975527 - Update SyncBatchNorm _all_gather_base to all_gather_into_tensor (#89521)

Commit
2 years ago
Update SyncBatchNorm _all_gather_base to all_gather_into_tensor (#89521) Summary: Fixes https://github.com/pytorch/pytorch/issues/88568 `_all_gather_base` is deprecated. So replacing its usage with `all_gather_into_tensor` Test Plan: CI Differential Revision: D41479983 Pull Request resolved: https://github.com/pytorch/pytorch/pull/89521 Approved by: https://github.com/wz337
Author
Committer
Parents
Loading