mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 11:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jax/extend/mlir
History
Peter Hawkins 78543f7bb8 Add jax.extend.mlir. 
Some users of JAX want to use the MLIR dialects defined in jaxlib. In particular, these need to be used by custom lowering rules. Add a semi-public (jax.extend) API to access these, rather than having them use jax._src.lib.mlir.

PiperOrigin-RevId: 588448489
2023-12-06 09:16:43 -08:00
..
dialects
Add jax.extend.mlir.
2023-12-06 09:16:43 -08:00
__init__.py
Add jax.extend.mlir.
2023-12-06 09:16:43 -08:00
BUILD
Add jax.extend.mlir.
2023-12-06 09:16:43 -08:00
ir.py
Add jax.extend.mlir.
2023-12-06 09:16:43 -08:00
passmanager.py
Add jax.extend.mlir.
2023-12-06 09:16:43 -08:00