vinx13 commented on code in PR #14109:
URL: https://github.com/apache/tvm/pull/14109#discussion_r1117453966
##########
src/relax/transform/fuse_ops.cc:
##########
@@ -908,23 +908,27 @@ class PatternBasedPartitioner : ExprVisitor {
using GroupMap = OperatorFusor::GroupMap;
using ExprVisitor::VisitExpr_;
- static GroupMap Run(String pattern_name, DFPattern pattern, Expr expr,
support::Arena* arena) {
- PatternBasedPartitioner part(pattern_name, pattern,
AnalyzeVar2Value(expr));
+ static GroupMap Run(String pattern_name, DFPattern pattern,
runtime::PackedFunc check, Expr expr,
+ support::Arena* arena) {
+ PatternBasedPartitioner part(pattern_name, pattern, check,
AnalyzeVar2Value(expr));
// Initialize each expr to have its own group
PostOrderVisit(
expr, [arena, &part](const Expr& e) { part.group_map_[e.get()] =
arena->make<Group>(); });
part.VisitExpr(expr);
return part.group_map_;
}
- PatternBasedPartitioner(String pattern_name, DFPattern pattern, const
Map<Var, Expr>& bindings)
+ PatternBasedPartitioner(String pattern_name, DFPattern pattern,
runtime::PackedFunc check,
Review Comment:
Updated, there are still a few places using `Array<PackedFunc>` because
`TypedPackedFunc` can't be Array element
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
