search

joelberkeley / spidr

Accelerated machine learning with dependent types
Apache License 2.0
73 stars 4 forks source link

Build StableHLO programs #365

Open joelberkeley opened 1 year ago

joelberkeley commented 1 year ago

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 commented 4 months ago

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