[python] Set up python binding for matcher convolution and attention op#22311
Merged
bangtianliu merged 1 commit intoOct 15, 2025
Merged
Conversation
Signed-off-by: Bangtian Liu <liubangtian@gmail.com>
Contributor
Author
|
I've combined MatchConvolutionOp and MatchAttentionOp in the current PR since they follow the same pattern as MatchContractionOp from #22227, and the implementation for each is relatively straightforward. But I am also happy to split this into two separate PRs if preferred. |
kuhar
approved these changes
Oct 15, 2025
weidel-p
pushed a commit
to weidel-p/iree
that referenced
this pull request
Oct 21, 2025
…op (iree-org#22311) This PR extends the preprocessing transform dialect Python bindings by adding `MatchConvolutionOp `and `MatchAttentionOp`, completing the matcher operation API suite started in iree-org#22227. **Motivation** Completes the Python binding API for all preprocessing transform matcher operations, enabling SHARK Tuner to programmatically generate Transform Dialect specs for convolutions and attention operations in addition to contractions. This removes the need for string-based spec generation and improves type safety and maintainability. Signed-off-by: Bangtian Liu <liubangtian@gmail.com> Signed-off-by: Philipp <philipp.weidel@intel.com>
bangtianliu
added a commit
to nod-ai/amd-shark-ai
that referenced
this pull request
Oct 23, 2025
This PR uses python binding to build td specs for convolution op. The python binding is exposed from IREE PR iree-org/iree#22311. Signed-off-by: Bangtian Liu <liubangtian@gmail.com>
bangtianliu
added a commit
to nod-ai/amd-shark-ai
that referenced
this pull request
Oct 27, 2025
This PR uses python binding to build td specs for attention op. The python binding is exposed from IREE PR iree-org/iree#22311. All the temporary solution code and string-based spec generation code have been removed, now SpecBuilder class can be used to build td spec using python bindings. --------- Signed-off-by: Bangtian Liu <liubangtian@gmail.com>
RattataKing
pushed a commit
to RattataKing/amd-shark-ai
that referenced
this pull request
Nov 4, 2025
This PR uses python binding to build td specs for attention op. The python binding is exposed from IREE PR iree-org/iree#22311. All the temporary solution code and string-based spec generation code have been removed, now SpecBuilder class can be used to build td spec using python bindings. --------- Signed-off-by: Bangtian Liu <liubangtian@gmail.com>
pstarkcdpr
pushed a commit
to pstarkcdpr/iree
that referenced
this pull request
Nov 28, 2025
…op (iree-org#22311) This PR extends the preprocessing transform dialect Python bindings by adding `MatchConvolutionOp `and `MatchAttentionOp`, completing the matcher operation API suite started in iree-org#22227. **Motivation** Completes the Python binding API for all preprocessing transform matcher operations, enabling SHARK Tuner to programmatically generate Transform Dialect specs for convolutions and attention operations in addition to contractions. This removes the need for string-based spec generation and improves type safety and maintainability. Signed-off-by: Bangtian Liu <liubangtian@gmail.com>
30 tasks
RattataKing
pushed a commit
to nod-ai/amd-shark-ai
that referenced
this pull request
Mar 2, 2026
This PR uses python binding to build td specs for convolution op. The python binding is exposed from IREE PR iree-org/iree#22311. Signed-off-by: Bangtian Liu <liubangtian@gmail.com>
RattataKing
pushed a commit
to nod-ai/amd-shark-ai
that referenced
this pull request
Mar 2, 2026
This PR uses python binding to build td specs for attention op. The python binding is exposed from IREE PR iree-org/iree#22311. All the temporary solution code and string-based spec generation code have been removed, now SpecBuilder class can be used to build td spec using python bindings. --------- Signed-off-by: Bangtian Liu <liubangtian@gmail.com>
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
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.
This PR extends the preprocessing transform dialect Python bindings by adding
MatchConvolutionOpandMatchAttentionOp, completing the matcher operation API suite started in #22227.Motivation
Completes the Python binding API for all preprocessing transform matcher operations, enabling SHARK Tuner to programmatically generate Transform Dialect specs for convolutions and attention operations in addition to contractions. This removes the need for string-based spec generation and improves type safety and maintainability.