提交 17d60b2c 编写于 作者: Z zhoufeng

fix open source ge backend compile

Signed-off-by: Nzhoufeng <zhoufeng54@huawei.com>
上级 9001dc48
......@@ -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();
......
......@@ -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 {
......
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册