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

History

Adam Paszke f1ab7514db Make sure we take libTPU version into account in the Pallas lowering

Also, strengthen the presubmit to make sure we catch more errors.

PiperOrigin-RevId: 726061633

2025-02-12 08:15:57 -08:00

envs

Add new Bazel remote cache configs

2025-01-25 05:32:42 -08:00

utilities

Add a retry logic to avoid flakes when downloading Bazel

2025-01-27 16:36:01 -08:00

build_artifacts.sh

Add new Bazel remote cache configs

2025-01-25 05:32:42 -08:00

README.md

Add new CI scripts for running Bazel CPU presubmits

2024-11-12 20:00:35 -08:00

run_bazel_test_cpu_rbe.sh

Add new CI script to run Bazel GPU (non-RBE) jobs

2024-11-26 18:42:27 -08:00

run_bazel_test_cuda_non_rbe.sh

Add new Bazel remote cache configs

2025-01-25 05:32:42 -08:00

run_bazel_test_cuda_rbe.sh

Replace "gpu" with "cuda" to be specific about the type of gpu tests that are running

2025-01-22 09:40:11 -08:00

run_pytest_cpu.sh

Add Github action workflows for running continuous tests with Pytest

2025-01-14 13:10:51 -08:00

run_pytest_cuda.sh

Add Github action workflows for running continuous tests with Pytest

2025-01-14 13:10:51 -08:00

run_pytest_tpu.sh

Make sure we take libTPU version into account in the Pallas lowering

2025-02-12 08:15:57 -08:00

README.md

JAX continuous integration

Warning

This folder is still under construction. It is part of an ongoing effort to improve the structure of CI and build related files within the JAX repo. This warning will be removed when the contents of this directory are stable and appropriate documentation around its usage is in place.