From 159633f73d0fcb300fc9336baeb5b7a31cc79fb1 Mon Sep 17 00:00:00 2001 From: unknown Date: Fri, 18 Dec 2020 20:36:27 +0800 Subject: [PATCH] For dynamic shape complie. --- ge/graph/build/graph_builder.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ge/graph/build/graph_builder.cc b/ge/graph/build/graph_builder.cc index c7a18350..76eea96a 100644 --- a/ge/graph/build/graph_builder.cc +++ b/ge/graph/build/graph_builder.cc @@ -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);