Browse Source

!472 bugfix for mem_base logic offset

From: @ni100die
Reviewed-by: @tangqunzhang,@ji_chen,@wqtshg
Signed-off-by: @ji_chen,@wqtshg
tags/v1.1.0
mindspore-ci-bot Gitee 4 years ago
parent
commit
989dce729a
1 changed files with 3 additions and 1 deletions
  1. +3
    -1
      ge/graph/build/graph_builder.cc

+ 3
- 1
ge/graph/build/graph_builder.cc View File

@@ -377,7 +377,9 @@ Status GraphBuilder::GetTaskInfo(const ge::ModelBuilder &builder, const ModelPtr
}

auto var_manager = VarManager::Instance(session_id);
auto *get_mem_base = reinterpret_cast<uint8_t *>(reinterpret_cast<uintptr_t>(var_manager->GetVarMemMaxSize()));
// since var_mem_logic_base_ = graph_mem_max_size_ + kGraphMemoryBuffer in graph_var_manager.cc,
// get_mem_base should not bigger than kGraphMemoryBuffer
auto *get_mem_base = reinterpret_cast<uint8_t *>(reinterpret_cast<uintptr_t>(kGraphMemoryBuffer>>1));
uint8_t *get_weight_mem_base = get_mem_base;
if (weight_size > 0) {
get_weight_mem_base = get_mem_base + memory_size + p2p_memory_size;


Loading…
Cancel
Save