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

Implement ConditionallySpeculatable for remaining dynamic ops #2242

Merged
merged 3 commits into from
May 1, 2024

Conversation

mlevesquedion
Copy link
Contributor

@mlevesquedion mlevesquedion commented Apr 19, 2024

Included ops:

  • DynamicPad
  • RealDynamicSlice
  • DynamicConv
  • DynamicGather

I refactored the logic to check speculatability for shaped ops to enable
reuse and allow ops with more than one shape-related operand to be
checked.

This should be the last change adding new speculatability
implementations. All the other ops are either done, pure, or
deprecated. I will confirm this shortly by reviewing the entire opset
and making sure everything is covered (I have been tracking progress in
a personal document).

@mlevesquedion
Copy link
Contributor Author

Converting to draft while I revisit the shared implementation for ops that have shape operands.

@mlevesquedion mlevesquedion force-pushed the dyn-cs branch 3 times, most recently from 40d2db2 to 5c4a2de Compare April 26, 2024 20:47
@mlevesquedion mlevesquedion marked this pull request as ready for review April 26, 2024 20:48
@mlevesquedion
Copy link
Contributor Author

This is once again ready for review.

Included ops:
- DynamicPad
- RealDynamicSlice
- DynamicConv
- DynamicGather

I refactored the logic to check speculatability for shaped ops to enable
reuse and allow ops with more than one shape-related operand to be
checked.

This should be the last change adding new speculatability
implementations. All the other ops are either done, pure, or
deprecated. I will confirm this shortly by reviewing the entire opset
and making sure everything is covered (I have been tracking progress in
a personal document).
@mlevesquedion
Copy link
Contributor Author

There were some merge conflicts. Those are fixed now.

stablehlo/dialect/Base.h Outdated Show resolved Hide resolved
stablehlo/dialect/Base.td Outdated Show resolved Hide resolved
@mlevesquedion mlevesquedion merged commit b725ed9 into openxla:main May 1, 2024
10 checks passed
@mlevesquedion mlevesquedion deleted the dyn-cs branch May 1, 2024 21:40
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants