pytorch
3ae612ba - [dtensor] remove assertions about submesh checks (#101229)

Commit
1 year ago
[dtensor] remove assertions about submesh checks (#101229) This PR removes assertions from submesh checks to directly return local tensor, this is so that all the other APIs can work with submesh Pull Request resolved: https://github.com/pytorch/pytorch/pull/101229 Approved by: https://github.com/fduwjj
Author
Committer
Parents
Loading