|
|
|
@ -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})
|
|
|
|
|