Skip to content

Commit

Permalink
Update TPU ARC cluster (#7677)
Browse files Browse the repository at this point in the history
  • Loading branch information
will-cromar authored Jul 12, 2024
1 parent 1651e76 commit 8390de2
Show file tree
Hide file tree
Showing 4 changed files with 46 additions and 5 deletions.
2 changes: 1 addition & 1 deletion .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@ torch_xla/csrc/version.cpp
/docs/src/*

# Local terraform state
.terraform*
.terraform


# Build system temporary files
Expand Down
4 changes: 2 additions & 2 deletions infra/terraform_modules/arc_v4_container_cluster/main.tf
Original file line number Diff line number Diff line change
Expand Up @@ -73,7 +73,7 @@ resource "google_container_node_pool" "arc_v4_tpu_nodes" {

resource "helm_release" "arc" {
name = "actions-runner-controller"
chart = "oci://ghcr.io/actions/actions-runner-controller-charts/gha-runner-scale-set-controller"
chart = "oci://ghcr.io/actions/actions-runner-controller-charts/gha-runner-scale-set-controller:0.9.3"
namespace = var.arc_namespace
create_namespace = true
}
Expand All @@ -83,7 +83,7 @@ resource "helm_release" "arc_runner_set" {
depends_on = [
helm_release.arc
]
chart = "oci://ghcr.io/actions/actions-runner-controller-charts/gha-runner-scale-set"
chart = "oci://ghcr.io/actions/actions-runner-controller-charts/gha-runner-scale-set:0.9.3"
namespace = var.runner_namespace
create_namespace = true

Expand Down
41 changes: 41 additions & 0 deletions infra/tpu-pytorch/.terraform.lock.hcl

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

4 changes: 2 additions & 2 deletions test/tpu/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -3,12 +3,12 @@ FROM us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/development:tpu as b
# Replace value with the latest runner release version
# source: https://github.com/actions/runner/releases
# ex: 2.303.0
ARG RUNNER_VERSION="2.316.1"
ARG RUNNER_VERSION="2.317.0"
ARG RUNNER_ARCH="x64"
# Replace value with the latest runner-container-hooks release version
# source: https://github.com/actions/runner-container-hooks/releases
# ex: 0.3.1
ARG RUNNER_CONTAINER_HOOKS_VERSION="0.6.0"
ARG RUNNER_CONTAINER_HOOKS_VERSION="0.6.1"

ARG USER=runner

Expand Down

0 comments on commit 8390de2

Please sign in to comment.