Add hyperpod-patch.sh

Author: iankouls-aws

.env

@@ -59,7 +59,7 @@ export MOD=""
 export MOD_TAG=$(if [ "$MOD" == "" ]; then echo ""; else echo "-${MOD}"; fi)
 ## VERSION: [optional] - Version tag for this Docker image. Example: v20180302
 #export VERSION=v$(date +%Y%m%d)
-export VERSION=v20240912
+export VERSION=v20241025
 export TAG=$(if [ -z "${VERSION}" ]; then echo ":latest${MOD_TAG}"; else echo ":${VERSION}${MOD_TAG}"; fi) 
 ## BUILD_OPTS: [optional] - arguments for the docker image build command
 export BUILD_OPTS="--progress plain --build-arg http_proxy=${http_proxy} --build-arg https_proxy=${https_proxy} --build-arg no_proxy=${no_proxy} --build-arg MOD=${MOD}"

Container-Root/hyperpod/hyperpod-patch.sh

@@ -0,0 +1,13 @@
+#!/bin/bash
+
+source conf/env.conf
+source ${CONF}/env_input
+source ${CONF}/env_vars
+
+if [ ! "$1" == "" ]; then export HYPERPOD_NAME=$1; fi
+
+echo ""
+echo "Patching HyperPod $IMPL cluster ${HYPERPOD_NAME}: "
+
+impl/${IMPL}/hyperpod-patch.sh ${HYPERPOD_NAME}
+

Container-Root/hyperpod/impl/eks/hyperpod-patch.sh

@@ -0,0 +1,28 @@
+#!/bin/bash
+
+# Patch HyperPod Cluster
+if [ ! "$1" == "" ]; then HYPERPOD_NAME=$1; fi
+
+echo ""
+echo "Patching HyperPod cluster: $HYPERPOD_NAME"
+echo ""
+read -r -p "Are you sure? [y/N] " response
+case "$response" in 
+    [yY][eE][sS]|[yY])
+        ./hyperpod-status.sh ${HYPERPOD_NAME}
+        if [ "$?" == "0" ]; then
+            CMD="aws sagemaker update-cluster-software --cluster-name ${HYPERPOD_NAME} ${ENDPOINT_ARG}"
+            if [ ! "$VERBOSE" == "false" ]; then echo -e "\n${CMD}\n"; fi
+            eval "$CMD"
+        else
+            echo "HyperPod cluster $HYPERPOD_NAME does not exist"
+        fi
+        ;;
+    *)
+        echo ""
+        echo "Cluster patching cancelled"
+        echo ""
+        ;;
+esac
+
+