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

!3172 Fix bugs of the MaybeDropNodes function 

Merge pull request !3172 from Kang/opt
This commit is contained in:
mindspore-ci-bot 2020-07-20 11:28:22 +08:00 committed by Gitee
parent be2cfa9ed6 61dc91a557
commit ef7cfaaf2e
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
mindspore/core/ir/manager.cc
View File

@ -387,12 +387,17 @@ FuncGraphSetPtr FuncGraphManager::MaybeDropNodes(const std::vector<AnfNodePtr> &
continue;
}
AnfNodeIndexSet &users = node_users_[node];
std::vector<AnfNodePtr> parameters;
if (!users.empty() ||
(node->isa<Parameter>() && parameters.end() != std::find(parameters.begin(), parameters.end(), node))) {
if (!users.empty()) {
continue;
}
if (node->isa<Parameter>() && node->func_graph() != nullptr) {
auto &parameters = node->func_graph()->parameters();
if (std::find(parameters.begin(), parameters.end(), node) != parameters.end()) {
continue;
}
}
if (IsValueNode<FuncGraph>(node)) {
auto fg = GetValueNode<FuncGraphPtr>(node);
func_graphs_to_check->add(fg);