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] Move pointwise to linalg conversion to source file #13044

Merged
merged 1 commit into from
Apr 12, 2023

Conversation

kuhar
Copy link
Member

@kuhar kuhar commented Apr 12, 2023

This moves the pointwise StableHLO op to linalg.generic conversion out of the common header and to the matching source file.

In addition, unify the common precondition checks code used by both pointwise patterns and pull it out to a non-template function. This is to reduce the amount of duplicated code. This reduces the compilation time of this source file from 28s to 22s on my machine.

Also clean up the moved pattern.

Issue: #12678

This moves the pointwise StableHLO op to `linalg.generic` conversion out
of the common header and to the matching source file.

In addition, unify the common precondition checks code used by both
pointwise patterns and pull it out to a non-template function. This is
to reduce the amount of duplicated code. This reduces the compilation
time of this source file from 28s to 22s on my machine.

Also clean up the moved pattern.

Issue: iree-org#12678
@kuhar kuhar enabled auto-merge (squash) April 12, 2023 15:48
@kuhar kuhar merged commit 180f595 into iree-org:main Apr 12, 2023
@ScottTodd ScottTodd added the integrations/stablehlo StableHLO (JAX/TensorFlow/etc) import and conversion label Apr 12, 2023
jpienaar pushed a commit that referenced this pull request May 1, 2023
…13044)

This moves the pointwise StableHLO op to `linalg.generic` conversion out
of the common header and to the matching source file.

In addition, unify the common precondition checks code used by both
pointwise patterns and pull it out to a non-template function. This is
to reduce the amount of duplicated code. This reduces the compilation
time of this source file from 28s to 22s on my machine.

Also clean up the moved pattern.

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

This moves the pointwise StableHLO op to `linalg.generic` conversion out
of the common header and to the matching source file.

In addition, unify the common precondition checks code used by both
pointwise patterns and pull it out to a non-template function. This is
to reduce the amount of duplicated code. This reduces the compilation
time of this source file from 28s to 22s on my machine.

Also clean up the moved pattern.

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

3 participants