From 78e44b1c7b2fc201e28cf5411e32fa58bcfa35b5 Mon Sep 17 00:00:00 2001 From: chujinjin Date: Mon, 16 Nov 2020 19:44:06 +0800 Subject: [PATCH] fix memory leak in pynative mode --- mindspore/ccsrc/pipeline/pynative/pynative_execute.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc b/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc index ecb14dfbee..b1c84bd86f 100644 --- a/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc +++ b/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc @@ -1985,6 +1985,7 @@ void PynativeExecutor::Clear(const std::string &flag) { graph_info_map_.clear(); op_id_map_.clear(); obj_to_forward_id_.clear(); + node_abs_map_.clear(); std::stack().swap(graph_context_); ConfigManager::GetInstance().ResetIterNum(); }