Skip to content

Commit

Permalink
Bump oldest support libtpu version to 20230927 which includes CopyToM…
Browse files Browse the repository at this point in the history
…emorySpace.

batch_device_put in libtpu will go through memory space path, and requires CopyToMemorySpace to be implemented if the backend uses memory space.

PiperOrigin-RevId: 591373300
  • Loading branch information
Jieying Luo authored and jax authors committed Dec 15, 2023
1 parent bc239fd commit 717059e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion .github/workflows/cloud-tpu-ci-nightly.yml
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ jobs:
tpu-type: ["v3-8", "v4-8", "v5e-4"]
name: "TPU test (jaxlib=${{ matrix.jaxlib-version }}, ${{ matrix.tpu-type }})"
env:
LIBTPU_OLDEST_VERSION_DATE: 20230915
LIBTPU_OLDEST_VERSION_DATE: 20230927
ENABLE_PJRT_COMPATIBILITY: ${{ matrix.jaxlib-version == 'nightly+oldest_supported_libtpu' }}
runs-on: ["self-hosted", "tpu", "${{ matrix.tpu-type }}"]
timeout-minutes: 120
Expand Down

0 comments on commit 717059e

Please sign in to comment.