From e6f62f79c3e783e3d8da4b76cb601a1590ea937e Mon Sep 17 00:00:00 2001 From: tensor-tang Date: Wed, 2 Aug 2017 23:55:30 +0800 Subject: [PATCH] add meesage and cmake cache arg --- cmake/external/mkldnn.cmake | 2 ++ 1 file changed, 2 insertions(+) diff --git a/cmake/external/mkldnn.cmake b/cmake/external/mkldnn.cmake index 33988c66b2..25c6b4ef52 100644 --- a/cmake/external/mkldnn.cmake +++ b/cmake/external/mkldnn.cmake @@ -43,6 +43,7 @@ IF(${CBLAS_PROVIDER} STREQUAL "MKLML") SET(MKLDNN_MKLROOT ${MKLML_ROOT}) SET(MKLDNN_IOMP_LIB ${MKLML_IOMP_LIB}) SET(MKLDNN_IOMP_DIR ${MKLML_LIB_DIR}) + MESSAGE(STATUS "Build MKLDNN with ${MKLDNN_MKLROOT}") ENDIF() ExternalProject_Add( @@ -56,6 +57,7 @@ ExternalProject_Add( CMAKE_ARGS -DCMAKE_INSTALL_PREFIX=${MKLDNN_INSTALL_DIR} CMAKE_ARGS -DMKLROOT=${MKLDNN_MKLROOT} CMAKE_CACHE_ARGS -DCMAKE_INSTALL_PREFIX:PATH=${MKLDNN_INSTALL_DIR} + -DMKLROOT:PATH=${MKLDNN_MKLROOT} ) ADD_LIBRARY(mkldnn SHARED IMPORTED GLOBAL) -- GitLab