pytorch
1880d293 - Expose lu_factor_batched_cublas

Commit
2 years ago
Expose lu_factor_batched_cublas We had bindings for this function, but they were not exposed as a function inside ATen. This PR exposes them. This function is tested in the next PR of the stack, where it is used in linalg.lu_solve. Pull Request resolved: https://github.com/pytorch/pytorch/pull/73877 Approved by: https://github.com/nikitaved, https://github.com/IvanYashchuk, https://github.com/mruberry
Author
Committer
Parents
Loading