tvm-dev mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From masahi <notificati...@github.com>
Subject Re: [dmlc/tvm] [RFC][DISCUSS] Tuple-related Fusion (#3039)
Date Thu, 18 Apr 2019 06:00:38 GMT
@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
Mime
  • Unnamed multipart/alternative (inline, 7-Bit, 0 bytes)
View raw message