Skip to content

Commit 273b053

Browse files
committed
Add deep health checkc status and ability to remove deep health checks from config
1 parent a45557b commit 273b053

2 files changed

Lines changed: 53 additions & 2 deletions

File tree

Container-Root/hyperpod/ops/eks/nodes-list.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,9 +20,9 @@ else
2020
shift
2121

2222
if [ "$node" == "" ]; then
23-
CMD="kubectl get nodes -L node.kubernetes.io/instance-type -L sagemaker.amazonaws.com/node-health-status $@"
23+
CMD="kubectl get nodes -L node.kubernetes.io/instance-type -L sagemaker.amazonaws.com/node-health-status -L sagemaker.amazonaws.com/deep-health-check-status $@"
2424
else
25-
CMD="kubectl get nodes -L node.kubernetes.io/instance-type -L sagemaker.amazonaws.com/node-health-status $@ | grep -E \"NODE-HEALTH-STATUS|$node\""
25+
CMD="kubectl get nodes -L node.kubernetes.io/instance-type -L sagemaker.amazonaws.com/node-health-status -L sagemaker.amazonaws.com/deep-health-check-status $@ | grep -E \"NODE-HEALTH-STATUS|$node\""
2626
fi
2727

2828
if [ "${VERBOSE}" == "true" ]; then

wd/conf/eks/aws-do-hyperpod-eks/generate-update-config.sh

Lines changed: 51 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,55 @@
33
source ./env_input
44
source ./env_vars
55

6+
if [ "$ONSTART_DEEP_HEALTHCHECKS" == "" ]; then
7+
8+
cat > hyperpod-update-config.json << EOL
9+
{
10+
"ClusterName": "${HYPERPOD_NAME}",
11+
"InstanceGroups": [
12+
{
13+
"InstanceGroupName": "worker-group-1",
14+
"InstanceType": "${ACCEL_INSTANCE_TYPE}",
15+
"InstanceCount": ${ACCEL_COUNT},
16+
"InstanceStorageConfigs": [
17+
{
18+
"EbsVolumeConfig": {
19+
"VolumeSizeInGB": ${ACCEL_VOLUME_SIZE}
20+
}
21+
}
22+
],
23+
"LifeCycleConfig": {
24+
"SourceS3Uri": "s3://${BUCKET_NAME}",
25+
"OnCreate": "on_create.sh"
26+
},
27+
"ExecutionRole": "${EXECUTION_ROLE}",
28+
"ThreadsPerCore": 1
29+
},
30+
{
31+
"InstanceGroupName": "worker-group-2",
32+
"InstanceType": "${GEN_INSTANCE_TYPE}",
33+
"InstanceCount": ${GEN_COUNT},
34+
"InstanceStorageConfigs": [
35+
{
36+
"EbsVolumeConfig": {
37+
"VolumeSizeInGB": ${GEN_VOLUME_SIZE}
38+
}
39+
}
40+
],
41+
"LifeCycleConfig": {
42+
"SourceS3Uri": "s3://${BUCKET_NAME}",
43+
"OnCreate": "on_create.sh"
44+
},
45+
"ExecutionRole": "${EXECUTION_ROLE}",
46+
"ThreadsPerCore": 1
47+
}
48+
],
49+
"NodeRecovery": "${NODE_RECOVERY}"
50+
}
51+
EOL
52+
53+
else
54+
655
cat > hyperpod-update-config.json << EOL
756
{
857
"ClusterName": "${HYPERPOD_NAME}",
@@ -48,3 +97,5 @@ cat > hyperpod-update-config.json << EOL
4897
"NodeRecovery": "${NODE_RECOVERY}"
4998
}
5099
EOL
100+
101+
fi

0 commit comments

Comments
 (0)