From 2ffacdebc2cf0917807094c79580aacf95f16869 Mon Sep 17 00:00:00 2001 From: baojun-nervana Date: Mon, 25 Feb 2019 18:44:38 +0000 Subject: [PATCH] Update ngraph version to v0.14 test=develop --- cmake/external/ngraph.cmake | 2 +- paddle/fluid/operators/ngraph/ngraph_engine.cc | 6 +++++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/cmake/external/ngraph.cmake b/cmake/external/ngraph.cmake index 5812a61f0..7edbc87be 100644 --- a/cmake/external/ngraph.cmake +++ b/cmake/external/ngraph.cmake @@ -37,7 +37,7 @@ INCLUDE(GNUInstallDirs) INCLUDE(ExternalProject) SET(NGRAPH_PROJECT "extern_ngraph") -SET(NGRAPH_GIT_TAG "20bd8bbc79ae3a81c57313846a2be7313e5d1dab") +SET(NGRAPH_GIT_TAG "a444f7a959b7d87f2c117c9b57a4c387759e481e") 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) diff --git a/paddle/fluid/operators/ngraph/ngraph_engine.cc b/paddle/fluid/operators/ngraph/ngraph_engine.cc index 660a3298c..41037d903 100644 --- a/paddle/fluid/operators/ngraph/ngraph_engine.cc +++ b/paddle/fluid/operators/ngraph/ngraph_engine.cc @@ -16,7 +16,10 @@ limitations under the License. */ #include #include +#include #include +#include +#include #include #include "paddle/fluid/framework/block_desc.h" @@ -483,7 +486,8 @@ void NgraphEngine::Run(const framework::Scope& scope, } } - backend_->call(backend_->compile(ngraph_function_), t_out, t_in); + auto handle = backend_->compile(ngraph_function_); + handle->call_with_validate(t_out, t_in); } // NgraphEngine::Run } // namespace operators } // namespace paddle -- GitLab