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

[RFC] StableHLO v1.0 Opset Deprecations & Cleanups #2283

Merged
merged 8 commits into from
May 13, 2024

Conversation

GleasonK
Copy link
Member

@GleasonK GleasonK commented May 3, 2024

A proposal to remove redundant operations from StableHLO before long-term compatibility guarantees go into place.

High level summary:

  • Remove CreateTokenOp, TraceOp, BroadcastOp, DotOp, UnaryEinsumOp, RealDynamicSliceOp.
  • Enhance DynamicSliceOp.
  • Move CrossReplicaSumOp to CHLO.
  • Hopefully remove/move to CHLO (need feedback) MapOp, RngOp, EinsumOp, TorchIndexSelectOp, GetTupleElementOp, and tuple type.

OpenXLA Discuss post: https://groups.google.com/a/openxla.org/g/openxla-discuss/c/sBAkvnd2bcA

Related tickets: #2176, #3

@GleasonK GleasonK added the RFC label May 3, 2024
@GleasonK GleasonK self-assigned this May 3, 2024
rfcs/20240503-opset-deprecations.md Outdated Show resolved Hide resolved
rfcs/20240503-opset-deprecations.md Outdated Show resolved Hide resolved
rfcs/20240503-opset-deprecations.md Outdated Show resolved Hide resolved
rfcs/20240503-opset-deprecations.md Outdated Show resolved Hide resolved
rfcs/20240503-opset-deprecations.md Outdated Show resolved Hide resolved
rfcs/20240503-opset-deprecations.md Show resolved Hide resolved
mlevesquedion pushed a commit that referenced this pull request May 9, 2024
…2296)

This pass is intended to be used to test the blast radius of the
proposed deprecation changes in #2283 and help with migration of passes
and other tooling.

Still todo - patterns for the following:
- Einsum pattern
- TorchIndexSelect pattern
- RNG pattern (if possible)
- RealDynamicSliceOp (requires dynamic_slice update first)
abhigunj pushed a commit to abhigunj/stablehlo that referenced this pull request May 9, 2024
…penxla#2296)

This pass is intended to be used to test the blast radius of the
proposed deprecation changes in openxla#2283 and help with migration of passes
and other tooling.

Still todo - patterns for the following:
- Einsum pattern
- TorchIndexSelect pattern
- RNG pattern (if possible)
- RealDynamicSliceOp (requires dynamic_slice update first)
mlevesquedion pushed a commit that referenced this pull request May 9, 2024
Part of #2283. This op doesn't seem to have any uses in frameworks or
compilers. It shouldn't cause any large issues to remove this op all
together, also since this op was never specced, it is exempt from
compatibility guarantees.

In the case that it is used somewhere, I would recommend migration to a
custom_call.
docs/spec.md Outdated Show resolved Hide resolved
docs/spec.md Outdated Show resolved Hide resolved
docs/spec.md Outdated Show resolved Hide resolved
docs/spec.md Show resolved Hide resolved
docs/spec.md Outdated Show resolved Hide resolved
docs/spec.md Outdated Show resolved Hide resolved
@ghpvnist ghpvnist self-requested a review May 13, 2024 19:59
@GleasonK GleasonK merged commit 06bcb0d into openxla:main May 13, 2024
10 checks passed
@GleasonK GleasonK deleted the stablehlo-op-deprecation branch May 14, 2024 03:20
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants