Add support for instance type ml.p6e-gb200.36xlarge

Author: zhaoqizqwang

helm_chart/HyperPodHelmChart/charts/health-monitoring-agent/templates/health-monitoring-agent.yaml

@@ -72,6 +72,7 @@ spec:
                   - key: node.kubernetes.io/instance-type
                     operator: In
                     values:
+                      - ml.p6e-gb200.36xlarge
                       - ml.p5en.48xlarge
                       - ml.p5e.48xlarge
                       - ml.p5.48xlarge

helm_chart/HyperPodHelmChart/values.yaml

@@ -172,6 +172,7 @@ nvidia-device-plugin:
               - ml.p5.48xlarge
               - ml.p5e.48xlarge
               - ml.p5en.48xlarge
+              - ml.p6e-gb200.36xlarge
   tolerations:
     - key: nvidia.com/gpu
       operator: Exists
@@ -229,6 +230,7 @@ aws-efa-k8s-device-plugin:
       - ml.p5.48xlarge
       - ml.p5e.48xlarge
       - ml.p5en.48xlarge
+      - ml.p6e-gb200.36xlarge
       - ml.r7i.large
       - ml.r7i.xlarge
       - ml.r7i.2xlarge

src/hyperpod_cli/constants/hyperpod_instance_types.py

@@ -82,6 +82,7 @@ class HyperpodInstanceType(Enum):
     ML_P5_48XLARGE = "ml.p5.48xlarge"
     ML_P5E_48XLARGE = "ml.p5e.48xlarge"
     ML_P5EN_48XLARGE = "ml.p5en.48xlarge"
+    ML_P6E_GB200_36XLARGE = "ml.p6e-gb200.36xlarge"
     ML_R7I_LARGE = "ml.r7i.large"
     ML_R7I_XLARGE = "ml.r7i.xlarge"
     ML_R7I_2XLARGE = "ml.r7i.2xlarge"