pytorch
f9e69af2 - Modify LU_backward and lu_solve_backward to use linalg_solve_triangular (#63569)

Commit
2 years ago
Modify LU_backward and lu_solve_backward to use linalg_solve_triangular (#63569) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/63569 This PR also rewrites `lu_solve_backward` from scratch going from solving 5 systems of equations to just 2. cc jianyuh nikitaved pearu mruberry walterddr IvanYashchuk xwang233 Lezcano Test Plan: Imported from OSS Reviewed By: mruberry Differential Revision: D32618014 Pulled By: anjali411 fbshipit-source-id: 0e915bcf7045a4db43ffd076d807beac816c8538
Author
Parents
Loading