jax
3349e7a1 - [Pallas] Allow inferring the backend from the provided compiler params.

Commit

58 days ago

[Pallas] Allow inferring the backend from the provided compiler params. This is a convenience on GPU, where two backends are available. PiperOrigin-RevId: 846147150

References

#34022 - [Pallas] Allow inferring the backend from the provided compiler params.

Author

bchetioui

Committer

Google-ML-Automation

Parents

97c557de

jax 3349e7a1 - [Pallas] Allow inferring the backend from the provided compiler params.

jax
3349e7a1 - [Pallas] Allow inferring the backend from the provided compiler params.