Browse Source

Custom pass register.

tags/v1.2.0
unknown 4 years ago
parent
commit
0566c6dc3f
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

@@ -769,7 +769,7 @@ Status GraphManager::SetRtContext(rtContext_t rt_context, rtCtxMode_t mode, uint
Status GraphManager::RunCustomPass(const GraphNodePtr &graph_node) { 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); auto comp_graph = GraphUtils::GetComputeGraph(*const_graph);
GE_DUMP(compute_graph, "RunCustomPassBegin");
GE_DUMP(comp_graph, "RunCustomPassBegin");


GE_TIMESTAMP_START(RunCustomPass); GE_TIMESTAMP_START(RunCustomPass);
GraphPtr graph = std::const_pointer_cast<Graph>(const_graph); GraphPtr graph = std::const_pointer_cast<Graph>(const_graph);


Loading…
Cancel
Save