From b1b98f17c99e3ef295bd69c37c4bcb4b65529fd0 Mon Sep 17 00:00:00 2001
From: wuweikang <wuweikang@huawei.com>
Date: Tue, 22 Jun 2021 16:35:06 +0800
Subject: [PATCH] fix build for single op

---
 ge/hybrid/model/hybrid_model_builder.cc | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/ge/hybrid/model/hybrid_model_builder.cc b/ge/hybrid/model/hybrid_model_builder.cc
index c050875e..f168c3ad 100755
--- a/ge/hybrid/model/hybrid_model_builder.cc
+++ b/ge/hybrid/model/hybrid_model_builder.cc
@@ -141,6 +141,7 @@ Status CollectDependenciesForFusedGraph(NodeItem &node_item, std::set<OpDesc *>
 HybridModelBuilder::HybridModelBuilder(HybridModel &hybrid_model)
     : hybrid_model_(hybrid_model), runtime_param_(hybrid_model.root_runtime_param_) {
   ge_root_model_ = hybrid_model_.ge_root_model_;
+  hybrid_model_.root_graph_ = ge_root_model_->GetRootGraph();
 }
 
 Status HybridModelBuilder::Build() {
@@ -852,6 +853,7 @@ Status HybridModelBuilder::LoadGraph() {
            root_graph->GetAllNodesSize());
   }
 
+  hybrid_model_.root_graph_ = root_graph;
   GE_CHK_STATUS_RET(RelinkNextIteration(), "[%s] Relink NextIteration failed", GetGraphName());
   // Reset node id by topological order across all subgraphs
   int64_t index = 0;