[mlir][xegpu] Create a test pass for subgroup distribution. #161592
Merged
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
charithaintc
requested review from
Jianhui-Li,
adam-smnk,
mshahneo,
nbpatel and
silee2
Jianhui-Li
reviewed
Oct 3, 2025
| @@ -0,0 +1,216 @@ | |||
| // RUN: mlir-opt --xevm-attach-target='module=xevm_* chip=pvc' -xegpu-subgroup-distribute \ | |||
There was a problem hiding this comment.
what's reason that the file name having "propogate-layout", the pass seems only having subgroup-distribute.
There was a problem hiding this comment.
good catch. layout prop is already done for these tests.
renamed back to subgroup-distribute.mlir
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.
Current subgroup distribution test employ the entire
xegpu-subgroup-distributepass which include multiple steps like layout propagation, move func body into warp op, and distribute to work items.This makes it harder to isolate the testing for xegpu subgroup distribution logic, because certain corner cases may be not supported yet by other steps mentioned above.
This PR introduces a test pass for subgroup distribution logic and isolate the testing for distribution logic. We plan to add more corner case (that were not possible before) covering non-xegpu ops (like vector) in next PRs.
This PR also include,