From 25bb9e7af8b2f58fc771ff859d48e8473aaa8b70 Mon Sep 17 00:00:00 2001
From: lichun <lichun30@hisilicon.com>
Date: Mon, 15 Mar 2021 19:31:24 +0800
Subject: [PATCH] fix compile fail bug

---
 inc/framework/generator/ge_generator.h | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/inc/framework/generator/ge_generator.h b/inc/framework/generator/ge_generator.h
index adc6e8c7..b5fc73c3 100644
--- a/inc/framework/generator/ge_generator.h
+++ b/inc/framework/generator/ge_generator.h
@@ -29,9 +29,9 @@
 #include "graph/op_desc.h"
 #include "graph/detail/attributes_holder.h"
 #include "omg/omg_inner_types.h"
-#include "model/ge_root_model.h"
 
 namespace ge {
+class GeRootModel;
 class GE_FUNC_VISIBILITY GeGenerator {
  public:
   static GeGenerator &GetInstance() {
@@ -99,6 +99,7 @@ class GE_FUNC_VISIBILITY GeGenerator {
                        const string &model_file_name, OpEngineType engine_type, ModelBufferData &model_buff,
                        bool is_offline = true);
   Status CheckForSingleOp(OpDescPtr &op_desc, const vector<GeTensor> &inputs, const vector<GeTensor> &outputs);
+  using GeRootModelPtr = std::shared_ptr<ge::GeRootModel>;
   Status SetModelNameForDump(const GeRootModelPtr &ge_root_model);
 
   class Impl;