jax
ff4e0b12 - Rearrange the LAPACK handler definitions in jaxlib to avoid duplicate handler errors.

Commit
1 year ago
Rearrange the LAPACK handler definitions in jaxlib to avoid duplicate handler errors. When linking the jaxlib `cpu_kernels` target and importing JAX, we currently silently fail to instantiate the CPU backend. This refactor means that we only ever define one version of the handlers. PiperOrigin-RevId: 657186057
Author
dfm dfm
Committer
Parents
Loading