-
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
[TVM][LANG] Add eager simplification for operations with FloatImm #2615
Conversation
Thank you @icemelon9 can you add some test cases as per https://docs.tvm.ai/contribute/code_review.html#ensure-test-coverage |
@tqchen Test case for eager simplification is added. |
Thanks @tqchen for the review. |
…ache#2615) * Add eager simplication for FloatImm * fix * fix lint * Fix gcc warning * fix * Add test case
This PR introduced a performance regression on resnet-18 on Mali GPUs. cc @yzhliu @icemelon9 |
Rough investigation, regression disappears by applying this patch: https://gist.github.com/yzhliu/0d2cdfc9fa92127b81a1298d5bec55a0 |
This may break the arange op because it relies on the simplification on floating expr to calculate the output shape. Maybe the new canonical simplification can handle this instead of eager simplification. Could you test it? |
Add eager simplification for operations of add, sub, mul, div, ceil, floor, round, trunc, and cast with float constants.