Merge pull request GoogleCloudPlatform#4640 from abbas1902/hotfix/h4d-blueprint

Hotfix: H4D Blueprint provisioning model option update

Author: abbas1902

community/modules/scheduler/schedmd-slurm-gcp-v6-controller/modules/slurm_files/scripts/tests/test_util.py

@@ -423,7 +423,7 @@ def test_MachineType_from_json(jo: dict, want: MachineType):
                                     name="e2",
                                     guest_cpus=12,
                                     memory_mb=87040,
-                                    accelerators=None),
+                                    accelerators=[]),
             }),
             None
         ),
@@ -456,7 +456,7 @@ def test_MachineType_from_json(jo: dict, want: MachineType):
                                     name="a2-highgpu-1g",
                                     guest_cpus=12,
                                     memory_mb=87040,
-                                    accelerators=None),
+                                    accelerators=[]),
                 "guestAccelerators":[ { "acceleratorCount": 1, "acceleratorType": "nvidia-tesla-a100" } ],
             }),
             AcceleratorInfo(type="nvidia-tesla-a100", count=1)

examples/hpc-slurm-h4d.yaml

@@ -20,6 +20,10 @@ vars:
   zone: us-central1-a
   rdma_net_range: 192.168.128.0/18
 
+  #Provisioning model, select one, lack of selection will rely on on-demand capacity
+  h4d_dws_flex_enabled: false
+  h4d_reservation_name: ""
+
 # Documentation for each of the modules used below can be found at
 # https://github.com/GoogleCloudPlatform/hpc-toolkit/blob/main/modules/README.md
 deployment_groups:
@@ -84,6 +88,12 @@ deployment_groups:
       node_count_static: 2
       node_count_dynamic_max: 0
       enable_placement: false
+
+      #Provisioning models
+      reservation_name: $(vars.h4d_reservation_name)
+      dws_flex:
+        enabled: $(vars.h4d_dws_flex_enabled)
+
       disk_type: hyperdisk-balanced
       on_host_maintenance: TERMINATE
       additional_networks: