mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 21:06:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Bump oldest support libtpu version to 20230927 which includes CopyToMemorySpace.

Browse Source 
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
This commit is contained in:
Jieying Luo 2023-12-15 15:18:23 -08:00 committed by jax authors
parent bc239fd1b3
commit 717059edcf
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -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