diff --git a/modules/gke-nodepool/main.tf b/modules/gke-nodepool/main.tf index f5a104bd..40f7fcf3 100644 --- a/modules/gke-nodepool/main.tf +++ b/modules/gke-nodepool/main.tf @@ -168,7 +168,10 @@ resource "google_container_node_pool" "nodepool" { gpu_partition_size = var.node_config.guest_accelerator.gpu_driver == null ? null : var.node_config.guest_accelerator.gpu_driver.partition_size dynamic "gpu_sharing_config" { - for_each = var.node_config.guest_accelerator.gpu_driver != null ? [""] : [] + for_each = lookup( + lookup(var.node_config.guest_accelerator, "gpu_driver", {}), + "max_shared_clients_per_gpu" + ) != null ? [""] : [] content { gpu_sharing_strategy = var.node_config.guest_accelerator.gpu_driver.max_shared_clients_per_gpu != null ? "TIME_SHARING" : null max_shared_clients_per_gpu = var.node_config.guest_accelerator.gpu_driver.max_shared_clients_per_gpu diff --git a/tests/modules/gke_nodepool/examples/guest-accelerator.yaml b/tests/modules/gke_nodepool/examples/guest-accelerator.yaml index 5f125ef6..8f505054 100644 --- a/tests/modules/gke_nodepool/examples/guest-accelerator.yaml +++ b/tests/modules/gke_nodepool/examples/guest-accelerator.yaml @@ -29,9 +29,7 @@ values: gpu_driver_installation_config: - gpu_driver_version: LATEST gpu_partition_size: null - gpu_sharing_config: - - gpu_sharing_strategy: null - max_shared_clients_per_gpu: null + gpu_sharing_config: null type: nvidia-tesla-a100 gvnic: [] machine_type: a2-highgpu-1g