Browse Source

!1874 bugfix for InitNetOutput

From: @yangyongqiang5033
Reviewed-by: @zhangxiaokun9,@wqtshg
Signed-off-by: @wqtshg
tags/v1.5.1
mindspore-ci-bot Gitee 3 years ago
parent
commit
c6f2fe1648
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/graph/load/model_manager/davinci_model.cc

+ 1
- 1
ge/graph/load/model_manager/davinci_model.cc View File

@@ -1156,7 +1156,6 @@ Status DavinciModel::InitNetOutput(const ComputeGraphPtr &graph, const NodePtr &
}

size_t num = output_data_info_.size();
bool fusion_flag = false;

size_t input_count = input_size_list.size();
is_getnext_sink_dynamic_ = false;
@@ -1166,6 +1165,7 @@ Status DavinciModel::InitNetOutput(const ComputeGraphPtr &graph, const NodePtr &
}
for (size_t idx = 0; idx < input_count; ++idx) {
ZeroCopyOffset zero_copy_offset;
bool fusion_flag = false;
Status ret = zero_copy_offset.InitOutputDataInfo(input_size_list, virtual_addr_list, op_desc, idx, fusion_flag);
GE_IF_BOOL_EXEC(ret != SUCCESS,
GELOGE(PARAM_INVALID, "[Init][DataInfo] of input_info %s failed.", op_desc->GetName().c_str());


Loading…
Cancel
Save