release resources debugger fix

This commit is contained in:
Islam Amin 2021-04-22 13:16:14 -04:00
parent 9daed0bea5
commit 2b20166d23
2 changed files with 5 additions and 0 deletions

View File

@ -288,6 +288,8 @@ void Debugger::Reset() {
last_overflow_bin_ = 0; last_overflow_bin_ = 0;
overflow_bin_path_.clear(); overflow_bin_path_.clear();
stream_task_to_opname_.clear(); stream_task_to_opname_.clear();
graph_proto_list_.clear();
graph_ptr_list_.clear();
} }
void Debugger::PreExecute(const KernelGraphPtr &graph_ptr, uint32_t graph_sum) { void Debugger::PreExecute(const KernelGraphPtr &graph_ptr, uint32_t graph_sum) {

View File

@ -1226,6 +1226,9 @@ void ClearResAtexit() {
device::KernelRuntimeManager::Instance().ClearRuntimeResource(); device::KernelRuntimeManager::Instance().ClearRuntimeResource();
ReleaseGeTsd(); ReleaseGeTsd();
parse::python_adapter::ResetPythonScope(); parse::python_adapter::ResetPythonScope();
#ifdef ENABLE_DEBUGGER
Debugger::GetInstance()->Reset();
#endif
} }
} // namespace pipeline } // namespace pipeline
} // namespace mindspore } // namespace mindspore