[PASS] Support for partition loops with thread_axis #81
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
tqchen
requested changes
Apr 7, 2017
| IRVisitor::Visit_(op); | ||
| relax_map_.erase(var); | ||
| hint_map_.erase(var); | ||
| } |
Member
There was a problem hiding this comment.
We need to check if op->type_key == ir::attr::thread_extent. Also thread_axis->dom may not be available, use op->value for the extent of the thread(thread always starts from 0).
Member
There was a problem hiding this comment.
We also need to add default condition here to run recursive visit. Otherwise this will results in a BUG that stops visit at AttrStmt that does not match the condition.
tqchen
approved these changes
Apr 10, 2017
tqchen
added a commit
to tqchen/tvm
that referenced
this pull request
May 26, 2018
tqchen
added a commit
that referenced
this pull request
May 29, 2018
tqchen
added a commit
to tqchen/tvm
that referenced
this pull request
Jul 6, 2018
sergei-mironov
pushed a commit
to sergei-mironov/tvm
that referenced
this pull request
Aug 8, 2018
icemelon
pushed a commit
to icemelon/tvm
that referenced
this pull request
Apr 14, 2020
vinx13
pushed a commit
to vinx13/tvm
that referenced
this pull request
Mar 9, 2022
cyx-6
added a commit
to cyx-6/tvm
that referenced
this pull request
Aug 6, 2022
csullivan
added a commit
to csullivan/tvm
that referenced
this pull request
Apr 27, 2024
MasterJH5574
pushed a commit
to MasterJH5574/tvm
that referenced
this pull request
Aug 17, 2025
* dev * dev * dev: remove barrier op
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
2 participants
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
No description provided.