Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Revert "fix param deletion bug, caused by non-refnode count err." 

This reverts commit 79179f128e.
This commit is contained in:
liangzelang 2021-06-15 10:16:20 +08:00
parent 43c61ff4f0
commit d7bc5313f1
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/ccsrc/backend/session/ascend_auto_monad.cc
View File

@ -1727,13 +1727,9 @@ class ExecuteOrderGenerator {
(void)refed_parameters.insert(validate_ref_parameter(std::get<1>(iter->second)));
}
}
// Search all refnodes for parameter write assigns.
// Search all nodes for parameter write assigns.
for (auto &item : search_list) {
auto &node = item.first;
if (ref_multimap.find(node) == ref_multimap.end()) {
// if node is not refnode which cannot write param, skip it.
continue;
}
for (auto &in : node->inputs()) {
auto visit_node = AnfAlgo::VisitKernelWithReturnType(in, 0).first;
visit_node = validate_ref_parameter(visit_node);