Skip to content

Commit d88e155

Browse files
ajtullochMarisaKirisame
authored andcommitted
Allow linking against MKLML (apache#2902)
1 parent 31ff0e3 commit d88e155

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

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)