Fix pynative_grad coredump bug

Signed-off-by: zjun <zhangjun0@huawei.com>
4 years ago · b4b4310c28
parent 71165dde9a
commit b4b4310c28
1 changed files with 2 additions and 1 deletions
--- a/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc
+++ b/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc
@ -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;
  }