Browse Source

fix memory leak in pynative mode

tags/v1.1.0
chujinjin 5 years ago
parent
commit
78e44b1c7b
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      mindspore/ccsrc/pipeline/pynative/pynative_execute.cc

+ 1
- 0
mindspore/ccsrc/pipeline/pynative/pynative_execute.cc View File

@@ -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<FuncGraphPtr>().swap(graph_context_);
ConfigManager::GetInstance().ResetIterNum();
}


Loading…
Cancel
Save