mirror of
https://github.com/ROCm/jax.git
synced 2025-04-17 12:26:07 +00:00

We previously registered the pass in the :_mosaic_gpu_ext which didn't work because the extension has its own pass registry. The fix instead is to move the registration to :register_jax_dialects in jaxlib. PiperOrigin-RevId: 719280601