|
|
@@ -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; |
|
|
|