From 6ed33df8aa0cad2708d99ca1e42152c1508bdae6 Mon Sep 17 00:00:00 2001
From: Yuwei Yang <yangyuwei@google.com>
Date: Mon, 28 Aug 2023 19:44:01 -0700
Subject: [PATCH] Add Terraform config for enabling image streaming and
 disabling auto upgrade.

---
 tools/kubernetes/terraform/module/main.tf | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/tools/kubernetes/terraform/module/main.tf b/tools/kubernetes/terraform/module/main.tf
index 38b127b3..c5869c2f 100644
--- a/tools/kubernetes/terraform/module/main.tf
+++ b/tools/kubernetes/terraform/module/main.tf
@@ -52,6 +52,7 @@ resource "google_container_cluster" "tpu_cluster" {
   release_channel {
     channel = "UNSPECIFIED"
   }
+  
   network            = google_compute_network.vpc.name
   subnetwork         = google_compute_subnetwork.subnet.name
   logging_service    = "logging.googleapis.com/kubernetes"
@@ -74,6 +75,10 @@ resource "google_container_node_pool" "multihost_tpu" {
   cluster        = google_container_cluster.tpu_cluster.name
 
   initial_node_count = var.tpu_node_pools[count.index].node_count
+  
+  management {
+    auto_upgrade = false
+  }
 
   node_config {
     oauth_scopes = [
@@ -89,7 +94,11 @@ resource "google_container_node_pool" "multihost_tpu" {
     gvnic {
       enabled = true
     }
-
+    gcfs_config {
+      enabled = true
+    }
+   
+    image_type = "COS_CONTAINERD"
     machine_type = var.tpu_node_pools[count.index].machine_type
     tags         = ["gke-node"]
     metadata = {
@@ -100,4 +109,4 @@ resource "google_container_node_pool" "multihost_tpu" {
     type         = "COMPACT"
     tpu_topology = var.tpu_node_pools[count.index].topology
   }
-}
\ No newline at end of file
+}