masahi opened a new pull request, #14971:
URL: https://github.com/apache/tvm/pull/14971
Currently, `CombineParallelMatmul` eagerly combines all parallel branches it
finds. Sometimes, it is desirable to ignore certain patterns based on, for
example, the number of matmuls.
This PR adds a user-defined "check function" to `CombineParallelMatmul` that
is used like this:
```
Var inp;
Array<Var> rhs, bias;
Map<Var, Expr> bindings; // Passed from DF rewriter
if (!check(inp, rhs, bias, bindings)) {
continue;
}
```
It also modifies the signature of the DF binding rewrite function to
additionally pass the current bindings (`Map<Var, Expr>`). This can be useful
for querying the attributes of the matched `CallNode` expressions.
--
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]
