Skip to content

Commit 81bbaa8

Browse files
ajtullochWei Chen
authored andcommitted
Allow linking against MKLML (apache#2902)
1 parent 42a5068 commit 81bbaa8

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

cmake/modules/contrib/BLAS.cmake

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ elseif(USE_BLAS STREQUAL "mkl")
1010
if(NOT IS_DIRECTORY ${USE_MKL_PATH})
1111
set(USE_MKL_PATH /opt/intel/mkl)
1212
endif()
13-
find_library(BLAS_LIBRARY mkl_rt ${USE_MKL_PATH}/lib/ ${USE_MKL_PATH}/lib/intel64)
13+
find_library(BLAS_LIBRARY NAMES mkl_rt mklml_gnu HINTS ${USE_MKL_PATH}/lib/ ${USE_MKL_PATH}/lib/intel64)
1414
include_directories(${USE_MKL_PATH}/include)
1515
list(APPEND TVM_RUNTIME_LINKER_LIBS ${BLAS_LIBRARY})
1616
list(APPEND RUNTIME_SRCS ${CBLAS_CONTRIB_SRC})

0 commit comments

Comments
 (0)