pytorch
ae6f07e7 - [MPS] Fix std/var cache issue (#80502)

Commit
2 years ago
[MPS] Fix std/var cache issue (#80502) Use `getTensorsStringKey` which has tensor shape info added as part of the key to prevent cache lookup issue when the shape of input tensor is changed. Fixes #80499 Pull Request resolved: https://github.com/pytorch/pytorch/pull/80502 Approved by: https://github.com/malfet, https://github.com/kulinseth
Author
Committer
Parents
Loading