From c6b66ab93272b5065fb8ad6ad70089553c9cf1b9 Mon Sep 17 00:00:00 2001
From: shiva kumar <shivaku@nvidia.com>
Date: Sun, 25 Aug 2024 15:17:47 +0530
Subject: [PATCH] 4.Pre-compiled end-to-end  gpu driver validation

---
 tests/scripts/nvidia-kernel-upgrade-aws.sh | 8 +-------
 1 file changed, 1 insertion(+), 7 deletions(-)

diff --git a/tests/scripts/nvidia-kernel-upgrade-aws.sh b/tests/scripts/nvidia-kernel-upgrade-aws.sh
index b8a4076d..4c49057b 100755
--- a/tests/scripts/nvidia-kernel-upgrade-aws.sh
+++ b/tests/scripts/nvidia-kernel-upgrade-aws.sh
@@ -61,13 +61,7 @@ if [ "${CURRENT_KERNEL}" != ${KERNEL_VERSION} ]; then
     }
     ' "$GRUB_FILE")
   sudo sed -i "s/^GRUB_DEFAULT=.*/GRUB_DEFAULT=\"1>${kernel_index}\"/" /etc/default/grub
-
-  echo "SHIVA"
-  cat ${GRUB_FILE}
-  echo "==================="
-  cat  /etc/default/grub
-
-  echo "SHIVA1"
+  sudo update-grub
 
   echo "Rebooting ..."
   # Run the reboot command with nohup to avoid abrupt SSH closure issues