File tree 3 files changed +8
-20
lines changed
tools/kubernetes/terraform 3 files changed +8
-20
lines changed Original file line number Diff line number Diff line change @@ -7,31 +7,23 @@ tpu_node_pools = [{
7
7
machine_type = " ct5lp-hightpu-4t"
8
8
topology = " 16x16"
9
9
policy = " sb-compact-1"
10
- disk_type = " pd-balanced"
11
- disk_size_gb = 50
12
10
}, {
13
11
zone = " us-east5-b"
14
12
node_count = 64
15
13
machine_type = " ct5lp-hightpu-4t"
16
14
topology = " 16x16"
17
15
policy = " sb-compact-1"
18
- disk_type = " pd-balanced"
19
- disk_size_gb = 50
20
16
}, {
21
17
zone = " us-east5-b"
22
18
node_count = 64
23
19
machine_type = " ct5lp-hightpu-4t"
24
20
topology = " 16x16"
25
21
policy = " sb-compact-1"
26
- disk_type = " pd-balanced"
27
- disk_size_gb = 50
28
22
}, {
29
23
zone = " us-east5-b"
30
24
node_count = 64
31
25
machine_type = " ct5lp-hightpu-4t"
32
26
topology = " 16x16"
33
27
policy = " sb-compact-1"
34
- disk_type = " pd-balanced"
35
- disk_size_gb = 50
36
- }]
28
+ }]
37
29
maintenance_interval = " PERIODIC"
Original file line number Diff line number Diff line change @@ -58,7 +58,7 @@ resource "google_container_cluster" "tpu_cluster" {
58
58
release_channel {
59
59
channel = " UNSPECIFIED"
60
60
}
61
-
61
+
62
62
network = google_compute_network. vpc . name
63
63
subnetwork = google_compute_subnetwork. subnet . name
64
64
logging_service = " logging.googleapis.com/kubernetes"
@@ -81,7 +81,7 @@ resource "google_container_node_pool" "multihost_tpu" {
81
81
cluster = google_container_cluster. tpu_cluster . name
82
82
83
83
initial_node_count = var. tpu_node_pools [count . index ]. node_count
84
-
84
+
85
85
management {
86
86
auto_upgrade = false
87
87
}
@@ -104,18 +104,16 @@ resource "google_container_node_pool" "multihost_tpu" {
104
104
gcfs_config {
105
105
enabled = true
106
106
}
107
-
108
- image_type = " COS_CONTAINERD"
107
+
108
+ image_type = " COS_CONTAINERD"
109
109
machine_type = var. tpu_node_pools [count . index ]. machine_type
110
- disk_type = var. tpu_node_pools [count . index ]. disk_type
111
- disk_size_gb = var. tpu_node_pools [count . index ]. disk_size_gb
112
110
tags = [" gke-node" ]
113
111
metadata = {
114
112
disable-legacy-endpoints = " true"
115
113
}
116
114
}
117
115
placement_policy {
118
- type = " COMPACT"
119
- policy_name = var. tpu_node_pools [count . index ]. policy
116
+ type = " COMPACT"
117
+ policy_name = var. tpu_node_pools [count . index ]. policy
120
118
}
121
119
}
Original file line number Diff line number Diff line change @@ -35,12 +35,10 @@ variable "tpu_node_pools" {
35
35
machine_type = string ,
36
36
topology = string ,
37
37
policy = string ,
38
- disk_type = string ,
39
- disk_size_gb = number ,
40
38
}))
41
39
}
42
40
43
41
variable "maintenance_interval" {
44
- default = " AS_NEEDED"
42
+ default = " AS_NEEDED"
45
43
description = " maintenance interval for TPU machines."
46
44
}
You can’t perform that action at this time.
0 commit comments