pytorch
e274c2e4 - [MPS] Restride output strides to contiguous format for inverse op (#102122)

Commit
1 year ago
[MPS] Restride output strides to contiguous format for inverse op (#102122) Remove unnecessary output allocation and reuse the current allocated output Tensor. This change restrides the output strides to contiguous format for inverse op. Pull Request resolved: https://github.com/pytorch/pytorch/pull/102122 Approved by: https://github.com/kulinseth
Author
Committer
Parents
Loading