Custom pass register.

4 years ago · b706aa1da3
parent 0566c6dc3f
commit b706aa1da3
1 changed files with 1 additions and 1 deletions
--- a/ge/graph/manager/graph_manager.cc
+++ b/ge/graph/manager/graph_manager.cc
@ -767,7 +767,7 @@ Status GraphManager::SetRtContext(rtContext_t rt_context, rtCtxMode_t mode, uint
 }

 Status GraphManager::RunCustomPass(const GraphNodePtr &graph_node) {
-  ConstGraphPtr const_graph = graph->node->GetGraph();
+  ConstGraphPtr const_graph = graph_node->GetGraph();
  auto comp_graph = GraphUtils::GetComputeGraph(*const_graph);
  GE_DUMP(comp_graph, "RunCustomPassBegin");