* update more supported instance types to efa plugin (#45)

* Add gr6 instance to efa plugin

Add more hyperpod supported instance types

Author: can-sun

helm_chart/HyperPodHelmChart/values.yaml

@@ -168,11 +168,25 @@ aws-efa-k8s-device-plugin:
       - ml.g5.16xlarge
       - ml.g5.24xlarge
       - ml.g5.48xlarge
+      - ml.g6.8xlarge
+      - ml.g6.12xlarge
+      - ml.g6.16xlarge
+      - ml.g6.24xlarge
+      - ml.g6.48xlarge
+      - ml.g6e.8xlarge
+      - ml.g6e.12xlarge
+      - ml.g6e.16xlarge
+      - ml.g6e.24xlarge
+      - ml.g6e.48xlarge
+      - ml.gr6.8xlarge
       - ml.p4d.24xlarge
       - ml.p4de.24xlarge
       - ml.p5.48xlarge
+      - ml.p5e.48xlarge
+      - ml.p5en.48xlarge
       - ml.trn1.32xlarge
       - ml.trn1n.32xlarge
+      - ml.trn2.48xlarge
   tolerations:
     - key: CriticalAddonsOnly
       operator: Exists

src/hyperpod_cli/constants/hyperpod_instance_types.py

@@ -52,3 +52,24 @@ class HyperpodInstanceType(Enum):
     ML_T3_LARGE = "ml.t3.large"
     ML_T3_XLARGE = "ml.t3.xlarge"
     ML_T3_2XLARGE = "ml.t3.2xlarge"
+    ML_G6_XLARGE = "ml.g6.xlarge"
+    ML_G6_2XLARGE = "ml.g6.2xlarge"
+    ML_G6_4XLARGE = "ml.g6.4xlarge"
+    ML_G6_8XLARGE = "ml.g6.8xlarge"
+    ML_G6_16XLARGE = "ml.g6.16xlarge"
+    ML_G6_12XLARGE = "ml.g6.12xlarge"
+    ML_G6_24XLARGE = "ml.g6.24xlarge"
+    ML_G6_48XLARGE = "ml.g6.48xlarge"
+    ML_GR6_4XLARGE = "ml.gr6.4xlarge"
+    ML_GR6_8XLARGE = "ml.gr6.8xlarge"
+    ML_G6E_XLARGE = "ml.g6e.xlarge"
+    ML_G6E_2XLARGE = "ml.g6e.2xlarge"
+    ML_G6E_4XLARGE = "ml.g6e.4xlarge"
+    ML_G6E_8XLARGE = "ml.g6e.8xlarge"
+    ML_G6E_16XLARGE = "ml.g6e.16xlarge"
+    ML_G6E_12XLARGE = "ml.g6e.12xlarge"
+    ML_G6E_24XLARGE = "ml.g6e.24xlarge"
+    ML_G6E_48XLARGE = "ml.g6e.48xlarge"
+    ML_P5E_48XLARGE = "ml.p5e.48xlarge"
+    ML_P5EN_48XLARGE = "ml.p5en.48xlarge"
+    ML_TRN2_48XLARGE = "ml.trn2.48xlarge"