From 41932726e4aaefd47ea44b3dd510a529ca380ea7 Mon Sep 17 00:00:00 2001 From: Andrew Tulloch Date: Tue, 26 Mar 2019 15:04:48 -0700 Subject: [PATCH] Allow linking against MKLML --- cmake/modules/contrib/BLAS.cmake | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cmake/modules/contrib/BLAS.cmake b/cmake/modules/contrib/BLAS.cmake index 45269a20715db..3a82860d10433 100644 --- a/cmake/modules/contrib/BLAS.cmake +++ b/cmake/modules/contrib/BLAS.cmake @@ -10,7 +10,7 @@ elseif(USE_BLAS STREQUAL "mkl") if(NOT IS_DIRECTORY ${USE_MKL_PATH}) set(USE_MKL_PATH /opt/intel/mkl) endif() - find_library(BLAS_LIBRARY mkl_rt ${USE_MKL_PATH}/lib/ ${USE_MKL_PATH}/lib/intel64) + find_library(BLAS_LIBRARY mkl_rt mklml_gnu ${USE_MKL_PATH}/lib/ ${USE_MKL_PATH}/lib/intel64) include_directories(${USE_MKL_PATH}/include) list(APPEND TVM_RUNTIME_LINKER_LIBS ${BLAS_LIBRARY}) list(APPEND RUNTIME_SRCS ${CBLAS_CONTRIB_SRC})