From 1057b6330416a129fac933a8a763a342d3752a7c Mon Sep 17 00:00:00 2001 From: Michael Whittaker Date: Thu, 8 May 2025 10:29:51 -0700 Subject: [PATCH] Bumping oldest libtpu version for 12 week window. --- .github/workflows/cloud-tpu-ci-nightly.yml | 2 +- .github/workflows/wheel_tests_continuous.yml | 5 ++++- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/.github/workflows/cloud-tpu-ci-nightly.yml b/.github/workflows/cloud-tpu-ci-nightly.yml index 5cc2aebe3cd0..0a605ccb2c28 100644 --- a/.github/workflows/cloud-tpu-ci-nightly.yml +++ b/.github/workflows/cloud-tpu-ci-nightly.yml @@ -44,7 +44,7 @@ jobs: jaxlib-version: "pypi_latest" name: "TPU test (jaxlib=${{ matrix.jaxlib-version }}, ${{ matrix.tpu.type }})" env: - LIBTPU_OLDEST_VERSION_DATE: 20241205 + LIBTPU_OLDEST_VERSION_DATE: 20250213 PYTHON: python${{ matrix.python-version }} runs-on: ${{ matrix.tpu.runner }} container: "us-central1-docker.pkg.dev/tensorflow-sigs/tensorflow/ml-build:latest" diff --git a/.github/workflows/wheel_tests_continuous.yml b/.github/workflows/wheel_tests_continuous.yml index 207075fd0340..22584be0b740 100644 --- a/.github/workflows/wheel_tests_continuous.yml +++ b/.github/workflows/wheel_tests_continuous.yml @@ -25,6 +25,9 @@ on: schedule: - cron: "0 */3 * * *" # Run once every 3 hours workflow_dispatch: # allows triggering the workflow run manually + pull_request: + branches: + - main concurrency: group: ${{ github.workflow }}-${{ github.head_ref || github.ref }} @@ -212,4 +215,4 @@ jobs: python: ${{ matrix.python }} run-full-tpu-test-suite: "1" libtpu-version-type: ${{ matrix.libtpu-version-type }} - gcs_download_uri: ${{ needs.build-jaxlib-artifact.outputs.gcs_upload_uri }} \ No newline at end of file + gcs_download_uri: ${{ needs.build-jaxlib-artifact.outputs.gcs_upload_uri }}