如何找到torch.solve的源代码？

Question

我担心是否 torch.solve() 检查线性系统的系数矩阵的条件并采用理想的预处理；因此我很好奇它的实现细节。我已经阅读了几个试图追踪源文件的答案，但没有成功。我希望有人能帮我找到它在 ATen 库中的定义。

Answer 1

我认为它只是将 LAPACK 用于 CPU，将 CUBLAS 用于 GPU，因为 torch.solve 列在 official docs 的“BLAS 和 LAPACK 操作”下。

然后我们正在寻找包装代码，我相信它是 this part。