-
Notifications
You must be signed in to change notification settings - Fork 27.7k
Request for conjugate transposed triangular solve #45734
Copy link
Copy link
Closed
Labels
module: complexRelated to complex number support in PyTorchRelated to complex number support in PyTorchmodule: linear algebraIssues related to specialized linear algebra operations in PyTorch; includes matrix multiply matmulIssues related to specialized linear algebra operations in PyTorch; includes matrix multiply matmultriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate moduleThis issue has been looked at a team member, and triaged and prioritized into an appropriate module
Metadata
Metadata
Assignees
Labels
module: complexRelated to complex number support in PyTorchRelated to complex number support in PyTorchmodule: linear algebraIssues related to specialized linear algebra operations in PyTorch; includes matrix multiply matmulIssues related to specialized linear algebra operations in PyTorch; includes matrix multiply matmultriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate moduleThis issue has been looked at a team member, and triaged and prioritized into an appropriate module
🚀 Feature
torch.triangular_solveshould have an option to solve A^H x = b, where "^H" is the conjugate transpose operation.Motivation
Now that PyTorch is expanding the complex number support, linear algebra methods that have transposed variants usually also have conjugate transpose variants for complex inputs. That should be supported as well.
Additional context
SciPy's triangular_solve has this option and the underlying BLAS/LAPACK libraries allow that (with
trans='C').cc @ezyang @anjali411 @dylanbespalko @mruberry @vincentqb @vishwakftw @jianyuh @nikitaved @pearu