pytorch
3b237529 - [PT-D][Sharding] Fix view op and matrix ops unit test

Commit
2 years ago
[PT-D][Sharding] Fix view op and matrix ops unit test To fix a corner case when the sharding dim is negative number we need to handle it correctly. Also disable RPC for matrix ops which are not necessary and they fail on AWS pytest. Differential Revision: [D36464797](https://our.internmc.facebook.com/intern/diff/D36464797/) Pull Request resolved: https://github.com/pytorch/pytorch/pull/77706 Approved by: https://github.com/pritamdamania87, https://github.com/wanchaol
Author
Committer
Parents
Loading