From c8f7624eae998ed397ae558638d4405875e4efef Mon Sep 17 00:00:00 2001 From: Bhavya Bahl Date: Tue, 1 Jul 2025 20:36:02 +0000 Subject: [PATCH 1/4] test ci --- setup.py | 1 + 1 file changed, 1 insertion(+) diff --git a/setup.py b/setup.py index dd6ad179405d..381c9d5b56d8 100644 --- a/setup.py +++ b/setup.py @@ -65,6 +65,7 @@ import platform +print(os.getcwd()) platform_machine = platform.machine() base_dir = os.path.dirname(os.path.abspath(__file__)) From 8bb791396fc07798fced279f3055f1bdc6e49d2f Mon Sep 17 00:00:00 2001 From: Bhavya Bahl Date: Tue, 1 Jul 2025 23:04:12 +0000 Subject: [PATCH 2/4] Update tpu_ci image --- infra/tpu-pytorch/tpu_ci.tf | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/infra/tpu-pytorch/tpu_ci.tf b/infra/tpu-pytorch/tpu_ci.tf index 4c724660c0ee..cf5c21e4c060 100644 --- a/infra/tpu-pytorch/tpu_ci.tf +++ b/infra/tpu-pytorch/tpu_ci.tf @@ -9,5 +9,5 @@ module "v4_arc_cluster" { max_tpu_nodes = 32 github_repo_url = "https://github.com/pytorch/xla" # Dockerfile for this image can be found at test/tpu/Dockerfile - runner_image = "gcr.io/tpu-pytorch/tpu-ci-runner:latest" + runner_image = "gcr.io/tpu-pytorch/tpu-ci-runner:test" } From 10141bb3e9c25922ab573cb7b3c32d3baf5824a5 Mon Sep 17 00:00:00 2001 From: Bhavya Bahl Date: Wed, 2 Jul 2025 00:44:05 +0000 Subject: [PATCH 3/4] Revert setup.py change --- setup.py | 1 - 1 file changed, 1 deletion(-) diff --git a/setup.py b/setup.py index 381c9d5b56d8..dd6ad179405d 100644 --- a/setup.py +++ b/setup.py @@ -65,7 +65,6 @@ import platform -print(os.getcwd()) platform_machine = platform.machine() base_dir = os.path.dirname(os.path.abspath(__file__)) From ecb4558cd20c7cd8f3db2ca8871c070f8fec2419 Mon Sep 17 00:00:00 2001 From: Bhavya Bahl Date: Wed, 2 Jul 2025 18:21:50 +0000 Subject: [PATCH 4/4] Add a TODO to update TPU image --- infra/tpu-pytorch/tpu_ci.tf | 1 + 1 file changed, 1 insertion(+) diff --git a/infra/tpu-pytorch/tpu_ci.tf b/infra/tpu-pytorch/tpu_ci.tf index cf5c21e4c060..fab86b18f60d 100644 --- a/infra/tpu-pytorch/tpu_ci.tf +++ b/infra/tpu-pytorch/tpu_ci.tf @@ -9,5 +9,6 @@ module "v4_arc_cluster" { max_tpu_nodes = 32 github_repo_url = "https://github.com/pytorch/xla" # Dockerfile for this image can be found at test/tpu/Dockerfile + # TODO(https://github.com/pytorch/xla/issues/8946): Update latest tag runner_image = "gcr.io/tpu-pytorch/tpu-ci-runner:test" }