Reservation Affinity for workbench instances (#13328) (#1055)

[upstream:ccc0df0c84d9cba5dbd53a7e02b1b63265f632d1]

Signed-off-by: Modular Magician <[email protected]>

Author: modular-magician

workbench_instance_basic_gpu/main.tf

@@ -1,3 +1,23 @@
+resource "google_compute_reservation" "gpu_reservation" {
+  name     = "wbi-reservation-${local.name_suffix}"
+  zone     = "us-central1-a"
+
+  specific_reservation {
+    count = 1
+    
+    instance_properties {
+      machine_type = "n1-standard-1"
+      
+      guest_accelerators {
+        accelerator_type  = "nvidia-tesla-t4"
+        accelerator_count = 1
+      }
+    }
+  }
+
+  specific_reservation_required = false
+}
+
 resource "google_workbench_instance" "instance" {
   name = "workbench-instance-${local.name_suffix}"
   location = "us-central1-a"
@@ -11,5 +31,13 @@ resource "google_workbench_instance" "instance" {
       project      = "cloud-notebooks-managed"
       family       = "workbench-instances"
     }
+    reservation_affinity {
+      consume_reservation_type = "RESERVATION_ANY"
+    }
   }
+
+  depends_on = [
+    google_compute_reservation.gpu_reservation
+  ]
+
 }