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

[StableHLO][NFC] Factor out and clean up pointwise patterns #13018

Merged
merged 1 commit into from
Apr 11, 2023

Conversation

kuhar
Copy link
Member

@kuhar kuhar commented Apr 11, 2023

This is primarily to improve compilation times when iterating on the rest of StableHLO to Linalg conversion patterns. These pointwise patterns create a large number of class template instantiations and take >20s to compile on my machine.

Also clean up the code:

  • Use free cast functions
  • Make the conversion patterns final
  • Prefer static/free helper functions

Issue: #12678

This is primarily to improve compilation times when iterating on the
rest of StableHLO to Linalg conversion patterns. These pointwise
patterns create a large number of class template instantiations and take
a few seconds to compile.

Also clean up the code:
-  Use free cast functions
-  Make the conversion patterns final
-  Prefer static/free helper functions
@kuhar kuhar added the integrations/stablehlo StableHLO (JAX/TensorFlow/etc) import and conversion label Apr 11, 2023
@kuhar kuhar merged commit 368a7a8 into iree-org:main Apr 11, 2023
jpienaar pushed a commit that referenced this pull request May 1, 2023
This is primarily to improve compilation times when iterating on the
rest of StableHLO to Linalg conversion patterns. These pointwise
patterns create a large number of class template instantiations and take
>20s to compile on my machine.

Also clean up the code:
-  Use free cast functions
-  Make the conversion patterns final
-  Prefer static/free helper functions

Issue: #12678
NatashaKnk pushed a commit to NatashaKnk/iree that referenced this pull request Jul 6, 2023
…#13018)

This is primarily to improve compilation times when iterating on the
rest of StableHLO to Linalg conversion patterns. These pointwise
patterns create a large number of class template instantiations and take
>20s to compile on my machine.

Also clean up the code:
-  Use free cast functions
-  Make the conversion patterns final
-  Prefer static/free helper functions

Issue: iree-org#12678
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
integrations/stablehlo StableHLO (JAX/TensorFlow/etc) import and conversion
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants