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.

Sign up for GitHub

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

Jump to bottom

Build StableHLO programs #365

Open
joelberkeley opened this issue Jul 17, 2023 · 2 comments · May be fixed by #425
Open

Build StableHLO programs #365

joelberkeley opened this issue Jul 17, 2023 · 2 comments · May be fixed by #425
Labels
enhancement New feature or request

Comments

@joelberkeley
Copy link
Owner

joelberkeley commented Jul 17, 2023 •
edited
Loading

Build StableHLO programs rather than XLA programs, then we can provide support for anything that accepts StableHLO, including XLA and IREE removing our dependency on XLA. This may require rewritng much of the XLA glue code, but since the XLA and StableHLO interfaces look very similar, a lot might carry over
@joelberkeley joelberkeley added the enhancement New feature or request label Jan 27, 2024
@joelberkeley
Copy link
Owner Author

I don't see much benefit to this atm since we can already convert XlaComputation to HLO and pass that to PJRT, thus running on IREE. At least that's what I've heard, we haven't done it yet

@joelberkeley joelberkeley pinned this issue Oct 7, 2024
@joelberkeley joelberkeley linked a pull request Oct 15, 2024 that will close this issue
migrate to StableHLO #425
Draft
@joelberkeley
Copy link
Owner Author

joelberkeley commented Dec 1, 2024 •
edited
Loading

We need StableHLO for some plugins incl. Apple Metal. We can, however, do this in two phases: 1) build XLA HLO then convert it to StableHLO. 2) build StableHLO directly. 1 is likely enough to implement plugins, autodiff and vmap

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
enhancement New feature or request
Projects
None yet
Milestone
No milestone
Development

Successfully merging a pull request may close this issue.

migrate to StableHLO joelberkeley/spidr
1 participant
@joelberkeley