Skip to content

Commit 71b23ea

Browse files
hawkinspGoogle-ML-Automation
authored andcommitted
Enable PJRT compatibility in cloud TPU CI.
There's no reason to limit this to only one CI job. PiperOrigin-RevId: 707227820
1 parent 4aebe55 commit 71b23ea

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

.github/workflows/cloud-tpu-ci-nightly.yml

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,6 @@ jobs:
3434
name: "TPU test (jaxlib=${{ matrix.jaxlib-version }}, ${{ matrix.tpu.type }})"
3535
env:
3636
LIBTPU_OLDEST_VERSION_DATE: 20240922
37-
ENABLE_PJRT_COMPATIBILITY: ${{ matrix.jaxlib-version == 'nightly+oldest_supported_libtpu' }}
3837
PYTHON: python${{ matrix.python-version }}
3938
runs-on: ${{ matrix.tpu.runner }}
4039
container: "us-central1-docker.pkg.dev/tensorflow-sigs/tensorflow/ml-build:latest"

0 commit comments

Comments
 (0)