From 8465bdc3eeaf07fff397d9579476eb8f4b17febf Mon Sep 17 00:00:00 2001
From: jzhaoqwa <jzhaoqwa@amazon.com>
Date: Tue, 1 Apr 2025 14:04:06 -0700
Subject: [PATCH] Suppor new instance type

**Description**
Suppor new instance type r7i, m7i and i3en

**Testing Done**
New instance is not officially released. No test needed on CLI side. Will be verified be HP team
---
 helm_chart/HyperPodHelmChart/values.yaml      | 25 +++++++++++++++++++
 .../constants/hyperpod_instance_types.py      | 25 +++++++++++++++++++
 2 files changed, 50 insertions(+)

diff --git a/helm_chart/HyperPodHelmChart/values.yaml b/helm_chart/HyperPodHelmChart/values.yaml
index 6589092c..98208306 100644
--- a/helm_chart/HyperPodHelmChart/values.yaml
+++ b/helm_chart/HyperPodHelmChart/values.yaml
@@ -179,11 +179,36 @@ aws-efa-k8s-device-plugin:
       - ml.g6e.24xlarge
       - ml.g6e.48xlarge
       - ml.gr6.8xlarge
+      - ml.i3en.large
+      - ml.i3en.xlarge
+      - ml.i3en.2xlarge
+      - ml.i3en.3xlarge
+      - ml.i3en.6xlarge
+      - ml.i3en.12xlarge
+      - ml.i3en.24xlarge
+      - ml.m7i.large
+      - ml.m7i.xlarge
+      - ml.m7i.2xlarge
+      - ml.m7i.4xlarge
+      - ml.m7i.8xlarge
+      - ml.m7i.12xlarge
+      - ml.m7i.16xlarge
+      - ml.m7i.24xlarge
+      - ml.m7i.48xlarge
       - ml.p4d.24xlarge
       - ml.p4de.24xlarge
       - ml.p5.48xlarge
       - ml.p5e.48xlarge
       - ml.p5en.48xlarge
+      - ml.r7i.large
+      - ml.r7i.xlarge
+      - ml.r7i.2xlarge
+      - ml.r7i.4xlarge
+      - ml.r7i.8xlarge
+      - ml.r7i.12xlarge
+      - ml.r7i.16xlarge
+      - ml.r7i.24xlarge
+      - ml.r7i.48xlarge
       - ml.trn1.32xlarge
       - ml.trn1n.32xlarge
       - ml.trn2.48xlarge
diff --git a/src/hyperpod_cli/constants/hyperpod_instance_types.py b/src/hyperpod_cli/constants/hyperpod_instance_types.py
index 3967f36a..7310081a 100644
--- a/src/hyperpod_cli/constants/hyperpod_instance_types.py
+++ b/src/hyperpod_cli/constants/hyperpod_instance_types.py
@@ -40,6 +40,13 @@ class HyperpodInstanceType(Enum):
     ML_C5N_4XLARGE = "ml.c5n.4xlarge"
     ML_C5N_9XLARGE = "ml.c5n.9xlarge"
     ML_C5N_18XLARGE = "ml.c5n.18xlarge"
+    ML_I3EN_LARGE = "ml.i3en.large"
+    ML_I3EN_XLARGE = "ml.i3en.xlarge"
+    ML_I3EN_2XLARGE = "ml.i3en.2xlarge"
+    ML_I3EN_3XLARGE = "ml.i3en.3xlarge"
+    ML_I3EN_6XLARGE = "ml.i3en.6xlarge"
+    ML_I3EN_12XLARGE = "ml.i3en.12xlarge"
+    ML_I3EN_24XLARGE = "ml.i3en.24xlarge"
     ML_M5_LARGE = "ml.m5.large"
     ML_M5_XLARGE = "ml.m5.xlarge"
     ML_M5_2XLARGE = "ml.m5.2xlarge"
@@ -48,6 +55,15 @@ class HyperpodInstanceType(Enum):
     ML_M5_12XLARGE = "ml.m5.12xlarge"
     ML_M5_16XLARGE = "ml.m5.16xlarge"
     ML_M5_24XLARGE = "ml.m5.24xlarge"
+    ML_M7I_LARGE = "ml.m7i.large"
+    ML_M7I_XLARGE = "ml.m7i.xlarge"
+    ML_M7I_2XLARGE = "ml.m7i.2xlarge"
+    ML_M7I_4XLARGE = "ml.m7i.4xlarge"
+    ML_M7I_8XLARGE = "ml.m7i.8xlarge"
+    ML_M7I_12XLARGE = "ml.m7i.12xlarge"
+    ML_M7I_16XLARGE = "ml.m7i.16xlarge"
+    ML_M7I_24XLARGE = "ml.m7i.24xlarge"
+    ML_M7I_48XLARGE = "ml.m7i.48xlarge"
     ML_T3_MEDIUM = "ml.t3.medium"
     ML_T3_LARGE = "ml.t3.large"
     ML_T3_XLARGE = "ml.t3.xlarge"
@@ -72,4 +88,13 @@ class HyperpodInstanceType(Enum):
     ML_G6E_48XLARGE = "ml.g6e.48xlarge"
     ML_P5E_48XLARGE = "ml.p5e.48xlarge"
     ML_P5EN_48XLARGE = "ml.p5en.48xlarge"
+    ML_R7I_LARGE = "ml.r7i.large"
+    ML_R7I_XLARGE = "ml.r7i.xlarge"
+    ML_R7I_2XLARGE = "ml.r7i.2xlarge"
+    ML_R7I_4XLARGE = "ml.r7i.4xlarge"
+    ML_R7I_8XLARGE = "ml.r7i.8xlarge"
+    ML_R7I_12XLARGE = "ml.r7i.12xlarge"
+    ML_R7I_16XLARGE = "ml.r7i.16xlarge"
+    ML_R7I_24XLARGE = "ml.r7i.24xlarge"
+    ML_R7I_48XLARGE = "ml.r7i.48xlarge"
     ML_TRN2_48XLARGE = "ml.trn2.48xlarge"