Skip to content

Commit f43a241

Browse files
umeshkumharapeabody
umeshkumhar
and
apeabody
authored
feat: add support for gpu_driver_installation_config on nodepool (#1767)
Co-authored-by: Andrew Peabody <[email protected]>
1 parent 3181f6c commit f43a241

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
@@ -830,6 +830,13 @@ resource "google_container_node_pool" "windows_pools" {
830830
type = lookup(each.value, "accelerator_type", "")
831831
count = lookup(each.value, "accelerator_count", 0)
832832
gpu_partition_size = lookup(each.value, "gpu_partition_size", null)
833+
834+
dynamic "gpu_driver_installation_config" {
835+
for_each = lookup(each.value, "gpu_driver_version", "") != "" ? [1] : []
836+
content {
837+
gpu_driver_version = lookup(each.value, "gpu_driver_version", "")
838+
}
839+
}
833840
}
834841
}
835842

‎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
@@ -716,6 +716,13 @@ resource "google_container_node_pool" "pools" {
716716
type = lookup(each.value, "accelerator_type", "")
717717
count = lookup(each.value, "accelerator_count", 0)
718718
gpu_partition_size = lookup(each.value, "gpu_partition_size", null)
719+
720+
dynamic "gpu_driver_installation_config" {
721+
for_each = lookup(each.value, "gpu_driver_version", "") != "" ? [1] : []
722+
content {
723+
gpu_driver_version = lookup(each.value, "gpu_driver_version", "")
724+
}
725+
}
719726
}
720727
}
721728

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

‎modules/beta-private-cluster/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

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

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

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -697,6 +697,13 @@ resource "google_container_node_pool" "pools" {
697697
type = lookup(each.value, "accelerator_type", "")
698698
count = lookup(each.value, "accelerator_count", 0)
699699
gpu_partition_size = lookup(each.value, "gpu_partition_size", null)
700+
701+
dynamic "gpu_driver_installation_config" {
702+
for_each = lookup(each.value, "gpu_driver_version", "") != "" ? [1] : []
703+
content {
704+
gpu_driver_version = lookup(each.value, "gpu_driver_version", "")
705+
}
706+
}
700707
}
701708
}
702709

@@ -925,6 +932,13 @@ resource "google_container_node_pool" "windows_pools" {
925932
type = lookup(each.value, "accelerator_type", "")
926933
count = lookup(each.value, "accelerator_count", 0)
927934
gpu_partition_size = lookup(each.value, "gpu_partition_size", null)
935+
936+
dynamic "gpu_driver_installation_config" {
937+
for_each = lookup(each.value, "gpu_driver_version", "") != "" ? [1] : []
938+
content {
939+
gpu_driver_version = lookup(each.value, "gpu_driver_version", "")
940+
}
941+
}
928942
}
929943
}
930944

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

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

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

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

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

@@ -811,6 +818,13 @@ resource "google_container_node_pool" "windows_pools" {
811818
type = lookup(each.value, "accelerator_type", "")
812819
count = lookup(each.value, "accelerator_count", 0)
813820
gpu_partition_size = lookup(each.value, "gpu_partition_size", null)
821+
822+
dynamic "gpu_driver_installation_config" {
823+
for_each = lookup(each.value, "gpu_driver_version", "") != "" ? [1] : []
824+
content {
825+
gpu_driver_version = lookup(each.value, "gpu_driver_version", "")
826+
}
827+
}
814828
}
815829
}
816830

‎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)