Add another MKL name alias for MKL installed through pypi

apache · Aug 28, 2019 · eca78e4 · eca78e4
1 parent 062f8cc
commit eca78e4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/cmake/config.cmake b/cmake/config.cmake
@@ -109,6 +109,7 @@ set(USE_BLAS none)
 # /path/to/mkl: mkl root path when use mkl blas library
 # set(USE_MKL_PATH /opt/intel/mkl) for UNIX
 # set(USE_MKL_PATH ../IntelSWTools/compilers_and_libraries_2018/windows/mkl) for WIN32
+# set(USE_MKL_PATH <path to venv or site-packages directory>) if using `pip install mkl`
 set(USE_MKL_PATH none)
 
 # Whether use contrib.random in runtime

diff --git a/cmake/modules/contrib/BLAS.cmake b/cmake/modules/contrib/BLAS.cmake
@@ -28,7 +28,7 @@ elseif(USE_BLAS STREQUAL "mkl")
     set(USE_MKL_PATH /opt/intel/mkl)
   endif()
   if(APPLE)
-    find_library(BLAS_LIBRARY NAMES mklml HINTS ${USE_MKL_PATH}/lib/ ${USE_MKL_PATH}/lib/intel64)
+    find_library(BLAS_LIBRARY NAMES mklml mkl_rt HINTS ${USE_MKL_PATH}/lib/ ${USE_MKL_PATH}/lib/intel64)
   elseif(UNIX)
     find_library(BLAS_LIBRARY NAMES mkl_rt mklml_gnu HINTS ${USE_MKL_PATH}/lib/ ${USE_MKL_PATH}/lib/intel64)
   endif()