CODEWITHSUNDEEP
pytorchlinear-algebraimplementation
Gravifer
Gravifer

Reputation: 141

How to find the source code of torch.solve?

I am concerned whether torch.solve() examine the condition of the coefficient matrix for a linear system and employ desirable preconditionings; thus I am curious about its implementation details. I have read through several answers trying to track down the source file but in vain. I hope somebody can help me to locate its definition in the ATen library.

Upvotes: 4

Views: 700

Answers (1)

Multihunter
Multihunter

Reputation: 5928

I think it just uses LAPACK for CPU and CUBLAS for GPU, since torch.solve is listed under "BLAS and LAPACK Operations" on the official docs.

Then we're looking for wrapper code, which I believe is this part.

Upvotes: 3

Related Questions