Merge pull request GoogleCloudPlatform#4818 from arpit974/A4xCoresAndSocketNumberUpdate

arpit974 · web-flow · commit 42c1f9c553ea · 2025-11-04T17:03:54.000+05:30
similar to other A* blueprints specifying correct node configurations in the nodesset.
diff --git a/examples/machine-learning/a4x-highgpu-4g/a4xhigh-slurm-blueprint.yaml b/examples/machine-learning/a4x-highgpu-4g/a4xhigh-slurm-blueprint.yaml
@@ -597,6 +597,12 @@ deployment_groups:
       instance_image_custom: true
       on_host_maintenance: TERMINATE
       reservation_name: $(vars.a4x_reservation_name)
+      advanced_machine_features:
+        threads_per_core: null # Use platform default value
+      node_conf:
+        CoresPerSocket: 70
+        SocketsPerBoard: 2
+        ThreadsPerCore: 1
       additional_networks:
         $(concat(
           [{
