Adding MTP DSV3 Tests (GoogleCloudPlatform#1229)

parambole · web-flow · commit 9fa1c0db6585 · 2026-04-21T11:33:19.000-07:00
diff --git a/dags/sparsity_diffusion_devx/maxtext_moe_tpu_e2e.py b/dags/sparsity_diffusion_devx/maxtext_moe_tpu_e2e.py
@@ -68,6 +68,11 @@
           "cluster": XpkClusters.TPU_V5P_128_CLUSTER,
           "time_out_in_min": 90,
       },
+      "deepseek3-671b-mtp": {
+          "script_name": "tpu/deepseek/v3-671b/2_test_deepseek_mtp",
+          "cluster": XpkClusters.TPU_V5P_128_CLUSTER,
+          "time_out_in_min": 90,
+      },
       "deepseek2-16b": {
           "script_name": "tpu/deepseek/v2-16b/test_deepseek",
           "cluster": XpkClusters.TPU_V5P_8_CLUSTER_V2,
