rocm_jax

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

History

Peter Hawkins 1b8be90801 Remove the jax_enable_mlir flag. MLIR is now the only supported code path.

This change does not yet remove all the XLA translation rule code since it may be used in various fallback paths. Only the top-level lowering function is removed. Further cleanup is left to subsequent changes.

PiperOrigin-RevId: 439324450

2022-04-04 08:40:09 -07:00

array.filecheck.py

Remove the jax_enable_mlir flag. MLIR is now the only supported code path.

2022-04-04 08:40:09 -07:00

jax_filecheck_helpers.py

[JAX] Add a dialect option to jit(...).lower(...).compiler_ir().

2021-12-10 13:02:25 -08:00

math.filecheck.py

Remove the jax_enable_mlir flag. MLIR is now the only supported code path.

2022-04-04 08:40:09 -07:00

names.filecheck.py

Remove the jax_enable_mlir flag. MLIR is now the only supported code path.

2022-04-04 08:40:09 -07:00

README.md

Add some initial filecheck tests for JAX->MHLO lowering.

2021-12-10 10:59:24 -08:00

shapes.filecheck.py

Remove the jax_enable_mlir flag. MLIR is now the only supported code path.

2022-04-04 08:40:09 -07:00

subcomputations.filecheck.py

Remove the jax_enable_mlir flag. MLIR is now the only supported code path.

2022-04-04 08:40:09 -07:00

README.md

This directory contains LLVM FileCheck tests that verify that JAX primitives can be lowered to MHLO.

These tests are intended to be a quick and easy-to-understand way to catch regressions from changes due the MLIR Python bindings and from changes to the various MLIR dialects used by JAX, without needing to run the full JAX test suite.