Browse Source

!584 fix node check bug in convert_tuple_output_to_maketuple pass

Merge pull request !584 from laiyongqiang/pass_fix
tags/v0.2.0-alpha
mindspore-ci-bot Gitee 5 years ago
parent
commit
3ef499e1ef
1 changed files with 2 additions and 3 deletions
  1. +2
    -3
      mindspore/ccsrc/pre_activate/pass/convert_tuple_output_to_maketuple.cc

+ 2
- 3
mindspore/ccsrc/pre_activate/pass/convert_tuple_output_to_maketuple.cc View File

@@ -68,9 +68,8 @@ const AnfNodePtr ConvertTupleOutputToMaketuple::Process(const FuncGraphPtr &func
if (AnfAlgo::GetCNodeName(cnode) == prim::kPrimTupleGetItem->name()) {
return nullptr;
}
if (std::any_of(cnode->inputs().begin() + 1, cnode->inputs().end(), [](const AnfNodePtr &node) {
return AnfAlgo::IsTupleOutput(node) && AnfAlgo::GetCNodeName(node) != prim::kPrimMakeTuple->name();
})) {
if (std::any_of(cnode->inputs().begin() + 1, cnode->inputs().end(),
[](const AnfNodePtr &node) { return AnfAlgo::IsRealKernel(node) && AnfAlgo::IsTupleOutput(node); })) {
return ConvertTupleInputToMakeTuple(func_graph, cnode);
}
return nullptr;


Loading…
Cancel
Save