From db782ac92e00e9d703434e9f5077b5efa99c405d Mon Sep 17 00:00:00 2001 From: laiyongqiang Date: Mon, 16 Nov 2020 10:14:55 +0800 Subject: [PATCH] remove context disable somas when variable_memory_max_size is set as somas feature switch --- .../ccsrc/runtime/device/kernel_runtime.cc | 17 +++++++---------- 1 file changed, 7 insertions(+), 10 deletions(-) diff --git a/mindspore/ccsrc/runtime/device/kernel_runtime.cc b/mindspore/ccsrc/runtime/device/kernel_runtime.cc index 5732df7f91..fc69f4f7e6 100644 --- a/mindspore/ccsrc/runtime/device/kernel_runtime.cc +++ b/mindspore/ccsrc/runtime/device/kernel_runtime.cc @@ -684,16 +684,13 @@ void KernelRuntime::AssignDynamicMemory(session::KernelGraph *graph) { if (is_enable_mem_reuse) { MS_LOG(INFO) << "Memory Reuse is enable..."; - auto context = MsContext::GetInstance(); - MS_EXCEPTION_IF_NULL(context); - auto variable_memory_max_size = context->get_param(MS_CTX_VARIABLE_MEMORY_MAX_SIZE); - if (variable_memory_max_size == "0") { - mem_manager_->MallocSomasDynamicMem(graph); - mem_type = kSomasReuseDynamicMem; - } else { - mem_manager_->MallocReusedDynamicMem(graph); - mem_type = kReuseDynamicMem; - } +#ifdef MEM_REUSE_DEBUG + mem_manager_->MallocReusedDynamicMem(graph); + mem_type = kReuseDynamicMem; +#else + mem_manager_->MallocSomasDynamicMem(graph); + mem_type = kSomasReuseDynamicMem; +#endif } else { MS_LOG(INFO) << "Memory Reuse is disable..."; }