From 17d60b2ccb78ca77bf3794008c74dc440d699ce0 Mon Sep 17 00:00:00 2001 From: zhoufeng Date: Wed, 19 Aug 2020 10:17:11 +0800 Subject: [PATCH] fix open source ge backend compile Signed-off-by: zhoufeng --- mindspore/ccsrc/pipeline/jit/pipeline.cc | 3 ++- mindspore/ccsrc/transform/graph_ir/convert.cc | 3 +++ mindspore/ccsrc/transform/graph_ir/graph_runner.cc | 1 + 3 files changed, 6 insertions(+), 1 deletion(-) diff --git a/mindspore/ccsrc/pipeline/jit/pipeline.cc b/mindspore/ccsrc/pipeline/jit/pipeline.cc index 7a70f68a18..35eb5b9a36 100644 --- a/mindspore/ccsrc/pipeline/jit/pipeline.cc +++ b/mindspore/ccsrc/pipeline/jit/pipeline.cc @@ -53,6 +53,7 @@ #include "pipeline/jit/pipeline_ge.h" #include "transform/graph_ir/convert.h" #include "transform/graph_ir/df_graph_manager.h" +#include "transform/graph_ir/op_adapter_map.h" #endif namespace mindspore { @@ -1038,7 +1039,7 @@ void ClearResAtexit() { opt::python_pass::PyPassManager::GetInstance()->ClearRes(); #ifdef ENABLE_GE transform::DfGraphManager::GetInstance().ClearGraph(); - transform::DfGraphConvertor::get_adpt_map().clear(); + transform::OpAdapterMap::get().clear(); #endif ReleaseGeTsd(); parse::python_adapter::ResetPythonScope(); diff --git a/mindspore/ccsrc/transform/graph_ir/convert.cc b/mindspore/ccsrc/transform/graph_ir/convert.cc index c48e0833e1..42e7004c24 100644 --- a/mindspore/ccsrc/transform/graph_ir/convert.cc +++ b/mindspore/ccsrc/transform/graph_ir/convert.cc @@ -34,6 +34,9 @@ #include "ops/array_ops.h" #include "ops/elewise_calculation_ops.h" #include "ops/math_ops.h" +#ifdef ENABLE_GE +#include "ops/save_ops.h" +#endif namespace mindspore { namespace transform { diff --git a/mindspore/ccsrc/transform/graph_ir/graph_runner.cc b/mindspore/ccsrc/transform/graph_ir/graph_runner.cc index 2d3eb028aa..15db6283f1 100644 --- a/mindspore/ccsrc/transform/graph_ir/graph_runner.cc +++ b/mindspore/ccsrc/transform/graph_ir/graph_runner.cc @@ -23,6 +23,7 @@ #include "sys/time.h" #include "utils/utils.h" #include "./common.h" +#include "utils/callbacks.h" #ifdef ENABLE_GE #include "utils/callbacks_ge.h" #endif