mirror of
https://github.com/ROCm/jax.git
synced 2025-04-16 03:46:06 +00:00

I re-used the same trick we do for the TPU dialect. Specifically, _triton_ext no longer depends on :triton_dialect_capi. Instead * we include Triton dialect C bindings into :jaxlib_mlir_capi_objects * and _triton_ext depends on :jaxlib_mlir_capi_objects and a header-only cc_library providing Triton dialect C bindings This is a fork of #19680 with a few internal-only fixes. PiperOrigin-RevId: 604929377