pytorch
04ad0134 - [FSDP] Use `reduce_scatter_tensor()` (#87240)

Commit
3 years ago
[FSDP] Use `reduce_scatter_tensor()` (#87240) Let us silence some more warnings 👍🏼 Pull Request resolved: https://github.com/pytorch/pytorch/pull/87240 Approved by: https://github.com/rohan-varma
Author
Andrew Gu
Committer
Parents
Loading