From: Jason Knight Date: Fri, 6 Sep 2019 13:30:13 +0000 (-0700) Subject: Add another MKL name alias for MKL (#3853) X-Git-Tag: upstream/0.7.0~1946 X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=d464d2baa773cd9508b95874c42d41230439ce76;p=platform%2Fupstream%2Ftvm.git Add another MKL name alias for MKL (#3853) Installed through pypi --- diff --git a/cmake/config.cmake b/cmake/config.cmake index a8ed966..2a61099 100644 --- 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 ) 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 index 44f3181..62fb1d8 100644 --- 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) elseif(MSVC)