提交 31301801 编写于 作者: L laiyongqiang

fix node check bug in convert_tuple_output_to_maketuple pass

上级 2cdb6262
......@@ -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;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册