forked from OSSInnovation/mindspore
!4715 Fix open source ge backend compiling
Merge pull request !4715 from zhoufeng/xiu-ba-ge
This commit is contained in:
commit
52b1b6a189
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue