Add support for guest accelerators in GKE node pool

kbst · Dec 6, 2024 · 89a5ece · 89a5ece
1 parent a77da45
commit 89a5ece
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 1 deletion.
diff --git a/google/cluster/node-pool/configuration.tf b/google/cluster/node-pool/configuration.tf
@@ -48,4 +48,6 @@ locals {
   instance_tags = local.cfg["instance_tags"]
 
   ephemeral_storage_local_ssd_config = local.cfg["ephemeral_storage_local_ssd_config"]
+
+  guest_accelerator = local.cfg["guest_accelerator"]
 }
diff --git a/google/cluster/node-pool/main.tf b/google/cluster/node-pool/main.tf
@@ -45,4 +45,6 @@ module "node_pool" {
   network_config = local.network_config
 
   ephemeral_storage_local_ssd_config = local.ephemeral_storage_local_ssd_config
+
+  guest_accelerator = local.guest_accelerator
 }
diff --git a/google/cluster/node-pool/variables.tf b/google/cluster/node-pool/variables.tf
@@ -26,7 +26,7 @@ variable "configuration" {
       value  = string
       effect = string
     })))
-    
+
     labels = optional(map(string))
 
     labels = optional(map(string))
@@ -62,6 +62,16 @@ variable "configuration" {
     ephemeral_storage_local_ssd_config = optional(object({
       local_ssd_count = number
     }))
+
+    guest_accelerator = optional(object({
+      type               = string
+      count              = number
+      gpu_partition_size = optional(string)
+      gpu_sharing_config = optional(object({
+        gpu_sharing_strategy       = optional(string)
+        max_shared_clients_per_gpu = optional(number)
+      }))
+    }))
   }))
 
   description = "Map with per workspace cluster configuration."