From: Haichen Shen Date: Sat, 7 Sep 2019 21:34:32 +0000 (-0700) Subject: [Fix] Fix blas cmake for mac os (#3898) X-Git-Tag: upstream/0.7.0~1935 X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=7a15aedf1ee72e0d4ff7cd69322705cab73081d2;p=platform%2Fupstream%2Ftvm.git [Fix] Fix blas cmake for mac os (#3898) * fix cmake for mac os * rename --- 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})