Browse Source

!4715 Fix open source ge backend compiling

Merge pull request !4715 from zhoufeng/xiu-ba-ge
tags/v0.7.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
52b1b6a189
3 changed files with 6 additions and 1 deletions
  1. +2
    -1
      mindspore/ccsrc/pipeline/jit/pipeline.cc
  2. +3
    -0
      mindspore/ccsrc/transform/graph_ir/convert.cc
  3. +1
    -0
      mindspore/ccsrc/transform/graph_ir/graph_runner.cc

+ 2
- 1
mindspore/ccsrc/pipeline/jit/pipeline.cc 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();


+ 3
- 0
mindspore/ccsrc/transform/graph_ir/convert.cc 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 {


+ 1
- 0
mindspore/ccsrc/transform/graph_ir/graph_runner.cc 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


Loading…
Cancel
Save