pytorch
58a74f34 - [17/N] Add _reduce_scatter_base custom op with CPU/CUDA implementation (#88903)

Commit
3 years ago
[17/N] Add _reduce_scatter_base custom op with CPU/CUDA implementation (#88903) Differential Revision: [D41415325](https://our.internmc.facebook.com/intern/diff/D41415325) Pull Request resolved: https://github.com/pytorch/pytorch/pull/88903 Approved by: https://github.com/kwen2501
Author
Committer
Parents
Loading