Browse Source

Variable memory reuse

tags/v1.3.0
TangQunzhang 3 years ago
parent
commit
6f35357236
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      ge/graph/manager/graph_var_manager.cc
  2. +1
    -1
      metadef

+ 1
- 1
ge/graph/manager/graph_var_manager.cc View File

@@ -556,7 +556,7 @@ ge::Status VarManager::AssignVarMem(const std::string &var_name, const ge::GeTen
// reuse old format variable memory // reuse old format variable memory
if (result == SUCCESS) { if (result == SUCCESS) {
result = var_resource_->GetVarAddr( result = var_resource_->GetVarAddr(
var_name, tensor_desc, reinterpret_cast<uint8_t **>(reinterpret_cast<uintptr_t>(&mem_offset)), memory_type);
var_name, cur_tensor_desc, reinterpret_cast<uint8_t **>(reinterpret_cast<uintptr_t>(&mem_offset)), memory_type);
if (result == SUCCESS) { if (result == SUCCESS) {
result = TensorUtils::GetSize(cur_tensor_desc, cur_tensor_desc_size); result = TensorUtils::GetSize(cur_tensor_desc, cur_tensor_desc_size);
GELOGD("tensor_desc_size is %ld, cur_tensor_desc_size is %ld, memoffset is %zu", tensor_desc_size, GELOGD("tensor_desc_size is %ld, cur_tensor_desc_size is %ld, memoffset is %zu", tensor_desc_size,


+ 1
- 1
metadef

@@ -1 +1 @@
Subproject commit 22ab76ecd8461f679606374be4b3b6b6f7cad321
Subproject commit 1c41e02f73b6e8f95369e052ee4de285145fb34f

Loading…
Cancel
Save