From 693ee3f03187b7fd2c8d2753245a2072c720c0b0 Mon Sep 17 00:00:00 2001
From: Pei Zhang <piz@google.com>
Date: Mon, 5 Feb 2024 14:11:05 -0800
Subject: [PATCH] update A100 to use us-central1-f (#100)

---
 dags/pytorch_xla/pytorchxla_torchbench.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/dags/pytorch_xla/pytorchxla_torchbench.py b/dags/pytorch_xla/pytorchxla_torchbench.py
index df1b3aa0..c1e97656 100644
--- a/dags/pytorch_xla/pytorchxla_torchbench.py
+++ b/dags/pytorch_xla/pytorchxla_torchbench.py
@@ -93,7 +93,7 @@
       image_family=resource.ImageFamily.COMMON_CU121_DEBIAN_11,
       accelerator_type=resource.GpuVersion.A100,
       count=1,
-      gpu_zone=resource.Zone.US_CENTRAL1_C,
+      gpu_zone=resource.Zone.US_CENTRAL1_F,
       model_name=model,
       time_out_in_min=1600,
       extraFlags=" ".join(torchbench_extra_flags),