!4715 Fix open source ge backend compiling

Merge pull request !4715 from zhoufeng/xiu-ba-ge
This commit is contained in:
mindspore-ci-bot 2020-08-19 14:55:52 +08:00 committed by Gitee
commit 52b1b6a189
3 changed files with 6 additions and 1 deletions

View File

@ -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();

View File

@ -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 {

View File

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