-
Notifications
You must be signed in to change notification settings - Fork 3.8k
[TIR] Fix handling of int64 extent in blockize and tensorize #13069
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
|
Thanks for contributing to TVM! Please refer to the contributing guidelines https://tvm.apache.org/docs/contribute/ for useful information and tips. Please request code reviews from Reviewers by @-ing them in a comment.
Generated by tvm-bot |
junrushao
left a comment
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
|
Also included fix for tensorize in this PR. |
2de3715 to
3062a0b
Compare
22b2120 to
6149cb8
Compare
6149cb8 to
e893372
Compare
…13069) * [TIR] Fix handling of int64 extent in blockize * Fix handling of int64 extent in tensorize * Update layout_transformation.cc
…13069) * [TIR] Fix handling of int64 extent in blockize * Fix handling of int64 extent in tensorize * Update layout_transformation.cc
This fixed arithmetic analysis to ensure the result of subspace division has the same dtype as the original expression.
cc @junrushao @spectrometerHBH @Hzfengsy