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

Remove StreamExecutor-based TPU runtime from Cloud TPU CI

Browse Source 
The old StreamExecutor-based backend is no longer supported as of
3e50fea29e
This commit is contained in:
Skye Wanderman-Milne 2023-08-09 10:05:46 -07:00
parent be543f020d
commit 0f30685dac
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
.github/workflows/cloud-tpu-ci-nightly.yml vendored
View File

@ -14,8 +14,7 @@ jobs:
matrix:
jaxlib-version: ["latest", "nightly"]
tpu-type: ["v3-8", "v4-8", "nightly"]
pjrt: ["true", "false"]
name: "TPU test (${{ matrix.jaxlib-version }}, pjrt=${{ matrix.pjrt }}, ${{ matrix.tpu-type }})"
name: "TPU test (${{ matrix.jaxlib-version }}, ${{ matrix.tpu-type }})"
runs-on: ["self-hosted", "tpu", "${{ matrix.tpu-type }}"]
timeout-minutes: 60
steps:
@ -55,7 +54,6 @@ jobs:
- name: Run tests
env:
JAX_PLATFORMS: tpu,cpu
JAX_USE_PJRT_C_API_ON_TPU: ${{ matrix.pjrt }}
PY_COLORS: 1
run: |
# Run single-accelerator tests in parallel