Browse Source

Custom pass register.

tags/v1.2.0
unknown 4 years ago
parent
commit
b706aa1da3
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/graph/manager/graph_manager.cc

+ 1
- 1
ge/graph/manager/graph_manager.cc View File

@@ -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");



Loading…
Cancel
Save