From d8be4eae7579bc2e975d5d939ce604479ee7116b Mon Sep 17 00:00:00 2001
From: lichun <lichun30@hisilicon.com>
Date: Wed, 30 Jun 2021 20:32:31 +0800
Subject: [PATCH] ReshapeRemovePass_not_remove_reshape_in_dynamic_subgraph

---
 .../ops_kernel_store/ge_local_ops_kernel_builder.cc           | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/ge/ge_local_engine/ops_kernel_store/ge_local_ops_kernel_builder.cc b/ge/ge_local_engine/ops_kernel_store/ge_local_ops_kernel_builder.cc
index b822f574..f6b88e7e 100644
--- a/ge/ge_local_engine/ops_kernel_store/ge_local_ops_kernel_builder.cc
+++ b/ge/ge_local_engine/ops_kernel_store/ge_local_ops_kernel_builder.cc
@@ -22,6 +22,7 @@
 #include "graph/utils/node_utils.h"
 #include "graph/utils/tensor_utils.h"
 #include "graph/utils/type_utils.h"
+#include "graph/compute_graph.h"
 #include "ge_local_engine/ops_kernel_store/op/op_factory.h"
 #include "ge_local_engine/common/constant/constant.h"
 #include "register/ops_kernel_builder_registry.h"
@@ -164,8 +165,7 @@ Status GeLocalOpsKernelBuilder::CalcConstantStrMemSize(const OpDescPtr &op_desc,
 Status GeLocalOpsKernelBuilder::GenerateTask(const Node &node, RunContext &context, std::vector<domi::TaskDef> &tasks) {
   bool is_in_unknown_subgraph = false;
   bool forced_unknown = false;
-  ComputeGraphPtr owner_graph = node.GetOwnerComputeGraph();
-  for (const auto &node : owner_graph->GetDirectNode()) {
+  for (const auto &node : node.GetOwnerComputeGraph()->GetDirectNode()) {
     GE_CHK_GRAPH_STATUS_RET(ge::NodeUtils::GetNodeUnknownShapeStatus(*node, is_in_unknown_subgraph),
                             "[Get][ShapeStatus] of node[%s] failed!", node->GetName().c_str());
     if (is_in_unknown_subgraph) {