Browse Source

Zero copy nodes are not allocated memory in the known subgraph

pull/1657/head
wqtshg 4 years ago
parent
commit
8ba6f74e98
2 changed files with 3 additions and 1 deletions
  1. +1
    -1
      ge/graph/load/model_manager/davinci_model.cc
  2. +2
    -0
      ge/hybrid/node_executor/compiledsubgraph/known_node_executor.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); 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); ret = ge::AttrUtils::GetInt(ge_model_, ATTR_MODEL_ZERO_COPY_MEMORY_SIZE, value);
runtime_param_.zero_copy_size = ret ? (uint64_t)value : 0;
runtime_param_.zero_copy_size = ret ? static_cast<size_t>value : 0;


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


+ 2
- 0
ge/hybrid/node_executor/compiledsubgraph/known_node_executor.cc View File

@@ -103,6 +103,8 @@ Status KnownNodeTask::Init(TaskContext &context) {
void *buffer = nullptr; void *buffer = nullptr;
size_t total_mem_size = davinci_model_->TotalMemSize(); size_t total_mem_size = davinci_model_->TotalMemSize();
size_t total_zero_copy_size = davinci_model_->TotalZeroCopySize(); size_t total_zero_copy_size = davinci_model_->TotalZeroCopySize();
GELOGI("####KnownNodeTask::Init total mem size is %lu, total zero size is %lu.",
total_mem_size, total_zero_copy_size);
if (total_mem_size != 0 && total_mem_size > total_zero_copy_size) { if (total_mem_size != 0 && total_mem_size > total_zero_copy_size) {
RECORD_EXECUTION_EVENT(context.GetExecutionContext(), context.GetNodeName(), RECORD_EXECUTION_EVENT(context.GetExecutionContext(), context.GetNodeName(),
"[KnownNodeTask_AllocateWorkspace] Start"); "[KnownNodeTask_AllocateWorkspace] Start");


Loading…
Cancel
Save