Skip to content

T
tic
Commit 230dfe8a by Andrew Tulloch Committed by Tianqi Chen
Options

Allow linking against MKLML (#2902)

parent 84cb712a
Showing with 1 additions and 1 deletions
cmake/modules/contrib/BLAS.cmake
...@@ -10,7 +10,7 @@ elseif(USE_BLAS STREQUAL "mkl") ...@@ -10,7 +10,7 @@ elseif(USE_BLAS STREQUAL "mkl")
if(NOT IS_DIRECTORY ${USE_MKL_PATH}) if(NOT IS_DIRECTORY ${USE_MKL_PATH})
set(USE_MKL_PATH /opt/intel/mkl) set(USE_MKL_PATH /opt/intel/mkl)
endif() endif()
find_library(BLAS_LIBRARY mkl_rt ${USE_MKL_PATH}/lib/ ${USE_MKL_PATH}/lib/intel64) find_library(BLAS_LIBRARY NAMES mkl_rt mklml_gnu HINTS ${USE_MKL_PATH}/lib/ ${USE_MKL_PATH}/lib/intel64)
include_directories(${USE_MKL_PATH}/include) include_directories(${USE_MKL_PATH}/include)
list(APPEND TVM_RUNTIME_LINKER_LIBS ${BLAS_LIBRARY}) list(APPEND TVM_RUNTIME_LINKER_LIBS ${BLAS_LIBRARY})
list(APPEND RUNTIME_SRCS ${CBLAS_CONTRIB_SRC}) list(APPEND RUNTIME_SRCS ${CBLAS_CONTRIB_SRC})
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment