Merge pull request #1045 from kangminxie/16pods-4blocks

Updating the v5e example of GKE terraform for 16 VLP pods
tensorflow · Sep 18, 2023 · 2758580 · 2758580
2 parents a1bec7a + 90fdc4a
commit 2758580
Showing 1 changed file with 77 additions and 5 deletions.
diff --git a/tools/kubernetes/terraform/examples/v5e/terraform.tfvars b/tools/kubernetes/terraform/examples/v5e/terraform.tfvars
@@ -6,24 +6,96 @@ tpu_node_pools = [{
   node_count   = 64
   machine_type = "ct5lp-hightpu-4t"
   topology     = "16x16"
-  policy       = "sb-compact-1"
+  policy       = "sb-compact-4a"
   }, {
   zone         = "us-east5-b"
   node_count   = 64
   machine_type = "ct5lp-hightpu-4t"
   topology     = "16x16"
-  policy       = "sb-compact-1"
+  policy       = "sb-compact-4a"
   }, {
   zone         = "us-east5-b"
   node_count   = 64
   machine_type = "ct5lp-hightpu-4t"
   topology     = "16x16"
-  policy       = "sb-compact-1"
+  policy       = "sb-compact-4a"
   }, {
   zone         = "us-east5-b"
   node_count   = 64
   machine_type = "ct5lp-hightpu-4t"
   topology     = "16x16"
-  policy       = "sb-compact-1"
-  }]
+  policy       = "sb-compact-4a"
+  }, {
+  zone         = "us-east5-b"
+  node_count   = 64
+  machine_type = "ct5lp-hightpu-4t"
+  topology     = "16x16"
+  policy       = "sb-compact-4b"
+  }, {
+  zone         = "us-east5-b"
+  node_count   = 64
+  machine_type = "ct5lp-hightpu-4t"
+  topology     = "16x16"
+  policy       = "sb-compact-4b"
+  }, {
+  zone         = "us-east5-b"
+  node_count   = 64
+  machine_type = "ct5lp-hightpu-4t"
+  topology     = "16x16"
+  policy       = "sb-compact-4b"
+  }, {
+  zone         = "us-east5-b"
+  node_count   = 64
+  machine_type = "ct5lp-hightpu-4t"
+  topology     = "16x16"
+  policy       = "sb-compact-4b"
+  }, {
+  zone         = "us-east5-b"
+  node_count   = 64
+  machine_type = "ct5lp-hightpu-4t"
+  topology     = "16x16"
+  policy       = "sb-compact-4c"
+  }, {
+  zone         = "us-east5-b"
+  node_count   = 64
+  machine_type = "ct5lp-hightpu-4t"
+  topology     = "16x16"
+  policy       = "sb-compact-4c"
+  }, {
+  zone         = "us-east5-b"
+  node_count   = 64
+  machine_type = "ct5lp-hightpu-4t"
+  topology     = "16x16"
+  policy       = "sb-compact-4c"
+  }, {
+  zone         = "us-east5-b"
+  node_count   = 64
+  machine_type = "ct5lp-hightpu-4t"
+  topology     = "16x16"
+  policy       = "sb-compact-4c"
+  }, {
+  zone         = "us-east5-b"
+  node_count   = 64
+  machine_type = "ct5lp-hightpu-4t"
+  topology     = "16x16"
+  policy       = "sb-compact-4d"
+  }, {
+  zone         = "us-east5-b"
+  node_count   = 64
+  machine_type = "ct5lp-hightpu-4t"
+  topology     = "16x16"
+  policy       = "sb-compact-4d"
+  }, {
+  zone         = "us-east5-b"
+  node_count   = 64
+  machine_type = "ct5lp-hightpu-4t"
+  topology     = "16x16"
+  policy       = "sb-compact-4d"
+  }, {
+  zone         = "us-east5-b"
+  node_count   = 64
+  machine_type = "ct5lp-hightpu-4t"
+  topology     = "16x16"
+  policy       = "sb-compact-4d"
+}]
 maintenance_interval = "PERIODIC"