fix: Point mxla_maxtext_nightly_gke to new cluster auto-v5p-8-bodaborg (GoogleCloudPlatform#1217)

Following the strategy to leverage increased compute capacity, this commit migrates all four test cases within the `mxla_maxtext_nightly_gke` DAG to the `auto-v5p-8-bodaborg` cluster. This move offloads pressure from the original `v5p-8-bodaborg` cluster and utilizes the expanded resources available in the new environment.

Author: camiloCienet

dags/multipod/mxla_maxtext_nightly_gke.py

@@ -23,8 +23,8 @@
 from dags.common.vm_resource import TpuVersion, Zone, DockerImage, XpkClusters, Project
 from dags.multipod.configs import gke_config
 
-# Run once a day at 9 am UTC (1 am PST)
-SCHEDULED_TIME = "0 2 * * *" if composer_env.is_prod_env() else None
+# Run once a day at 5 pm UTC (1 am PST)
+SCHEDULED_TIME = "0 17 * * *" if composer_env.is_prod_env() else None
 
 with models.DAG(
     dag_id="mxla_maxtext_nightly_gke",
@@ -51,7 +51,7 @@
 
   # v5p tests
   maxtext_nightly_1slice_v5p_8 = gke_config.get_gke_maxtext_nightly_config(
-      cluster=XpkClusters.TPU_V5P_8_CLUSTER,
+      cluster=XpkClusters.TPU_V5P_8_CLUSTER_V2,
       time_out_in_min=60,
       test_name=default_test_name,
       docker_image=jax_nightly_image.value,
@@ -60,7 +60,7 @@
 
   maxtext_nightly_2slice_v5p_8 = gke_config.get_gke_maxtext_nightly_config(
       num_slices=2,
-      cluster=XpkClusters.TPU_V5P_8_CLUSTER,
+      cluster=XpkClusters.TPU_V5P_8_CLUSTER_V2,
       time_out_in_min=60,
       test_name=default_test_name,
       docker_image=jax_nightly_image.value,
@@ -69,7 +69,7 @@
 
   maxtext_nightly_4slice_v5p_8 = gke_config.get_gke_maxtext_nightly_config(
       num_slices=4,
-      cluster=XpkClusters.TPU_V5P_8_CLUSTER,
+      cluster=XpkClusters.TPU_V5P_8_CLUSTER_V2,
       time_out_in_min=60,
       test_name=default_test_name,
       docker_image=jax_nightly_image.value,
@@ -78,7 +78,7 @@
 
   maxtext_nightly_8slice_v5p_8 = gke_config.get_gke_maxtext_nightly_config(
       num_slices=8,
-      cluster=XpkClusters.TPU_V5P_8_CLUSTER,
+      cluster=XpkClusters.TPU_V5P_8_CLUSTER_V2,
       time_out_in_min=60,
       test_name=default_test_name,
       docker_image=jax_nightly_image.value,