Skip to content

T
tic
Commit 710ac146 by Jon Soifer Committed by Tianqi Chen
Options

Support MKL on Windows (#3837)

parent 062f8cc4
Showing with 2 additions and 0 deletions
cmake/modules/contrib/BLAS.cmake
...@@ -31,6 +31,8 @@ elseif(USE_BLAS STREQUAL "mkl") ...@@ -31,6 +31,8 @@ elseif(USE_BLAS STREQUAL "mkl")
find_library(BLAS_LIBRARY NAMES mklml HINTS ${USE_MKL_PATH}/lib/ ${USE_MKL_PATH}/lib/intel64) find_library(BLAS_LIBRARY NAMES mklml HINTS ${USE_MKL_PATH}/lib/ ${USE_MKL_PATH}/lib/intel64)
elseif(UNIX) elseif(UNIX)
find_library(BLAS_LIBRARY NAMES mkl_rt mklml_gnu HINTS ${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)
elseif(MSVC)
find_library(BLAS_LIBRARY NAMES mkl_rt HINTS ${USE_MKL_PATH}/lib/ ${USE_MKL_PATH}/lib/intel64_win)
endif() endif()
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})
......
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