From ca09bb810139a0a98ed5b9211aa404cf36b182ca Mon Sep 17 00:00:00 2001
From: Jingxin Ye <yejingxin@google.com>
Date: Mon, 28 Aug 2023 22:08:25 -0700
Subject: [PATCH] Update Terraform script for enabling image streaming and
 disabling auto upgrade

---
 tools/kubernetes/terraform/module/main.tf | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/tools/kubernetes/terraform/module/main.tf b/tools/kubernetes/terraform/module/main.tf
index 38b127b3..b392253d 100644
--- a/tools/kubernetes/terraform/module/main.tf
+++ b/tools/kubernetes/terraform/module/main.tf
@@ -76,6 +76,10 @@ resource "google_container_node_pool" "multihost_tpu" {
   initial_node_count = var.tpu_node_pools[count.index].node_count
 
   node_config {
+    management {
+      auto_upgrade = false
+    }
+
     oauth_scopes = [
       "https://www.googleapis.com/auth/logging.write",
       "https://www.googleapis.com/auth/monitoring",
@@ -89,6 +93,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"]
@@ -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
+}