Browse Source

Fix dymamic top cell clear bug

Signed-off-by: zjun <zhangjun0@huawei.com>
tags/v1.2.0-rc1
zjun 5 years ago
parent
commit
8216e543a6
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/pipeline/pynative/pynative_execute.cc

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

@@ -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


Loading…
Cancel
Save