From b706aa1da3044d6fb2c02951b190e70f1683433d Mon Sep 17 00:00:00 2001 From: unknown Date: Thu, 31 Dec 2020 11:00:04 +0800 Subject: [PATCH] Custom pass register. --- ge/graph/manager/graph_manager.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ge/graph/manager/graph_manager.cc b/ge/graph/manager/graph_manager.cc index c102ec2e..706908af 100755 --- 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");