From 7d87e5423599694a5a282e7702c096be215517f3 Mon Sep 17 00:00:00 2001
From: lichun <lichun30@hisilicon.com>
Date: Mon, 15 Mar 2021 21:28:57 +0800
Subject: [PATCH] fix compile error

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

diff --git a/inc/framework/generator/ge_generator.h b/inc/framework/generator/ge_generator.h
index adc6e8c7..7d59def7 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,8 @@ 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;