Peter Hawkins 70f91db853 Set PYTHONWARNINGS=error in bazel tests. ...

The goal of this change is to catch PRs that introduce new warnings sooner.

To help pass the environment variable more easily, rename the jax_test Bazel test macro to jax_multiplatform_test, and introduce a new jax_py_test macro that wraps py_test. Add code to both to set the environment variable.

Add code to suppress some new warnings uncovered in CI.

PiperOrigin-RevId: 678352286

2024-09-24 12:30:11 -07:00

..

mosaic

Set PYTHONWARNINGS=error in bazel tests.

2024-09-24 12:30:11 -07:00

api_benchmark.py

Fixes to api_benchmark.py. Testcases always fail without these fixes.

2024-08-27 23:37:49 -07:00

linalg_benchmark.py

Use a Jacobi SVD solver for unbatched SVDs up to 1024x1024 on NVIDIA GPUs.

2023-11-13 12:04:13 -08:00

math_benchmark.py

Update flags to use the ABSL typed flag API.

2023-07-27 12:15:58 -07:00

random_benchmark.py

fast dispatch for functions over typed PRNG key arrays

2023-09-13 09:43:58 -07:00

shape_poly_benchmark.py

[export] Create the jax.export module APIs.

2024-06-10 19:31:51 +02:00

sparse_benchmark.py

update README and several docs to typed RNG keys

2024-08-11 08:09:47 -07:00