diff --git a/.github/workflows/bazel_gpu_rbe.yml b/.github/workflows/bazel_cuda_rbe.yml similarity index 81% rename from .github/workflows/bazel_gpu_rbe.yml rename to .github/workflows/bazel_cuda_rbe.yml index 0b3d126ab..33baf274c 100644 --- a/.github/workflows/bazel_gpu_rbe.yml +++ b/.github/workflows/bazel_cuda_rbe.yml @@ -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 \ No newline at end of file + - name: Run Bazel CUDA Tests with RBE + run: ./ci/run_bazel_test_cuda_rbe.sh \ No newline at end of file diff --git a/ci/run_bazel_test_gpu_rbe.sh b/ci/run_bazel_test_cuda_rbe.sh similarity index 100% rename from ci/run_bazel_test_gpu_rbe.sh rename to ci/run_bazel_test_cuda_rbe.sh diff --git a/ci/utilities/run_docker_container.sh b/ci/utilities/run_docker_container.sh old mode 100644 new mode 100755