Replace "gpu" with "cuda" to be specific about the type of gpu tests that are running

Also, make `run_docker_container.sh` executable

PiperOrigin-RevId: 718420712
This commit is contained in:
Nitin Srinivasan 2025-01-22 09:39:33 -08:00 committed by jax authors
parent 6c76cc4e36
commit 14029c6762
3 changed files with 4 additions and 4 deletions

View File

@ -1,4 +1,4 @@
name: CI - Bazel GPU tests (RBE)
name: CI - Bazel CUDA tests (RBE)
on:
workflow_dispatch:
@ -34,7 +34,7 @@ jobs:
JAXCI_HERMETIC_PYTHON_VERSION: "3.12"
JAXCI_ENABLE_X64: ${{ matrix.enable-x_64 }}
name: "Bazel single accelerator GPU tests (${{ matrix.runner }}, Python 3.12, x64=${{ matrix.enable-x_64 }})"
name: "Bazel single accelerator CUDA tests (${{ matrix.runner }}, Python 3.12, x64=${{ matrix.enable-x_64 }})"
steps:
- uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
@ -42,5 +42,5 @@ jobs:
uses: google-ml-infra/actions/ci_connection@main
with:
halt-dispatch-input: ${{ inputs.halt-for-connection }}
- name: Run Bazel GPU Tests with RBE
run: ./ci/run_bazel_test_gpu_rbe.sh
- name: Run Bazel CUDA Tests with RBE
run: ./ci/run_bazel_test_cuda_rbe.sh

0
ci/utilities/run_docker_container.sh Normal file → Executable file
View File