pytorch
e9e1bb1a - Fix device of info tensor for torch.linalg.inv_ex with MAGMA backend (#59223)

Commit
3 years ago
Fix device of info tensor for torch.linalg.inv_ex with MAGMA backend (#59223) Summary: This PR fixes `torch.linalg.inv_ex` with MAGMA backend. `info` tensor was returned on CPU device even for CUDA inputs. Now it's on the same device as input. Fixes https://github.com/pytorch/pytorch/issues/58769 Pull Request resolved: https://github.com/pytorch/pytorch/pull/59223 Reviewed By: ngimel Differential Revision: D28814876 Pulled By: mruberry fbshipit-source-id: f66c6f06fb8bc305cb2e22b08750a25c8888fb65
Author
Parents
Loading