Browse Source

!13024 Fix bprop dynamic top cell

From: @zjun3021
Reviewed-by: @kisnwang,@chujinjin
Signed-off-by: @chujinjin
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 4 years ago
parent
commit
0c3152fe6b
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