Browse Source

For dynamic shape complie.

pull/656/head
unknown 5 years ago
parent
commit
159633f73d
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      ge/graph/build/graph_builder.cc

+ 2
- 2
ge/graph/build/graph_builder.cc View File

@@ -277,7 +277,7 @@ Status GraphBuilder::BuildForUnknownShapeGraph(ComputeGraphPtr &comp_graph, GeMo
}

Graph2SubGraphInfoList subgraph_map;
ge::ModelBuilder builder(session_id, com_graph, subgraph_map, stream_max_parallel_num_, hcom_parallel_, build_mode_);
ge::ModelBuilder builder(session_id, comp_graph, subgraph_map, stream_max_parallel_num_, hcom_parallel_, build_mode_);
GE_CHK_STATUS_RET(builder.RreBuildModel(), "Failed to pre build model.");
GELOGI("Begin to build unknown shape graph[%s].", comp_graph->GetName().c_str());
@@ -288,7 +288,7 @@ Status GraphBuilder::BuildForUnknownShapeGraph(ComputeGraphPtr &comp_graph, GeMo
GE_DUMP(comp_graph, "AfterCalcOpParam");


for (auto &node : comp_graph->GetsDirectNode()) {
for (auto &node : comp_graph->GetDirectNode()) {
GE_CHECK_NOTNULL(node);
auto op_desc = node->GetOpDesc();
GE_CHECK_NOTNULL(op_desc);


Loading…
Cancel
Save