review fixes; remove setting disk_type

Signed-off-by: Piotr Pawłowski <ppawl@google.com>
AI-Hypercomputer · Dec 20, 2024 · 722922c · 722922c
1 parent 190a33c
commit 722922c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/src/xpk/blueprints/a3ultra/config-map.yaml.tftpl b/src/xpk/blueprints/a3ultra/config-map.yaml.tftpl
@@ -3,5 +3,5 @@ apiVersion: v1
 metadata:
   name: ${resource_config_name}
 data:
-  h100-mega-80gb-8: "${num_nodes}"
-  h200-141gb-8: "${num_nodes}"
+  h200-141gb-8: "${num_nodes}"
+  nvidia-h200-141gb: "${num_nodes}"
diff --git a/src/xpk/core/blueprint/blueprint_generator.py b/src/xpk/core/blueprint/blueprint_generator.py
@@ -508,16 +508,14 @@ def generate_a3_ultra_blueprint(
           f" {a3ultra_device_type} device_type."
       )
       xpk_exit(1)
-    gpu_pool_id = f"{cluster_name}-a3u-pool"
     gpu_pool = DeploymentModule(
-        id=gpu_pool_id,
+        id=f"{cluster_name}-a3u-pool",
         source="github.com/GoogleCloudPlatform/cluster-toolkit.git//modules/compute/gke-node-pool?ref=e0c690b",
         use=[cluster_id],
         settings={
             "machine_type": system.gce_machine_type,
             "auto_upgrade": True,
             "zones": [zone],
-            "disk_type": "hyperdisk-balanced",
             "static_node_count": static_node_count,
             "spot": spot,
             "max_pods_per_node": 32,