From 9e39b30398cf45fb69d48586ad8802e3a5ec8c36 Mon Sep 17 00:00:00 2001
From: ighosh98 <indraneelghosh@google.com>
Date: Wed, 18 Dec 2024 06:18:04 +0000
Subject: [PATCH] add reservations for kueue integration tests

---
 .../daily-tests/blueprints/gke-a2-highgpu.yaml         |  3 ++-
 .../daily-tests/tests/gke-a2-highgpu-kueue.yml         | 10 ++++++++--
 2 files changed, 10 insertions(+), 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
index 726768ffa8..b715448e65 100644
--- 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
index 549fbac367..0735f4f970 100644
--- 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: