Browse Source

Zero copy nodes are not allocated memory in the known subgraph

pull/1721/head
wqtshg 4 years ago
parent
commit
ade78930b0
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      ge/hybrid/node_executor/compiledsubgraph/known_node_executor.cc

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

@@ -156,8 +156,8 @@ Status KnownNodeTask::InitDavinciModel(const HybridModel &model, TensorBuffer *w
weight_size = weight_buffer->GetSize();
}
GELOGD("Start to init davinci model, weight size = %zu", weight_size);
GE_CHK_STATUS_RET(DoInitDavinciModel(weight, weight_size), "[Init][Model] Failed to init davinci model.");
GELOGD("[Init][Model] success");
GE_CHK_STATUS_RET(DoInitDavinciModel(weight, weight_size), "[Init][DavinciModel] Failed to init davinci model.");
GELOGD("[Init][DavinciModel] success");
return SUCCESS;
}



Loading…
Cancel
Save