pytorch
ada7dfff - fix allgather func collective to use maybe_wrap_tensor (#98866)

Commit
1 year ago
fix allgather func collective to use maybe_wrap_tensor (#98866) It looks like we forgot to switch allgather to use maybe_wrap_tensor, this PR switch to use that and added test to guard tracing behavior Pull Request resolved: https://github.com/pytorch/pytorch/pull/98866 Approved by: https://github.com/mrshenli
Author
Committer
Parents
Loading