Skip to content

Commit d1b88a0

Browse files
authored
Suppor new instance type (#75)
**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
1 parent 16734e0 commit d1b88a0

File tree

2 files changed

+50
-0
lines changed

2 files changed

+50
-0
lines changed

helm_chart/HyperPodHelmChart/values.yaml

Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -179,11 +179,36 @@ aws-efa-k8s-device-plugin:
179179
- ml.g6e.24xlarge
180180
- ml.g6e.48xlarge
181181
- ml.gr6.8xlarge
182+
- ml.i3en.large
183+
- ml.i3en.xlarge
184+
- ml.i3en.2xlarge
185+
- ml.i3en.3xlarge
186+
- ml.i3en.6xlarge
187+
- ml.i3en.12xlarge
188+
- ml.i3en.24xlarge
189+
- ml.m7i.large
190+
- ml.m7i.xlarge
191+
- ml.m7i.2xlarge
192+
- ml.m7i.4xlarge
193+
- ml.m7i.8xlarge
194+
- ml.m7i.12xlarge
195+
- ml.m7i.16xlarge
196+
- ml.m7i.24xlarge
197+
- ml.m7i.48xlarge
182198
- ml.p4d.24xlarge
183199
- ml.p4de.24xlarge
184200
- ml.p5.48xlarge
185201
- ml.p5e.48xlarge
186202
- ml.p5en.48xlarge
203+
- ml.r7i.large
204+
- ml.r7i.xlarge
205+
- ml.r7i.2xlarge
206+
- ml.r7i.4xlarge
207+
- ml.r7i.8xlarge
208+
- ml.r7i.12xlarge
209+
- ml.r7i.16xlarge
210+
- ml.r7i.24xlarge
211+
- ml.r7i.48xlarge
187212
- ml.trn1.32xlarge
188213
- ml.trn1n.32xlarge
189214
- ml.trn2.48xlarge

src/hyperpod_cli/constants/hyperpod_instance_types.py

Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,13 @@ class HyperpodInstanceType(Enum):
4040
ML_C5N_4XLARGE = "ml.c5n.4xlarge"
4141
ML_C5N_9XLARGE = "ml.c5n.9xlarge"
4242
ML_C5N_18XLARGE = "ml.c5n.18xlarge"
43+
ML_I3EN_LARGE = "ml.i3en.large"
44+
ML_I3EN_XLARGE = "ml.i3en.xlarge"
45+
ML_I3EN_2XLARGE = "ml.i3en.2xlarge"
46+
ML_I3EN_3XLARGE = "ml.i3en.3xlarge"
47+
ML_I3EN_6XLARGE = "ml.i3en.6xlarge"
48+
ML_I3EN_12XLARGE = "ml.i3en.12xlarge"
49+
ML_I3EN_24XLARGE = "ml.i3en.24xlarge"
4350
ML_M5_LARGE = "ml.m5.large"
4451
ML_M5_XLARGE = "ml.m5.xlarge"
4552
ML_M5_2XLARGE = "ml.m5.2xlarge"
@@ -48,6 +55,15 @@ class HyperpodInstanceType(Enum):
4855
ML_M5_12XLARGE = "ml.m5.12xlarge"
4956
ML_M5_16XLARGE = "ml.m5.16xlarge"
5057
ML_M5_24XLARGE = "ml.m5.24xlarge"
58+
ML_M7I_LARGE = "ml.m7i.large"
59+
ML_M7I_XLARGE = "ml.m7i.xlarge"
60+
ML_M7I_2XLARGE = "ml.m7i.2xlarge"
61+
ML_M7I_4XLARGE = "ml.m7i.4xlarge"
62+
ML_M7I_8XLARGE = "ml.m7i.8xlarge"
63+
ML_M7I_12XLARGE = "ml.m7i.12xlarge"
64+
ML_M7I_16XLARGE = "ml.m7i.16xlarge"
65+
ML_M7I_24XLARGE = "ml.m7i.24xlarge"
66+
ML_M7I_48XLARGE = "ml.m7i.48xlarge"
5167
ML_T3_MEDIUM = "ml.t3.medium"
5268
ML_T3_LARGE = "ml.t3.large"
5369
ML_T3_XLARGE = "ml.t3.xlarge"
@@ -72,4 +88,13 @@ class HyperpodInstanceType(Enum):
7288
ML_G6E_48XLARGE = "ml.g6e.48xlarge"
7389
ML_P5E_48XLARGE = "ml.p5e.48xlarge"
7490
ML_P5EN_48XLARGE = "ml.p5en.48xlarge"
91+
ML_R7I_LARGE = "ml.r7i.large"
92+
ML_R7I_XLARGE = "ml.r7i.xlarge"
93+
ML_R7I_2XLARGE = "ml.r7i.2xlarge"
94+
ML_R7I_4XLARGE = "ml.r7i.4xlarge"
95+
ML_R7I_8XLARGE = "ml.r7i.8xlarge"
96+
ML_R7I_12XLARGE = "ml.r7i.12xlarge"
97+
ML_R7I_16XLARGE = "ml.r7i.16xlarge"
98+
ML_R7I_24XLARGE = "ml.r7i.24xlarge"
99+
ML_R7I_48XLARGE = "ml.r7i.48xlarge"
75100
ML_TRN2_48XLARGE = "ml.trn2.48xlarge"

0 commit comments

Comments
 (0)