Skip to content
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

Skipping generic from root op when it computes slice indices #19767

Open
wants to merge 6 commits into
base: main
Choose a base branch
from

Conversation

jerryyin
Copy link
Member

@jerryyin jerryyin commented Jan 22, 2025

This PR prevents the linalg.generic to become root op when it is used to compute slice indices. This PR fix #19746. @IanWood1 has help confirmed this will not cause performance degradation compared with another alternative: having scatter op take precedence over generics.

I'd like to thank @qedawkins for providing mentorship to this PR.

Copy link
Contributor

@qedawkins qedawkins left a comment

Choose a reason for hiding this comment

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

Thanks, just one comment on how to construct the set of ops to skip.

@jerryyin jerryyin force-pushed the users/zyin/skip-generic-result-scatter-indices branch from 77696dc to 23dc093 Compare January 22, 2025 14:59
Copy link
Member

@kuhar kuhar left a comment

Choose a reason for hiding this comment

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

Just some drive-by coding style nits

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.

[Codegen] Failure compiling scatter with fused indices
5 participants