Browse Source

!3172 Fix bugs of the MaybeDropNodes function

Merge pull request !3172 from Kang/opt
tags/v0.6.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
ef7cfaaf2e
1 changed files with 9 additions and 4 deletions
  1. +9
    -4
      mindspore/core/ir/manager.cc

+ 9
- 4
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);


Loading…
Cancel
Save