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

Mark GEP as inbounds (memref) and unlock derivative of indexing/slicing #552

Merged
merged 10 commits into from
Feb 29, 2024

Conversation

rmoyard
Copy link
Contributor

@rmoyard rmoyard commented Feb 26, 2024

Description of the Change:

We mark GEPOp inbounds when they do not have a null ptr as defining operation, this allows Enzyme to do a better activity analysis and it unlocks the derivative of indexing arrays.

@rmoyard
Copy link
Contributor Author

rmoyard commented Feb 26, 2024

[sc-57426]

@dime10 dime10 added this to the v0.5.0 milestone Feb 26, 2024
@rmoyard rmoyard changed the title Mark GEP as inbounds (memref) Mark GEP as inbounds (memref) and unlock derivative of indexing Feb 29, 2024
@rmoyard rmoyard marked this pull request as ready for review February 29, 2024 00:40
Copy link
Contributor

Hello. You may have forgotten to update the changelog!
Please edit doc/changelog.md on your branch with:

  • A one-to-two sentence description of the change. You may include a small working example for new features.
  • A link back to this PR.
  • Your name (or GitHub username) in the contributors section.

@rmoyard
Copy link
Contributor Author

rmoyard commented Feb 29, 2024

[sc-55109]

@rmoyard rmoyard changed the title Mark GEP as inbounds (memref) and unlock derivative of indexing Mark GEP as inbounds (memref) and unlock derivative of indexing/slicing Feb 29, 2024
Copy link
Collaborator

@dime10 dime10 left a comment

Choose a reason for hiding this comment

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

Great 🥳

mlir/test/Catalyst/GEPInbounds.mlir Outdated Show resolved Hide resolved
mlir/include/Catalyst/Transforms/Passes.td Outdated Show resolved Hide resolved
@dime10 dime10 merged commit 1692a4d into main Feb 29, 2024
34 checks passed
@dime10 dime10 deleted the gep_inbounds branch February 29, 2024 05:53
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