Skip to content

Commit 2d2ce84

Browse files
authored
Update v5p-8 cluster (#543)
Update v5p-8
1 parent c65b0be commit 2d2ce84

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

dags/common/vm_resource.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,7 @@ class Project(enum.Enum):
6262
CLOUD_TPU_INFERENCE_TEST = "cloud-tpu-inference-test"
6363
TPU_PROD_ENV_LARGE_ADHOC = "tpu-prod-env-large-adhoc"
6464
TPU_PROD_ENV_ONE_VM = "tpu-prod-env-one-vm"
65+
TPU_PROD_ENV_LARGE_CONT = "tpu-prod-env-large-cont"
6566

6667

6768
class ImageProject(enum.Enum):
@@ -233,7 +234,7 @@ class XpkClusters:
233234
name="v5p-8-bodaborg-us-east5-a",
234235
device_version=TpuVersion.V5P,
235236
core_count=8,
236-
project=Project.CLOUD_TPU_MULTIPOD_DEV.value,
237+
project=Project.TPU_PROD_ENV_LARGE_CONT.value,
237238
zone=Zone.US_EAST5_A.value,
238239
)
239240
TPU_V5E_256_CLUSTER = XpkClusterConfig(

0 commit comments

Comments
 (0)