rocm_jax

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

History

Eugene Burmako b8ae8e3fa1 (NFC) Prepare for migration from producing MHLO to producing StableHLO

This CL renames occurrences of "mhlo" in: 1) names, 2) tests, 3) prose in order
to prepare for the upcoming migration.

Unchanged occurrences:
  1) Public API that contains "mhlo", e.g. XlaLowering.mhlo and the "mhlo"
     argument value in Lowering.as_text and Lowering.compiler_ir.
  2) Documentation (changelog, JEPs, IR examples, etc).
  3) One rare situation where prose says "StableHLO" and "MHLO" in one sentence,
     so both are necessary to disambiguate.

PiperOrigin-RevId: 495771153

2022-12-15 21:00:07 -08:00

array.filecheck.py

(NFC) Prepare for migration from producing MHLO to producing StableHLO

2022-12-15 21:00:07 -08:00

jax_filecheck_helpers.py

(NFC) Prepare for migration from producing MHLO to producing StableHLO

2022-12-15 21:00:07 -08:00

math.filecheck.py

(NFC) Prepare for migration from producing MHLO to producing StableHLO

2022-12-15 21:00:07 -08:00

names.filecheck.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

README.md

(NFC) Prepare for migration from producing MHLO to producing StableHLO

2022-12-15 21:00:07 -08:00

shapes.filecheck.py

(NFC) Prepare for migration from producing MHLO to producing StableHLO

2022-12-15 21:00:07 -08:00

subcomputations.filecheck.py

(NFC) Prepare for migration from producing MHLO to producing StableHLO

2022-12-15 21:00:07 -08:00

README.md

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

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.