Which linear algebra library been used by ATen?

1

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
2

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

5 Likes