Join GitHub today
GitHub is home to over 31 million developers working together to host and review code, manage projects, and build software together.Sign up
cleanup BLAS detection #8561
PyTorch and Caffe2 use the following libraries as preferred for BLAS/LAPACK capabilities:
There are however additional subtelities.
Additionally, we also use fortran-blas interface in pytorch, cblas interface in Caffe2 (if not Eigen), and eigen::MatrixMap interface if Eigen is found and
Thanks! Pasting replies from internal chat here:
In general, it sounds reasonable to use the BLAS found by CMake first, and use Eigen as the last option if others are not found.
FYI, the fallback to Eigen from MKL roughly matches what is being done at https://github.com/pytorch/pytorch/blob/master/cmake/Dependencies.cmake#L73-L77 so makes sense.