xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!13487 Fix top cell bug 

From: @zjun3021
Reviewed-by: @chujinjin,@zhoufeng54
Signed-off-by: @chujinjin
This commit is contained in:
mindspore-ci-bot 2021-03-17 20:06:18 +08:00 committed by Gitee
parent b20e964760 adf47d3d92
commit 11a20a4ece
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/ccsrc/pipeline/pynative/pynative_execute.cc
View File

@ -1647,7 +1647,7 @@ bool GradExecutor::IsTopGraph(const std::string &cell_id) {
bool GradExecutor::IsTopestGraph(const std::string &cell_id) {
return std::any_of(top_cell_list_.begin(), top_cell_list_.end(), [&cell_id](const TopCellInfoPtr &value) {
return value->cell_id() == cell_id && value->is_topest();
return (value->cell_id() == cell_id || cell_id.find(value->cell_id()) != std::string::npos) && value->is_topest();
});
}
@ -1764,7 +1764,7 @@ void GradExecutor::ClearResidualRes(const std::string &cell_id) {
return;
}
auto is_real_dynamic = pre_top_cell_->is_real_dynamic();
if (is_real_dynamic) {
if (is_real_dynamic && cell_id == pre_top_cell_->cell_id()) {
// Clear previous step resource
auto resource = GetResource(cell_id);
if (resource != nullptr && resource->results().find(pipeline::kBackend) != resource->results().end()) {