mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-15 19:36:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/ci
History
Nitin Srinivasan 721d1a3211 Add functionality to allow promoting RC wheels during release 
List of changes:
1. Allow us to build a RC wheel when building release artifacts. This is done by modifying the build CLI to use the new JAX build rule and passing in the build options that control the wheel tag. A new build argument `use_new_wheel_build_rule` is introduced to the build CLI to avoid breaking anyone that uses the CLI and the old build rule. Note that this option will go way in the future when the build CLI migrates fully to the new build rule.
2. Change the upload script to upload both rc and release tagged wheels (changes internal)

PiperOrigin-RevId: 733464219
2025-03-04 14:21:12 -08:00
..
envs
Add functionality to allow promoting RC wheels during release
2025-03-04 14:21:12 -08:00
utilities
Add functionality to allow promoting RC wheels during release
2025-03-04 14:21:12 -08: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
Print the list of installed packages before running pytests
2025-02-28 11:46:12 -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.