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 c78487d23d Add Github action workflows for running continuous tests with Pytest 
Changes:
- Adds `wheel_tests.yml` that will be used to run continuous jobs that builds artifacts and runs CPU/CUDA tests. Jobs will run by workflow calls to `build_artifacts.yml`/`pytest_cpu.yml`/`pytest_gpu.yml`.
- Adds testing of CUDA tests on H100 gpus
- Make script executable
- Change the name of GPU scripts and workflows to CUDA to be more clear as to what is being tested
PiperOrigin-RevId: 715500412
2025-01-14 13:10:51 -08:00
..
envs
Add utility script and env for running the CI scripts under Docker
2024-12-16 12:19:23 -08:00
utilities
Add Github action workflows for running continuous tests with Pytest
2025-01-14 13:10:51 -08:00
build_artifacts.sh
Add Github action workflows for running continuous tests with Pytest
2025-01-14 13:10:51 -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 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
Add an experimental Cloud TPU presubmit job
2024-12-13 18:06: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.