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

!18064 Fix param elimination bug, caused by non-refnode count err 

Merge pull request !18064 from liangzelang/gpt3_assign
This commit is contained in:
i-robot 2021-06-10 20:01:46 +08:00 committed by Gitee
parent 3e6a869976 79179f128e
commit a4af96e498
1 changed files with 5 additions and 1 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

@ -1723,9 +1723,13 @@ class ExecuteOrderGenerator {
(void)refed_parameters.insert(validate_ref_parameter(std::get<1>(iter->second))); (void)refed_parameters.insert(validate_ref_parameter(std::get<1>(iter->second)));
} }
} }
// Search all nodes for parameter write assigns. // Search all refnodes for parameter write assigns.
for (auto &item : search_list) { for (auto &item : search_list) {
auto &node = item.first; 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()) { for (auto &in : node->inputs()) {
auto visit_node = AnfAlgo::VisitKernelWithReturnType(in, 0).first; auto visit_node = AnfAlgo::VisitKernelWithReturnType(in, 0).first;
visit_node = validate_ref_parameter(visit_node); visit_node = validate_ref_parameter(visit_node);