-
Notifications
You must be signed in to change notification settings - Fork 3.4k
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
[TIR][Schedule] Enhance compute-inline
for fusion
#15142
[TIR][Schedule] Enhance compute-inline
for fusion
#15142
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 |
0bc3eed
to
ae42a89
Compare
I just realized that |
tests/python/unittest/test_meta_schedule_schedule_rule_mlt_tc.py
Outdated
Show resolved
Hide resolved
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.
Please address the sm
version comment, and we can get it in
This PR enhances `compute-inline` and `reverse-compute-inline` to handle more complicated fusion patterns where the RHS of the equation could be generic expressions rather than strict buffer indexing. This could be used in multiple cases in LLM inference.
ae42a89
to
dd07026
Compare
This PR enhances
compute-inline
andreverse-compute-inline
to handle more complicated fusion patterns where the RHS of the equation could be generic expressions rather than strict buffer indexing. This could be used in multiple cases in LLM inference.