Skip to content

feat(linalg): add a way to pass controlFn to foldIntoPackUnpackPatterns #22

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

Conversation

egebeysel
Copy link

@egebeysel egebeysel commented Jun 10, 2025

This is step 1/3 of fixing the large vector sizes found compilation failure (roo-389). This PR adds a mechanism, so that downstream consumers can pass in control functions for the application of these patterns. This change shouldn't affect any consumers of this method that do not specify a controlFn.

In IREE, we (will) use it to control preventing folding patterns that would inhibit fusion. See IREE issue llvm#20896 for more details.

Copy link
Author

This stack of pull requests is managed by Graphite. Learn more about stacking.

@egebeysel egebeysel requested a review from a team June 10, 2025 12:14
@egebeysel egebeysel marked this pull request as ready for review June 10, 2025 12:14
@egebeysel egebeysel force-pushed the 06-10-feat_linalg_add_a_way_to_pass_controlfn_to_foldintopackunpackpatterns_ branch from cf69d3e to e1447ad Compare June 10, 2025 12:17
Copy link

@AGindinson AGindinson left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM overall

`foldIntoPackUnpackPatterns`
@egebeysel egebeysel force-pushed the 06-10-feat_linalg_add_a_way_to_pass_controlfn_to_foldintopackunpackpatterns_ branch from e1447ad to 16dc684 Compare June 10, 2025 12:26
@egebeysel egebeysel merged commit de57bbf into integrate-llvm-project-20250508 Jun 11, 2025
6 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants