From 299ed02735dbcbd2f69d028d96455a82c6818e5d Mon Sep 17 00:00:00 2001 From: "gengchao4@huawei.com" Date: Tue, 1 Dec 2020 12:46:46 +0800 Subject: [PATCH] bugfix for mem_base logic offset --- ge/graph/build/graph_builder.cc | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/ge/graph/build/graph_builder.cc b/ge/graph/build/graph_builder.cc index e52f84ed..79e46f50 100644 --- a/ge/graph/build/graph_builder.cc +++ b/ge/graph/build/graph_builder.cc @@ -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(reinterpret_cast(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(reinterpret_cast(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;