From 722922cef06322a37334bda8fb2f19d34ca9a403 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Piotr=20Paw=C5=82owski?= Date: Fri, 20 Dec 2024 11:35:06 +0000 Subject: [PATCH] review fixes; remove setting disk_type MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Piotr Pawłowski --- src/xpk/blueprints/a3ultra/config-map.yaml.tftpl | 4 ++-- src/xpk/core/blueprint/blueprint_generator.py | 4 +--- 2 files changed, 3 insertions(+), 5 deletions(-) diff --git a/src/xpk/blueprints/a3ultra/config-map.yaml.tftpl b/src/xpk/blueprints/a3ultra/config-map.yaml.tftpl index 03f9307d..eeddce91 100644 --- 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}" \ No newline at end of file + 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 index 1d3a5636..47c910b5 100644 --- 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,