Skip to content

Commit 9432d5b

Browse files
umeshkumharumeshkumhar
committed
add support for gpu_driver_installation_config
1 parent 7c28e82 commit 9432d5b

File tree

8 files changed

+105
-0
lines changed

8 files changed

+105
-0
lines changed

‎autogen/main/cluster.tf.tmpl

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -846,6 +846,13 @@ resource "google_container_node_pool" "windows_pools" {
846846
type = lookup(each.value, "accelerator_type", "")
847847
count = lookup(each.value, "accelerator_count", 0)
848848
gpu_partition_size = lookup(each.value, "gpu_partition_size", null)
849+
850+
dynamic "gpu_driver_installation_config" {
851+
for_each = lookup(each.value, "gpu_driver_version", "") != "" ? [1] : []
852+
content {
853+
gpu_driver_version = lookup(each.value, "gpu_driver_version", "")
854+
}
855+
}
849856
}
850857
}
851858

‎cluster.tf

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -528,6 +528,13 @@ resource "google_container_node_pool" "pools" {
528528
type = lookup(each.value, "accelerator_type", "")
529529
count = lookup(each.value, "accelerator_count", 0)
530530
gpu_partition_size = lookup(each.value, "gpu_partition_size", null)
531+
532+
dynamic "gpu_driver_installation_config" {
533+
for_each = lookup(each.value, "gpu_driver_version", "") != "" ? [1] : []
534+
content {
535+
gpu_driver_version = lookup(each.value, "gpu_driver_version", "")
536+
}
537+
}
531538
}
532539
}
533540

@@ -716,6 +723,13 @@ resource "google_container_node_pool" "windows_pools" {
716723
type = lookup(each.value, "accelerator_type", "")
717724
count = lookup(each.value, "accelerator_count", 0)
718725
gpu_partition_size = lookup(each.value, "gpu_partition_size", null)
726+
727+
dynamic "gpu_driver_installation_config" {
728+
for_each = lookup(each.value, "gpu_driver_version", "") != "" ? [1] : []
729+
content {
730+
gpu_driver_version = lookup(each.value, "gpu_driver_version", "")
731+
}
732+
}
719733
}
720734
}
721735

‎modules/beta-private-cluster-update-variant/cluster.tf

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -734,6 +734,13 @@ resource "google_container_node_pool" "pools" {
734734
type = lookup(each.value, "accelerator_type", "")
735735
count = lookup(each.value, "accelerator_count", 0)
736736
gpu_partition_size = lookup(each.value, "gpu_partition_size", null)
737+
738+
dynamic "gpu_driver_installation_config" {
739+
for_each = lookup(each.value, "gpu_driver_version", "") != "" ? [1] : []
740+
content {
741+
gpu_driver_version = lookup(each.value, "gpu_driver_version", "")
742+
}
743+
}
737744
}
738745
}
739746

@@ -962,6 +969,13 @@ resource "google_container_node_pool" "windows_pools" {
962969
type = lookup(each.value, "accelerator_type", "")
963970
count = lookup(each.value, "accelerator_count", 0)
964971
gpu_partition_size = lookup(each.value, "gpu_partition_size", null)
972+
973+
dynamic "gpu_driver_installation_config" {
974+
for_each = lookup(each.value, "gpu_driver_version", "") != "" ? [1] : []
975+
content {
976+
gpu_driver_version = lookup(each.value, "gpu_driver_version", "")
977+
}
978+
}
965979
}
966980
}
967981

‎modules/beta-private-cluster/cluster.tf

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -640,6 +640,13 @@ resource "google_container_node_pool" "pools" {
640640
type = lookup(each.value, "accelerator_type", "")
641641
count = lookup(each.value, "accelerator_count", 0)
642642
gpu_partition_size = lookup(each.value, "gpu_partition_size", null)
643+
644+
dynamic "gpu_driver_installation_config" {
645+
for_each = lookup(each.value, "gpu_driver_version", "") != "" ? [1] : []
646+
content {
647+
gpu_driver_version = lookup(each.value, "gpu_driver_version", "")
648+
}
649+
}
643650
}
644651
}
645652

@@ -867,6 +874,13 @@ resource "google_container_node_pool" "windows_pools" {
867874
type = lookup(each.value, "accelerator_type", "")
868875
count = lookup(each.value, "accelerator_count", 0)
869876
gpu_partition_size = lookup(each.value, "gpu_partition_size", null)
877+
878+
dynamic "gpu_driver_installation_config" {
879+
for_each = lookup(each.value, "gpu_driver_version", "") != "" ? [1] : []
880+
content {
881+
gpu_driver_version = lookup(each.value, "gpu_driver_version", "")
882+
}
883+
}
870884
}
871885
}
872886

‎modules/beta-public-cluster-update-variant/cluster.tf

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -715,6 +715,13 @@ resource "google_container_node_pool" "pools" {
715715
type = lookup(each.value, "accelerator_type", "")
716716
count = lookup(each.value, "accelerator_count", 0)
717717
gpu_partition_size = lookup(each.value, "gpu_partition_size", null)
718+
719+
dynamic "gpu_driver_installation_config" {
720+
for_each = lookup(each.value, "gpu_driver_version", "") != "" ? [1] : []
721+
content {
722+
gpu_driver_version = lookup(each.value, "gpu_driver_version", "")
723+
}
724+
}
718725
}
719726
}
720727

@@ -943,6 +950,13 @@ resource "google_container_node_pool" "windows_pools" {
943950
type = lookup(each.value, "accelerator_type", "")
944951
count = lookup(each.value, "accelerator_count", 0)
945952
gpu_partition_size = lookup(each.value, "gpu_partition_size", null)
953+
954+
dynamic "gpu_driver_installation_config" {
955+
for_each = lookup(each.value, "gpu_driver_version", "") != "" ? [1] : []
956+
content {
957+
gpu_driver_version = lookup(each.value, "gpu_driver_version", "")
958+
}
959+
}
946960
}
947961
}
948962

‎modules/beta-public-cluster/cluster.tf

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -621,6 +621,13 @@ resource "google_container_node_pool" "pools" {
621621
type = lookup(each.value, "accelerator_type", "")
622622
count = lookup(each.value, "accelerator_count", 0)
623623
gpu_partition_size = lookup(each.value, "gpu_partition_size", null)
624+
625+
dynamic "gpu_driver_installation_config" {
626+
for_each = lookup(each.value, "gpu_driver_version", "") != "" ? [1] : []
627+
content {
628+
gpu_driver_version = lookup(each.value, "gpu_driver_version", "")
629+
}
630+
}
624631
}
625632
}
626633

@@ -848,6 +855,13 @@ resource "google_container_node_pool" "windows_pools" {
848855
type = lookup(each.value, "accelerator_type", "")
849856
count = lookup(each.value, "accelerator_count", 0)
850857
gpu_partition_size = lookup(each.value, "gpu_partition_size", null)
858+
859+
dynamic "gpu_driver_installation_config" {
860+
for_each = lookup(each.value, "gpu_driver_version", "") != "" ? [1] : []
861+
content {
862+
gpu_driver_version = lookup(each.value, "gpu_driver_version", "")
863+
}
864+
}
851865
}
852866
}
853867

‎modules/private-cluster-update-variant/cluster.tf

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -641,6 +641,13 @@ resource "google_container_node_pool" "pools" {
641641
type = lookup(each.value, "accelerator_type", "")
642642
count = lookup(each.value, "accelerator_count", 0)
643643
gpu_partition_size = lookup(each.value, "gpu_partition_size", null)
644+
645+
dynamic "gpu_driver_installation_config" {
646+
for_each = lookup(each.value, "gpu_driver_version", "") != "" ? [1] : []
647+
content {
648+
gpu_driver_version = lookup(each.value, "gpu_driver_version", "")
649+
}
650+
}
644651
}
645652
}
646653

@@ -830,6 +837,13 @@ resource "google_container_node_pool" "windows_pools" {
830837
type = lookup(each.value, "accelerator_type", "")
831838
count = lookup(each.value, "accelerator_count", 0)
832839
gpu_partition_size = lookup(each.value, "gpu_partition_size", null)
840+
841+
dynamic "gpu_driver_installation_config" {
842+
for_each = lookup(each.value, "gpu_driver_version", "") != "" ? [1] : []
843+
content {
844+
gpu_driver_version = lookup(each.value, "gpu_driver_version", "")
845+
}
846+
}
833847
}
834848
}
835849

‎modules/private-cluster/cluster.tf

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -547,6 +547,13 @@ resource "google_container_node_pool" "pools" {
547547
type = lookup(each.value, "accelerator_type", "")
548548
count = lookup(each.value, "accelerator_count", 0)
549549
gpu_partition_size = lookup(each.value, "gpu_partition_size", null)
550+
551+
dynamic "gpu_driver_installation_config" {
552+
for_each = lookup(each.value, "gpu_driver_version", "") != "" ? [1] : []
553+
content {
554+
gpu_driver_version = lookup(each.value, "gpu_driver_version", "")
555+
}
556+
}
550557
}
551558
}
552559

@@ -735,6 +742,13 @@ resource "google_container_node_pool" "windows_pools" {
735742
type = lookup(each.value, "accelerator_type", "")
736743
count = lookup(each.value, "accelerator_count", 0)
737744
gpu_partition_size = lookup(each.value, "gpu_partition_size", null)
745+
746+
dynamic "gpu_driver_installation_config" {
747+
for_each = lookup(each.value, "gpu_driver_version", "") != "" ? [1] : []
748+
content {
749+
gpu_driver_version = lookup(each.value, "gpu_driver_version", "")
750+
}
751+
}
738752
}
739753
}
740754

0 commit comments

Comments
 (0)