Add support for guest accelerators in GKE node pool

Author: rzk

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"]
 }

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
 }

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."