Browse Source

!8964 Fix pynative grad core dump bug

From: @zjun3021
Reviewed-by: @kisnwang,@chujinjin
Signed-off-by: @chujinjin
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
cab7d86ec1
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      mindspore/ccsrc/pipeline/pynative/pynative_execute.cc

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

@@ -2022,8 +2022,9 @@ void PynativeExecutor::Clear(const std::string &flag) {
ConfigManager::GetInstance().ResetIterNum();
if (top_graph_cells_.find(flag) != top_graph_cells_.end()) {
op_forward_map_.clear();
Clean();
}
Clean();
node_abs_map_.clear();
return;
}



Loading…
Cancel
Save