From 174f4077b8bad4cdaec4d537b89167b9ad77ec56 Mon Sep 17 00:00:00 2001 From: Will Cromar Date: Fri, 26 Apr 2024 16:16:55 -0700 Subject: [PATCH] Run TPU CI when label is on PR (#6984) --- .github/workflows/build_and_test.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/build_and_test.yml b/.github/workflows/build_and_test.yml index ce90448738a..ffb75c3ef8d 100644 --- a/.github/workflows/build_and_test.yml +++ b/.github/workflows/build_and_test.yml @@ -42,7 +42,7 @@ jobs: # New CI workflow build-torch-xla: - name: "Build PyTorch/XLA (TPU)" + name: "Build PyTorch/XLA" uses: ./.github/workflows/_build_torch_xla.yml with: dev-image: us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/development:3.10_tpuvm @@ -86,5 +86,5 @@ jobs: uses: ./.github/workflows/_tpu_ci.yml needs: build-torch-xla # Only run this for HEAD and releases - if: github.event_name == 'push' + if: github.event_name == 'push' || contains(github.event.pull_request.labels.*.name, 'tpuci')