mirror of
https://github.com/ROCm/jax.git
synced 2025-04-14 10:56:06 +00:00
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:
parent
6c76cc4e36
commit
14029c6762
@ -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
0
ci/utilities/run_docker_container.sh
Normal file → Executable file
Loading…
x
Reference in New Issue
Block a user