rocm_jax

mirror of https://github.com/ROCm/jax.git synced 2025-04-24 10:56:06 +00:00

History

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

See https://opensource.google/documentation/reference/releasing/contributions#copyright for more details.

PiperOrigin-RevId: 476167538

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

array.filecheck.py

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

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

jax_filecheck_helpers.py

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

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

math.filecheck.py

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

2022-09-22 12:27:19 -07: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

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

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

shapes.filecheck.py

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

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

subcomputations.filecheck.py

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

2022-09-22 12:27:19 -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.