Enable PJRT compatibility in cloud TPU CI.

There's no reason to limit this to only one CI job. PiperOrigin-RevId: 707227820
jax-ml · Dec 17, 2024 · 71b23ea · 71b23ea
1 parent 4aebe55
commit 71b23ea
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/.github/workflows/cloud-tpu-ci-nightly.yml b/.github/workflows/cloud-tpu-ci-nightly.yml
@@ -34,7 +34,6 @@ jobs:
     name: "TPU test (jaxlib=${{ matrix.jaxlib-version }}, ${{ matrix.tpu.type }})"
     env:
       LIBTPU_OLDEST_VERSION_DATE: 20240922
-      ENABLE_PJRT_COMPATIBILITY: ${{ matrix.jaxlib-version == 'nightly+oldest_supported_libtpu' }}
       PYTHON: python${{ matrix.python-version }}
     runs-on: ${{ matrix.tpu.runner }}
     container: "us-central1-docker.pkg.dev/tensorflow-sigs/tensorflow/ml-build:latest"