Skip to content

Commit ca04583

Browse files
authored
Update TPU CI with latest docker container (#9483)
1 parent 93a5e58 commit ca04583

File tree

2 files changed

+3
-4
lines changed

2 files changed

+3
-4
lines changed

infra/tpu-pytorch/tpu_ci.tf

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,5 @@ module "v4_arc_cluster" {
99
max_tpu_nodes = 32
1010
github_repo_url = "https://github.com/pytorch/xla"
1111
# Dockerfile for this image can be found at test/tpu/Dockerfile
12-
# TODO(https://github.com/pytorch/xla/issues/8946): Update latest tag
13-
runner_image = "gcr.io/tpu-pytorch/tpu-ci-runner:test"
12+
runner_image = "gcr.io/tpu-pytorch/tpu-ci-runner:latest"
1413
}

test/tpu/Dockerfile

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,12 +3,12 @@ FROM us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/development:tpu as b
33
# Replace value with the latest runner release version
44
# source: https://github.com/actions/runner/releases
55
# ex: 2.303.0
6-
ARG RUNNER_VERSION="2.323.0"
6+
ARG RUNNER_VERSION="2.326.0"
77
ARG RUNNER_ARCH="x64"
88
# Replace value with the latest runner-container-hooks release version
99
# source: https://github.com/actions/runner-container-hooks/releases
1010
# ex: 0.3.1
11-
ARG RUNNER_CONTAINER_HOOKS_VERSION="0.6.2"
11+
ARG RUNNER_CONTAINER_HOOKS_VERSION="0.7.0"
1212

1313
ARG USER=runner
1414

0 commit comments

Comments
 (0)