|
|
|
@ -45,9 +45,9 @@ IF(${CBLAS_PROVIDER} STREQUAL "MKLML")
|
|
|
|
|
ELSE()
|
|
|
|
|
MESSAGE(FATAL_ERROR "Should enable MKLML when build MKLDNN")
|
|
|
|
|
ENDIF()
|
|
|
|
|
|
|
|
|
|
SET(MKLDNN_CFLAG "${CMAKE_C_FLAGS} -Wno-error=strict-overflow")
|
|
|
|
|
SET(MKLDNN_CXXFLAG "${CMAKE_CXX_FLAGS} -Wno-error=strict-overflow")
|
|
|
|
|
SET(MKLDNN_FLAG "-Wno-error=strict-overflow -Wno-error=unused-result -Wno-unused-result")
|
|
|
|
|
SET(MKLDNN_CFLAG "${CMAKE_C_FLAGS} ${MKLDNN_FLAG}")
|
|
|
|
|
SET(MKLDNN_CXXFLAG "${CMAKE_CXX_FLAGS} ${MKLDNN_FLAG}")
|
|
|
|
|
ExternalProject_Add(
|
|
|
|
|
${MKLDNN_PROJECT}
|
|
|
|
|
${EXTERNAL_PROJECT_LOG_ARGS}
|
|
|
|
@ -61,6 +61,7 @@ ExternalProject_Add(
|
|
|
|
|
CMAKE_ARGS -DMKLROOT=${MKLML_ROOT}
|
|
|
|
|
CMAKE_ARGS -DCMAKE_C_FLAGS=${MKLDNN_CFLAG}
|
|
|
|
|
CMAKE_ARGS -DCMAKE_CXX_FLAGS=${MKLDNN_CXXFLAG}
|
|
|
|
|
CMAKE_ARGS -DWITH_TEST=OFF -DWITH_EXAMPLE=OFF
|
|
|
|
|
CMAKE_CACHE_ARGS -DCMAKE_INSTALL_PREFIX:PATH=${MKLDNN_INSTALL_DIR}
|
|
|
|
|
-DMKLROOT:PATH=${MKLML_ROOT}
|
|
|
|
|
)
|
|
|
|
|