-
Notifications
You must be signed in to change notification settings - Fork 3.7k
[Relay]Disable InferType if it was done and no changes after previous pass #17585
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
Conversation
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.
LGTM! Thanks!
e522684
to
bd02406
Compare
@tvm-bot rerun |
a minor note, we are moving towards phasing out relay and use relax |
@tvm-bot rerun |
It is true, and we must switch to relax and we are moving toward it. At the same time legacy relay flow sometimes has deeper and wider support. Do we know if relax flow covered all frontend capabilities which were covered by relay? Do we know which backed switched and have same support from performance point of view? |
This optimizatin allows to speedup PatternRewriter transformations by reusing of preious type inferred expression instead of perform InferType multiple times
bd02406
to
01ccffc
Compare
As of now relax flow covers most of the onnx flows and rightfully the development focuses on the latest use-cases. So there might be delta on some legacy use-cases. However, as we move toward relax, hopefully these deltas can be closed and improved |
… pass (apache#17585) Disable InferType if it was done and no changes after previous pass This optimizatin allows to speedup PatternRewriter transformations by reusing of preious type inferred expression instead of perform InferType multiple times
This optimization allows to speedup PatternRewriter transformations by reusing of previous type inferred expression instead of perform InferType multiple times
The test_pass_simplify_expr.py tests on my machine started to pass for 40 seconds vs 116 seconds main branch state