[Fix] Fix blas cmake for mac os (#3898)

* fix cmake for mac os * rename
apache · Sep 7, 2019 · 7a15aed · 7a15aed
1 parent 8c50469
commit 7a15aed
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/cmake/modules/contrib/BLAS.cmake b/cmake/modules/contrib/BLAS.cmake
@@ -28,17 +28,17 @@ elseif(USE_BLAS STREQUAL "mkl")
     set(USE_MKL_PATH /opt/intel/mkl)
   endif()
   if(APPLE)
-    find_library(BLAS_LIBRARY NAMES mklml mkl_rt HINTS ${USE_MKL_PATH}/lib/ ${USE_MKL_PATH}/lib/intel64)
+    find_library(BLAS_LIBRARY_MKL NAMES mklml mkl_rt 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)
+    find_library(BLAS_LIBRARY_MKL 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)
+    find_library(BLAS_LIBRARY_MKL 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})
+  list(APPEND TVM_RUNTIME_LINKER_LIBS ${BLAS_LIBRARY_MKL})
   list(APPEND RUNTIME_SRCS ${CBLAS_CONTRIB_SRC})
   add_definitions(-DUSE_MKL_BLAS=1)
-  message(STATUS "Use BLAS library " ${BLAS_LIBRARY})
+  message(STATUS "Use BLAS library " ${BLAS_LIBRARY_MKL})
 elseif(USE_BLAS STREQUAL "atlas" OR USE_BLAS STREQUAL "blas")
   find_library(BLAS_LIBRARY cblas)
   list(APPEND TVM_RUNTIME_LINKER_LIBS ${BLAS_LIBRARY})