-
Notifications
You must be signed in to change notification settings - Fork 3.5k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[Relay] Fix Fuse #3035
[Relay] Fix Fuse #3035
Conversation
src/relay/pass/fuse_ops.cc
Outdated
@@ -865,9 +865,17 @@ class FuseMutator : private ExprMutator { | |||
} | |||
|
|||
Expr MakeNewFunction(GraphPartitioner::Group* group, Type ret_type, Expr body) { | |||
// If the function has no call, it is not a primitive function. | |||
struct HasCallVisitor : ExprVisitor { | |||
bool HasCall = false; |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
should be snake case
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Great thanks. Just a minor nit
I believe that we will need a bit more discussion about what is happening here. In particular, the usage of fusor and primitive functions. There might be a better fix to this problem e.g. (not fuse these ops) |
Move discussion to #3039 |
* save * fix * Update fuse_ops.cc
* save * fix * Update fuse_ops.cc
Thanks for contributing to TVM! Please refer to guideline https://docs.tvm.ai/contribute/ for useful information and tips. After the pull request is submitted, please request code reviews from Reviewers.