From 4c49566a892a5f1d61b9460ced049834d512a4cd Mon Sep 17 00:00:00 2001
From: Michael Vines <mvines@gmail.com>
Date: Sat, 21 Sep 2019 10:45:16 -0700
Subject: [PATCH] Enable nvidia persistence mode on instance reboots

---
 net/gce.sh | 22 ++++++++++++++--------
 1 file changed, 14 insertions(+), 8 deletions(-)

diff --git a/net/gce.sh b/net/gce.sh
index 5586ea6fc6..73fa8fab54 100755
--- a/net/gce.sh
+++ b/net/gce.sh
@@ -607,7 +607,14 @@ EOF
 set -ex
 
 if [[ -f /solana-scratch/.instance-startup-complete ]]; then
-  # Skip on instance reboot
+  $(
+    cd "$here"/scripts/
+    if "$enableGpu"; then
+      cat enable-nvidia-persistence-mode.sh
+    fi
+  )
+
+  # Skip most setup on instance reboot
   exit 0
 fi
 
@@ -656,14 +663,13 @@ $(
     network-config.sh \
     remove-docker-interface.sh \
 
-    if "$enableGpu"; then
-      cat enable-nvidia-persistence-mode.sh
-    fi
-
-    if [[ -n $fullNodeAdditionalDiskSizeInGb ]]; then
-      cat mount-additional-disk.sh
-    fi
+  if "$enableGpu"; then
+    cat enable-nvidia-persistence-mode.sh
+  fi
 
+  if [[ -n $fullNodeAdditionalDiskSizeInGb ]]; then
+    cat mount-additional-disk.sh
+  fi
 )
 
 cat > /etc/motd <<EOM