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

问题描述

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

解决方法

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

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