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

History

Nitin Srinivasan 7ac6355262 Add TPU test jobs to the new CI continuous and nightly/release test workflows

Also, modify the TPU presubmit workflow to reuse the `build_artifacts.yml` and `pytest_tpu.yml`

PiperOrigin-RevId: 735832964

2025-03-11 11:42:21 -07:00

envs

Add TPU test jobs to the new CI continuous and nightly/release test workflows

2025-03-11 11:42:21 -07:00

utilities

Add TPU test jobs to the new CI continuous and nightly/release test workflows

2025-03-11 11:42:21 -07:00

build_artifacts.sh

Add functionality to allow promoting RC wheels during release

2025-03-04 14:21:12 -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

Update the calculation for num_processes and num_test_jobs that are used in CUDA test jobs

2025-02-27 08:11:51 -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

Print the list of installed packages before running pytests

2025-02-28 11:46:12 -08:00

run_pytest_cuda.sh

Print the list of installed packages before running pytests

2025-02-28 11:46:12 -08:00

run_pytest_tpu.sh

Add TPU test jobs to the new CI continuous and nightly/release test workflows

2025-03-11 11:42:21 -07: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.