-
Notifications
You must be signed in to change notification settings - Fork 0
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
[M2a] Storage align #422
[M2a] Storage align #422
Conversation
aa8bfa0
to
8f431f0
Compare
cc @Hzfengsy please help to shepherd this PR |
python/tvm/tir/schedule/schedule.py
Outdated
|
||
########## Schedule: loop binding/annotation ########## | ||
def storage_align(self, block: BlockRV, buffer_index: int, axis: int, factor: int, |
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.
It's a bit strange to put storage_align()
in the "loop binding/annotation" category, as the primitive doesn't play with any loop. My proposal is to add a new category named "buffer transformation", which can contain primitives like storage_align()
, set_scope()
, etc.
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.
+1
python/tvm/tir/schedule/schedule.py
Outdated
|
||
########## Schedule: loop binding/annotation ########## | ||
def storage_align(self, block: BlockRV, buffer_index: int, axis: int, factor: int, |
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.
+1
327627f
to
2912342
Compare
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 if you can address current comments and check the lint.
Please address the comments and send the PR to mainline. I'd love to do another round of review there |
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. I think we can merge it and send it to mainline.
Co-authored-by: Siyuan Feng <Hzfengsy@sjtu.edu.cn>
Co-authored-by: Ruihang Lai <lairuihangdongdong@qq.com>
Co-authored-by: Ruihang Lai <lairuihangdongdong@qq.com>
Co-authored-by: Ruihang Lai <lairuihangdongdong@qq.com>
6cdd3ab
to
730ecab
Compare
730ecab
to
1bf6f27
Compare
Comments have been addressed, please review again |
Please send this PR to mainline for a second review :-) |
No description provided.