Browse Source

copy opdesc

pull/815/head
lwx897429 5 years ago
parent
commit
514ab3299a
1 changed files with 6 additions and 2 deletions
  1. +6
    -2
      ge/graph/load/new_model_manager/davinci_model.cc

+ 6
- 2
ge/graph/load/new_model_manager/davinci_model.cc View File

@@ -969,7 +969,9 @@ Status DavinciModel::InitDataOp(const ComputeGraphPtr &graph, const NodePtr &nod
}

data_by_index[data_index] = op_desc;
data_op_list_.push_back(op_desc);
auto data_op = AttrUtils::CopyOpDesc(op_desc);
GE_CHECK_NOTNULL(data_op);
data_op_list_.push_back(data_op);
if (known_node_) {
return SUCCESS;
}
@@ -1025,7 +1027,9 @@ Status DavinciModel::OptInputOutputInfo(const map<uint32_t, OpDescPtr> &data_by_

data_op_list_.clear();
for (auto &item : data_by_index) {
data_op_list_.emplace_back(item.second);
auto data_op = AttrUtils::CopyOpDesc(item.second);
GE_CHECK_NOTNULL(data_op);
data_op_list_.emplace_back(data_op);
auto output_addrs = ModelUtils::GetOutputDataAddrs(runtime_param_, item.second);
GELOGD("Data node: %s, output addr size: %zu", item.second->GetName().c_str(), output_addrs.size());
input_addrs_list_.emplace_back(output_addrs);


Loading…
Cancel
Save