fix(google_gke): Add variable for GPU driver version

google_gke/README.md

@@ -183,7 +183,7 @@ module "gke" {
 | <a name="input_network"></a> [network](#input\_network) | Shared VPC Network (formulated as a URL) wherein the cluster will be created. Overidden by shared\_vpc\_outputs. | `string` | `null` | no |
 | <a name="input_node_pool_sa_roles"></a> [node\_pool\_sa\_roles](#input\_node\_pool\_sa\_roles) | n/a | `list` | <pre>[<br/>  "roles/logging.logWriter",<br/>  "roles/monitoring.metricWriter",<br/>  "roles/monitoring.viewer",<br/>  "roles/stackdriver.resourceMetadata.writer"<br/>]</pre> | no |
 | <a name="input_node_pools"></a> [node\_pools](#input\_node\_pools) | Map containing node pools, with each node pool's name (or name\_prefix if `use_name_prefix` is true) being the key and the values being that node pool's configurations. Configurable options per node pool include: `disk_size_gb` (string), `disk_type` (string), `machine_type` (string), `max_count` (number), `max_surge` (number), `max_unavailable` (number), `min_count` (number), `use_name_prefix` (bool). See locals.tf for defaults. | `list(map(string))` | <pre>[<br/>  {<br/>    "name": "tf-default-node-pool"<br/>  }<br/>]</pre> | no |
-| <a name="input_node_pools_guest_accelerator"></a> [node\_pools\_guest\_accelerator](#input\_node\_pools\_guest\_accelerator) | Map containing node pools guest accelerator. Each node pool's name is the key. See locals.tf for defaults. | `map(map(string))` | <pre>{<br/>  "tf-default-node-pool": {}<br/>}</pre> | no |
+| <a name="input_node_pools_guest_accelerator"></a> [node\_pools\_guest\_accelerator](#input\_node\_pools\_guest\_accelerator) | Map containing node pools guest accelerator. Each node pool's name is the key. See locals.tf for defaults. | <pre>map(object({<br/>    type               = string<br/>    count              = number<br/>    gpu_driver_version = optional(string, "LATEST")<br/>  }))</pre> | `{}` | no |
 | <a name="input_node_pools_labels"></a> [node\_pools\_labels](#input\_node\_pools\_labels) | Map containing node pools non-default labels (as a map of strings). Each key is used as node pool's name prefix. See locals.tf for defaults. | `map(map(string))` | <pre>{<br/>  "tf-default-node-pool": {}<br/>}</pre> | no |
 | <a name="input_node_pools_oauth_scopes"></a> [node\_pools\_oauth\_scopes](#input\_node\_pools\_oauth\_scopes) | Map containing node pools non-default OAuth scopes (as an list). Each node pool's name is the key. See locals.tf for defaults. | `map(list(string))` | <pre>{<br/>  "tf-default-node-pool": []<br/>}</pre> | no |
 | <a name="input_node_pools_spot_enabled"></a> [node\_pools\_spot\_enabled](#input\_node\_pools\_spot\_enabled) | Map containing node pools spot enabled. Each node pool's name is the key. See locals.tf for defaults. | `map(bool)` | <pre>{<br/>  "tf-default-node-pool": false<br/>}</pre> | no |

google_gke/cluster.tf

@@ -288,6 +288,10 @@ resource "google_container_node_pool" "pools" {
       content {
         type  = local.node_pools_guest_accelerator[each.key].type
         count = local.node_pools_guest_accelerator[each.key].count
+
+        gpu_driver_installation_config {
+          gpu_driver_version = local.node_pools_guest_accelerator[each.key].gpu_driver_version
+        }
       }
     }
 

google_gke/variables.tf

@@ -267,10 +267,12 @@ variable "node_pools_taints" {
 
 variable "node_pools_guest_accelerator" {
   description = "Map containing node pools guest accelerator. Each node pool's name is the key. See locals.tf for defaults."
-  type        = map(map(string))
-  default = {
-    tf-default-node-pool = {}
-  }
+  type = map(object({
+    type               = string
+    count              = number
+    gpu_driver_version = optional(string, "LATEST")
+  }))
+  default = {}
 }
 
 variable "node_pools_spot_enabled" {