mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 11:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

[CI] Update naming comments in workflow files.

Browse Source 
Comments have been added to indicate fields that if changed require a blocking presubmit check to be updated.

PiperOrigin-RevId: 721753931
This commit is contained in:
Michael Hudgins 2025-01-31 06:35:34 -08:00 committed by jax authors
parent cadfcc7a1b
commit 873db96f9f
3 changed files with 18 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
.github/workflows/bazel_cpu_rbe.yml vendored
View File

@ -22,21 +22,19 @@ concurrency:
jobs:
run_tests:
if: github.event.repository.fork == false
runs-on: ${{ matrix.runner }}
container: ${{ (contains(matrix.runner, 'linux-x86') && 'us-central1-docker.pkg.dev/tensorflow-sigs/tensorflow/ml-build:latest') ||
(contains(matrix.runner, 'linux-arm64') && 'us-central1-docker.pkg.dev/tensorflow-sigs/tensorflow/ml-build-arm64:latest') }}
env:
JAXCI_HERMETIC_PYTHON_VERSION: "3.12"
JAXCI_ENABLE_X64: ${{ matrix.enable-x_64 }}
# Begin Presubmit Naming Check - name modification requires internal check to be updated
strategy:
matrix:
runner: ["linux-x86-n2-16", "linux-arm64-c4a-16"]
enable-x_64: [1, 0]
runs-on: ${{ matrix.runner }}
container: ${{ (contains(matrix.runner, 'linux-x86') && 'us-central1-docker.pkg.dev/tensorflow-sigs/tensorflow/ml-build:latest') ||
(contains(matrix.runner, 'linux-arm64') && 'us-central1-docker.pkg.dev/tensorflow-sigs/tensorflow/ml-build-arm64:latest') }}
env:
JAXCI_HERMETIC_PYTHON_VERSION: "3.12"
JAXCI_ENABLE_X64: ${{ matrix.enable-x_64 }}
name: "Bazel CPU tests (${{ matrix.runner }}, Python 3.12, x64=${{ matrix.enable-x_64 }})"
# End Presubmit Naming Check github-cpu-presubmits
steps:
- uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
- name: Wait For Connection

16
.github/workflows/bazel_cuda_rbe.yml vendored
View File

@ -22,20 +22,18 @@ concurrency:
jobs:
run_tests:
if: github.event.repository.fork == false
runs-on: ${{ matrix.runner }}
container: 'us-central1-docker.pkg.dev/tensorflow-sigs/tensorflow/ml-build:latest'
env:
JAXCI_HERMETIC_PYTHON_VERSION: "3.12"
JAXCI_ENABLE_X64: ${{ matrix.enable-x_64 }}
# Begin Presubmit Naming Check - name modification requires internal check to be updated
strategy:
matrix:
runner: ["linux-x86-n2-16"]
enable-x_64: [1, 0]
runs-on: ${{ matrix.runner }}
container: 'us-central1-docker.pkg.dev/tensorflow-sigs/tensorflow/ml-build:latest'
env:
JAXCI_HERMETIC_PYTHON_VERSION: "3.12"
JAXCI_ENABLE_X64: ${{ matrix.enable-x_64 }}
name: "Bazel single accelerator CUDA tests (${{ matrix.runner }}, Python 3.12, x64=${{ matrix.enable-x_64 }})"
# End Presubmit Naming Check github-cuda-presubmits
steps:
- uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
- name: Wait For Connection

7
.github/workflows/cloud-tpu-ci-presubmit.yml vendored
View File

@ -30,6 +30,7 @@ concurrency:
jobs:
cloud-tpu-test:
if: github.event.repository.fork == false
# Begin Presubmit Naming Check - name modification requires internal check to be updated
strategy:
fail-fast: false # don't cancel all jobs on failure
matrix:
@ -37,9 +38,8 @@ jobs:
{type: "v5e-8", cores: "8", runner: "linux-x86-ct5lp-224-8tpu"}
]
python-version: ["3.10"]
name: "TPU test (jaxlib=head, ${{ matrix.tpu.type }})"
# End Presubmit Naming Check github-tpu-presubmits
env:
JAXCI_PYTHON: python${{ matrix.python-version }}
JAXCI_TPU_CORES: ${{ matrix.tpu.cores }}
@ -52,7 +52,6 @@ jobs:
defaults:
run:
shell: bash -ex {0}
steps:
# https://opensource.google/documentation/reference/github/services#actions
# mandates using a specific commit for non-Google actions. We use
@ -90,4 +89,4 @@ jobs:
with:
halt-dispatch-input: ${{ inputs.halt-for-connection }}
- name: Install jaxlib wheel and run tests
run: ./ci/run_pytest_tpu.sh
run: ./ci/run_pytest_tpu.sh