Which linear algebra library been used by ATen?

Looking into the tensor library used by pytorch, wondering which linear algebra package was used for some of the tensor operations.

TensorFlow seems using Eigen from their code.

2 Likes

On the CPU, we use Intel MKL to ship with our binaries. ATen can use other Linear Algebra libraries as well, that expose a LAPACK interface, such as OpenBLAS, Eigen.
On the GPU, we use cuBLAS, cuSolver and MAGMA

4 Likes