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

[TorchToLinalg] remove extract_slice grid_sample lowering #3483

Merged
merged 3 commits into from
Aug 20, 2024

Conversation

IanWood1
Copy link
Contributor

@IanWood1 IanWood1 commented Jun 20, 2024

Instead of using extract_slice for grid sampler, use affine constants to access the X and Y values in the generic op's region.

@renxida renxida marked this pull request as ready for review June 20, 2024 22:03
@IanWood1 IanWood1 force-pushed the grid_sampler_index_maps branch from 43ce1a0 to 9f43e04 Compare June 20, 2024 23:17
@AmosLewis
Copy link
Collaborator

iree-org/iree#17692

@vivekkhandelwal1
Copy link
Collaborator

Hi @IanWood1, can you please address the comment and update the PR?

@IanWood1
Copy link
Contributor Author

Hi @IanWood1, can you please address the comment and update the PR?

updated!

@zjgarvey
Copy link
Collaborator

@IanWood1 If you don't have merge permissions, let me know and I'll squash and merge this when you are ready.

@IanWood1 IanWood1 merged commit a24114e into llvm:main Aug 20, 2024
3 checks passed
@IanWood1 IanWood1 changed the title [TorchToLinalg] remove extract_slice from lowering of grid_sample [TorchToLinalg] remove extract_slice grid_sample lowering Aug 20, 2024
@IanWood1 IanWood1 deleted the grid_sampler_index_maps branch August 20, 2024 21:23
@IanWood1
Copy link
Contributor Author

@zjgarvey I just got them last week, but thank you. and thanks for the review

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.

4 participants