add reservations for kueue integration tests

GoogleCloudPlatform · Dec 19, 2024 · 9e39b30 · 9e39b30
1 parent 0b570bc
commit 9e39b30
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/tools/cloud-build/daily-tests/blueprints/gke-a2-highgpu.yaml b/tools/cloud-build/daily-tests/blueprints/gke-a2-highgpu.yaml
@@ -20,7 +20,7 @@ vars:
   project_id: hpc-toolkit-dev ## Set GCP Project ID Here ##
   deployment_name: gke-a2-highgpu
   region: us-central1
-  zone: us-central1-c
+  zone: us-central1-f
 
   # Cidr block containing the IP of the machine calling terraform.
   # The following line must be updated for this example to work.
@@ -84,6 +84,7 @@ deployment_groups:
       zones: [$(vars.zone)]
       image_type: UBUNTU_CONTAINERD
       placement_policy:
+        name: a2-highgpu-compact
         type: "COMPACT"
     outputs: [instructions]
 

diff --git a/tools/cloud-build/daily-tests/tests/gke-a2-highgpu-kueue.yml b/tools/cloud-build/daily-tests/tests/gke-a2-highgpu-kueue.yml
@@ -22,13 +22,19 @@ workspace: /workspace
 blueprint_yaml: "{{ workspace }}/tools/cloud-build/daily-tests/blueprints/gke-a2-highgpu.yaml"
 network: "gke-a2high-net-{{ build }}"
 region: us-central1
-zone: us-central1-c
+zone: us-central1-f
 remote_node: "{{ deployment_name }}-remote-node-0"
+reservation_affinity:
+  consume_reservation_type: SPECIFIC_RESERVATION
+  specific_reservations:
+  - name: a2-reservation-0
+    project: "{{ project }}"
 cli_deployment_vars:
   region: "{{ region }}"
   zone: "{{ zone }}"
   network_name: "{{ network }}"
-  local_ssd_count_nvme_block: 8
+  reservation_affinity: "{{ reservation_affinity }}"
+  local_ssd_count_nvme_block: 2
 custom_vars:
   project: "{{ project }}"
 post_deploy_tests: