pytorch
ac334326 - Fixed out= variants of non-symmetric eigendecomposition and QR decomposition (#54056)

Commit
3 years ago
Fixed out= variants of non-symmetric eigendecomposition and QR decomposition (#54056) Summary: This PR modifies the behavior of _out variants of `torch.eig`, `torch.qr`, `torch.linalg.qr` to match the description here https://github.com/pytorch/pytorch/wiki/Developer-FAQ#how-does-out-work-in-pytorch With this PR result and input tensors must be on the same device and have the same "type kind". Tested with OpInfo's `supports_out=True`. Ref. https://github.com/pytorch/pytorch/issues/42666 Pull Request resolved: https://github.com/pytorch/pytorch/pull/54056 Reviewed By: heitorschueroff Differential Revision: D27230275 Pulled By: mruberry fbshipit-source-id: 3fe1ce6c0e2c20bdfd6742305a20f3cf3632a4d6
Author
Parents
Loading