Fix dymamic top cell clear bug

Signed-off-by: zjun <zhangjun0@huawei.com>
4 years ago · 8216e543a6
parent a855cb2d24
commit 8216e543a6
1 changed files with 1 additions and 1 deletions
--- a/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc
+++ b/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc
@ -2537,7 +2537,7 @@ void PynativeExecutor::GradNetInner(const GradOperationPtr &grad, const py::obje
 }

 void PynativeExecutor::ClearDynamicTopRes(const std::string &cell_id, const FuncGraphPtr &df_builder) {
-  if (IsTopestGraph(cell_id)) {
+  if (df_builder == nullptr && IsTopestGraph(cell_id)) {
    op_index_map_.clear();
  }
  // Delete unused top cell resource