mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-22 08:36:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
031c0acf50
rocm_jax/ci
History
Nitin Srinivasan 031c0acf50 Add new CI scripts for running Pytests 
This commit adds the new CI scripts for running Pytests. It makes use of the pytest envs inside the "ci/envs/run_tests" folder to control the build behavior. For e.g: for running the GPU tests with Pytest, we will need to run `./ci/run_pytest.sh ./ci/envs/run_tests/pytest_gpu.env`. Note that Pytests need JAX wheels to be installed on the system to work. The `install_wheels_locally.sh` script installs these wheels in CI builds.

PiperOrigin-RevId: 701331411
2024-11-29 12:08:17 -08:00
..
envs
Add new CI scripts for running Pytests
2024-11-29 12:08:17 -08:00
utilities
Add new CI scripts for running Pytests
2024-11-29 12:08:17 -08:00
build_artifacts.sh
Add new CI scripts for building JAX artifacts
2024-11-25 14:37:02 -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_gpu_non_rbe.sh
Add new CI script to run Bazel GPU (non-RBE) jobs
2024-11-26 18:42:27 -08:00
run_bazel_test_gpu_rbe.sh
Add new CI script to run Bazel GPU (non-RBE) jobs
2024-11-26 18:42:27 -08:00
run_pytest_cpu.sh
Add new CI scripts for running Pytests
2024-11-29 12:08:17 -08:00
run_pytest_gpu.sh
Add new CI scripts for running Pytests
2024-11-29 12:08:17 -08:00
run_pytest_tpu.sh
Add new CI scripts for running Pytests
2024-11-29 12:08:17 -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.