@zhiics can you replace [this block](https://github.com/dmlc/tvm/blob/master/src/relay/pass/fuse_ops.cc#L822-L848) with this
``` Expr VisitExpr_(const TupleNode* tuple) { auto* ret_group = gmap_.at(tuple)->FindRoot(); if (ret_group == gmap_.at(tuple)) { return ExprMutator::VisitExpr_(tuple); } Array<Expr> new_fields = GetNewArguments(tuple->fields, ret_group); // This tuple is an intermediate node in the group return TupleNode::make(new_fields); } ``` and try your PR? -- You are receiving this because you are subscribed to this thread. Reply to this email directly or view it on GitHub: https://github.com/dmlc/tvm/issues/3039#issuecomment-484367143