pytorch
627a3312 - Port CPU torch.orgqr to ATen (#50502)

Commit
3 years ago
Port CPU torch.orgqr to ATen (#50502) Summary: Now we can remove `_th_orgqr`! Compared to the original TH-based `orgqr`, complex (https://github.com/pytorch/pytorch/issues/33152) and batched inputs are now supported. CUDA support will be added in a follow-up PR. Closes https://github.com/pytorch/pytorch/issues/24747 Ref. https://github.com/pytorch/pytorch/issues/49421, https://github.com/pytorch/pytorch/issues/42666 Pull Request resolved: https://github.com/pytorch/pytorch/pull/50502 Reviewed By: mrshenli Differential Revision: D25953300 Pulled By: mruberry fbshipit-source-id: f52a74e1c8f51b5e24f7b461430ca8fc96e4d149
Author
Parents
Loading