rocm_jax/filecheck at rocm-main - rocm_jax - Gitea For EOELAB

mirrors/rocm_jax

mirror of https://github.com/ROCm/jax.git synced 2025-04-14 19:06:07 +00:00

History

Jake VanderPlas e389b707ba Add public APIs for jax.lax monoidal reductions

2025-02-11 16:00:03 -08:00

..

array.filecheck.py

Add public APIs for jax.lax monoidal reductions

2025-02-11 16:00:03 -08:00

custom_call.filecheck.py

Integrate StableHLO at openxla/stablehlo@8817ff1d

2024-07-15 10:38:09 -07:00

jax_filecheck_helpers.py

tests: access tree utilities via jax.tree.*

2024-02-26 14:17:18 -08:00

math.filecheck.py

Register acos primitive to lower to CHLO acos.

2024-10-25 13:20:36 -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

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

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

shapes.filecheck.py

Fix complex sin and cos on inputs with small absolute value or large pure imaginary part

2024-02-22 23:42:18 +02:00

subcomputations.filecheck.py

filecheck test: use lax.cumsum directly to prevent false-positive

2024-09-04 12:31:19 -07: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.