Support MKL on Windows (#3837)

710ac146 · Jon Soifer · Tianqi Chen · 062f8cc4 · 710ac146
Commit 710ac146 authored Aug 28, 2019 by Jon Soifer Committed by Tianqi Chen Aug 29, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

cmake/modules/contrib/BLAS.cmake
+2 -0

No files found.
--- a/cmake/modules/contrib/BLAS.cmake
+++ b/cmake/modules/contrib/BLAS.cmake
@@ -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)
  elseif(UNIX)
    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()
  include_directories(${USE_MKL_PATH}/include)
  list(APPEND TVM_RUNTIME_LINKER_LIBS ${BLAS_LIBRARY})