upgrade ngraph to support mkldnn v1.0 (#19689)

new_fix
baojun 5 years ago committed by Tao Luo
parent 25dcd74d34
commit 87f13f7569

@ -37,7 +37,7 @@ INCLUDE(GNUInstallDirs)
INCLUDE(ExternalProject)
SET(NGRAPH_PROJECT "extern_ngraph")
SET(NGRAPH_GIT_TAG "v0.24.0-rc.2")
SET(NGRAPH_GIT_TAG "e26d602a756f5f83e6c8220f910b61d7089fa951")
SET(NGRAPH_SOURCES_DIR ${THIRD_PARTY_PATH}/ngraph)
SET(NGRAPH_INSTALL_DIR ${THIRD_PARTY_PATH}/install/ngraph)
SET(NGRAPH_INC_DIR ${NGRAPH_INSTALL_DIR}/include)
@ -76,6 +76,7 @@ ExternalProject_Add(
CMAKE_ARGS -DMKLDNN_INCLUDE_DIR=${MKLDNN_INC_DIR}
CMAKE_ARGS -DMKLDNN_LIB_DIR=${MKLDNN_INSTALL_DIR}/${CMAKE_INSTALL_LIBDIR}
CMAKE_ARGS -DMKLML_LIB_DIR=${MKLML_INSTALL_DIR}/lib
CMAKE_ARGS -NGRAPH_USE_LEGACY_MKLDNN=TRUE
)
add_dependencies(ngraph ${NGRAPH_PROJECT})

Loading…
Cancel
Save