mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 04:16:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/.github/workflows
History
Nitin Srinivasan a6ab6bbc20 Ignore Pallas TPU tests when testing with the oldest supported libtpu 
I missed adding this in from https://github.com/jax-ml/jax/blob/main/.github/workflows/cloud-tpu-ci-nightly.yml when I added the TPU jobs to the new CI workflows

PiperOrigin-RevId: 736094492
2025-03-12 05:20:42 -07:00
..
self_hosted_runner_utils
Change references in setup.py and utilities to reference the JAX repo move to the JAX-ML org
2024-09-20 07:32:15 -07:00
asan.yaml
Reduce pytest workers for asan to resolve memory usage causing OOM
2025-02-27 14:54:29 -08:00
bazel_cpu_rbe.yml
Replace Python 3.12 with Python 3.13 and add Python 3.10 to the matrix
2025-02-05 08:27:22 -08:00
bazel_cuda_non_rbe.yml
Use new ML Build CUDA images
2025-02-21 10:16:16 -08:00
bazel_cuda_rbe.yml
Replace Python 3.12 with Python 3.13 and add Python 3.10 to the matrix
2025-02-05 08:27:22 -08:00
build_artifacts.yml
Use the 64 core Windows runner to build artifacts
2025-02-25 13:42:16 -08:00
ci-build.yaml
Use uv instead of pip for installing Python packages
2025-02-25 17:48:22 -08:00
cloud-tpu-ci-nightly.yml
Bump oldest supported libtpu to match the compatibility window (12 weeks)
2025-02-28 12:13:42 -08:00
cloud-tpu-ci-presubmit.yml
Add TPU test jobs to the new CI continuous and nightly/release test workflows
2025-03-11 11:42:21 -07:00
jax-array-api.yml
Update array-api-tests to latest commit
2025-03-04 10:17:51 -08:00
metal_plugin_ci.yml
Use uv to install Python packages
2025-02-24 10:13:39 -08:00
pytest_cpu.yml
Remove installation of NVIDIA wheels for CPU tests
2025-03-11 13:43:13 -07:00
pytest_cuda.yml
Build JAX wheels instead of installing it from the source repository
2025-03-06 06:48:16 -08:00
pytest_tpu.yml
Ignore Pallas TPU tests when testing with the oldest supported libtpu
2025-03-12 05:20:42 -07:00
README.md
CI: pin actions/checkout to v4
2023-09-05 09:26:15 -07:00
release-notification.yml
Update Github Actions workflows using Ratchet
2022-11-16 18:45:59 +00:00
requirements_lock_3_13_ft.patch
Added CI job with TSAN and free-threading
2025-01-20 17:12:47 +01:00
rocm-ci.yml
[ROCm] Only run core tests for ROCm post-build checks
2025-02-10 18:30:44 +00:00
tsan-suppressions.txt
Remove a TSAN suppression.
2025-03-05 08:39:58 -05:00
tsan.yaml
Add file and zip to tsan.yaml
2025-03-10 16:51:05 +00:00
upstream-nightly.yml
Add --system to uv commands in upstream-nightly workflow.
2025-02-26 13:21:41 -05:00
wheel_tests_continuous.yml
Add TPU test jobs to the new CI continuous and nightly/release test workflows
2025-03-11 11:42:21 -07:00
wheel_tests_nightly_release.yml
Run single python version for v4-8 and min & max for v5e-8 for TPU tests in nightly/release test workflow
2025-03-11 20:03:05 -07:00
wheel_win_x64.yml
Use uv to install Python packages
2025-02-24 10:13:39 -08:00
windows_ci.yml
Fix incorrect line separator
2025-02-25 09:28:06 -08:00

README.md

Github Actions workflows

See the Github documentation for more information on Github Actions in general.

Notes

  • https://opensource.google/documentation/reference/github/services#actions mandates using a specific commit for non-Google actions. We use Ratchet to pin specific versions. If you'd like to update an action, you can write something like uses: 'actions/checkout@v4', and then run ./ratchet pin workflow.yml to convert to a commit hash. See the Ratchet README for installation and more detailed instructions.