@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

Reply via email to