fix: Use a new cluster for DAG maxtext_muti_tier_checkpointing (GoogleCloudPlatform#1123)

This change fixes DAG `maxtext_muti_tier_checkpointing` by pointing it to a new cluster `auto-v5p-8-bodaborg` due to overlapping DAG's pointing same cluster.

Author: camiloCienet

dags/common/vm_resource.py

@@ -255,6 +255,13 @@ class XpkClusters:
       project=Project.CLOUD_TPU_MULTIPOD_DEV.value,
       zone=Zone.EUROPE_WEST4_B.value,
   )
+  TPU_V5P_8_CLUSTER_V2 = XpkClusterConfig(
+      name="auto-v5p-8-bodaborg",
+      device_version=TpuVersion.V5P,
+      core_count=8,
+      project=Project.CLOUD_TPU_MULTIPOD_DEV.value,
+      zone=Zone.EUROPE_WEST4_B.value,
+  )
   TPU_V5P_128_CLUSTER = XpkClusterConfig(
       name="v5p-128-bodaborg-europe-west4-b",
       device_version=TpuVersion.V5P,

dags/multipod/maxtext_multi_tier_checkpointing.py

@@ -53,7 +53,7 @@
   }
   clusters = {
       # accelerator: cluster name
-      "v5p-8": XpkClusters.TPU_V5P_8_CLUSTER,
+      "v5p-8": XpkClusters.TPU_V5P_8_CLUSTER_V2,
   }
 
   for mode, image in docker_images:
@@ -72,4 +72,4 @@
             run_model_cmds=command,
             docker_image=image.value,
             test_owner=test_owner.ABHINAV_S,
-        ).run(ramdisk_directory="local")
+        ).run(ramdisk_directory="local", mtc_enabled=True)