From 7a15aedf1ee72e0d4ff7cd69322705cab73081d2 Mon Sep 17 00:00:00 2001 From: Haichen Shen Date: Sat, 7 Sep 2019 14:34:32 -0700 Subject: [PATCH] [Fix] Fix blas cmake for mac os (#3898) * fix cmake for mac os * rename --- cmake/modules/contrib/BLAS.cmake | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/cmake/modules/contrib/BLAS.cmake b/cmake/modules/contrib/BLAS.cmake index 62fb1d8..2efe877 100644 --- 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}) -- 2.7.4