Browse Source

Zero copy nodes are not allocated memory in the known subgraph

pull/1659/head
wqtshg 4 years ago
parent
commit
f186038600
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

@@ -489,7 +489,7 @@ void DavinciModel::InitRuntimeParams() {
runtime_param_.memory_infos[kSessionScopeMemory | RT_MEMORY_HBM] = std::move(session_scope_mem_info);

ret = ge::AttrUtils::GetInt(ge_model_, ATTR_MODEL_ZERO_COPY_MEMORY_SIZE, value);
runtime_param_.zero_copy_size = ret ? static_cast<size_t>value : 0;
runtime_param_.zero_copy_size = static_cast<size_t>(ret ? value : 0);

GELOGI("InitRuntimeParams(), %s.", runtime_param_.ToString().c_str());
}


Loading…
Cancel
Save