pytorch
c7bd9b94 - Switch AsyncCollectiveTensor to be a wrapper subclass. (#96105)

Commit
1 year ago
Switch AsyncCollectiveTensor to be a wrapper subclass. (#96105) Our usage is of a wrapper, so it makes sense that we use one. This makes it possible for FakeTensorMode to work. Pull Request resolved: https://github.com/pytorch/pytorch/pull/96105 Approved by: https://github.com/wanchaol, https://github.com/wconstab
Author
Rodrigo Kumpera
Committer
Parents
Loading