fix: update constants (#639)

Co-authored-by: hoonoh <hoonoh@users.noreply.github.com>
hoonoh · Apr 14, 2023 · 3e3e382 · 3e3e382
1 parent 0683a62
commit 3e3e382
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 6 deletions.
diff --git a/src/constants/ec2-info.ts b/src/constants/ec2-info.ts
@@ -231,6 +231,10 @@ export const ec2Info: Record<InstanceType | string, Ec2InstanceInfo> = {
   'inf1.2xlarge': { vCpu: 8, memoryGiB: 16 },
   'inf1.6xlarge': { vCpu: 24, memoryGiB: 48 },
   'inf1.xlarge': { vCpu: 4, memoryGiB: 8 },
+  'inf2.24xlarge': { vCpu: 96, memoryGiB: 384 },
+  'inf2.48xlarge': { vCpu: 192, memoryGiB: 768 },
+  'inf2.8xlarge': { vCpu: 32, memoryGiB: 128 },
+  'inf2.xlarge': { vCpu: 4, memoryGiB: 16 },
   'is4gen.2xlarge': { vCpu: 8, memoryGiB: 48 },
   'is4gen.4xlarge': { vCpu: 16, memoryGiB: 96 },
   'is4gen.8xlarge': { vCpu: 32, memoryGiB: 192 },
@@ -582,6 +586,7 @@ export const ec2Info: Record<InstanceType | string, Ec2InstanceInfo> = {
   't4g.xlarge': { vCpu: 4, memoryGiB: 16 },
   'trn1.2xlarge': { vCpu: 8, memoryGiB: 32 },
   'trn1.32xlarge': { vCpu: 128, memoryGiB: 512 },
+  'trn1n.32xlarge': { vCpu: 128, memoryGiB: 512 },
   'u-12tb1.112xlarge': { vCpu: 448, memoryGiB: 12288 },
   'u-18tb1.112xlarge': { vCpu: 448, memoryGiB: 18432 },
   'u-24tb1.112xlarge': { vCpu: 448, memoryGiB: 24576 },

diff --git a/src/constants/ec2-types.ts b/src/constants/ec2-types.ts
@@ -96,11 +96,13 @@ export const instanceFamilyAcceleratedComputing = [
   'g5',
   'g5g',
   'inf1',
+  'inf2',
   'p2',
   'p3',
   'p3dn',
   'p4d',
   'trn1',
+  'trn1n',
   'vt1',
 ] as const;
 
@@ -756,6 +758,10 @@ export const allInstances = [
   'inf1.2xlarge',
   'inf1.6xlarge',
   'inf1.24xlarge',
+  'inf2.xlarge',
+  'inf2.8xlarge',
+  'inf2.24xlarge',
+  'inf2.48xlarge',
   'p2.xlarge',
   'p2.8xlarge',
   'p2.16xlarge',
@@ -766,6 +772,7 @@ export const allInstances = [
   'p4d.24xlarge',
   'trn1.2xlarge',
   'trn1.32xlarge',
+  'trn1n.32xlarge',
   'vt1.3xlarge',
   'vt1.6xlarge',
   'vt1.24xlarge',

diff --git a/test/__snapshots__/cli.spec.ts.snap b/test/__snapshots__/cli.spec.ts.snap
@@ -295,9 +295,11 @@ Options:
         "g5.4xlarge", "g5.8xlarge", "g5.12xlarge", "g5.16xlarge", "g5.24xlarge",
        "g5.48xlarge", "g5g.xlarge", "g5g.2xlarge", "g5g.4xlarge", "g5g.8xlarge",
      "g5g.16xlarge", "g5g.metal", "inf1.xlarge", "inf1.2xlarge", "inf1.6xlarge",
-        "inf1.24xlarge", "p2.xlarge", "p2.8xlarge", "p2.16xlarge", "p3.2xlarge",
+                "inf1.24xlarge", "inf2.xlarge", "inf2.8xlarge", "inf2.24xlarge",
+        "inf2.48xlarge", "p2.xlarge", "p2.8xlarge", "p2.16xlarge", "p3.2xlarge",
    "p3.8xlarge", "p3.16xlarge", "p3dn.24xlarge", "p4d.24xlarge", "trn1.2xlarge",
-                  "trn1.32xlarge", "vt1.3xlarge", "vt1.6xlarge", "vt1.24xlarge"]
+                "trn1.32xlarge", "trn1n.32xlarge", "vt1.3xlarge", "vt1.6xlarge",
+                                                                 "vt1.24xlarge"]
       --family              EC2 instance family.
                     [array] [choices: "general", "compute", "memory", "storage",
                                                          "acceleratedComputing"]
@@ -310,7 +312,8 @@ Options:
     "r6gd", "r6i", "r6id", "r6idn", "r6in", "r7g", "x1", "x1e", "x2gd", "x2idn",
         "x2iedn", "x2iezn", "z1d", "d2", "d3", "d3en", "h1", "i2", "i3", "i3en",
        "i4i", "im4gn", "is4gen", "dl1", "f1", "g2", "g3", "g3s", "g4ad", "g4dn",
-                  "g5", "g5g", "inf1", "p2", "p3", "p3dn", "p4d", "trn1", "vt1"]
+        "g5", "g5g", "inf1", "inf2", "p2", "p3", "p3dn", "p4d", "trn1", "trn1n",
+                                                                          "vt1"]
   -s, --size                EC2 instance sizes.
         [array] [choices: "nano", "micro", "small", "medium", "large", "xlarge",
    "2xlarge", "3xlarge", "4xlarge", "6xlarge", "8xlarge", "9xlarge", "10xlarge",
@@ -649,9 +652,11 @@ Options:
         "g5.4xlarge", "g5.8xlarge", "g5.12xlarge", "g5.16xlarge", "g5.24xlarge",
        "g5.48xlarge", "g5g.xlarge", "g5g.2xlarge", "g5g.4xlarge", "g5g.8xlarge",
      "g5g.16xlarge", "g5g.metal", "inf1.xlarge", "inf1.2xlarge", "inf1.6xlarge",
-        "inf1.24xlarge", "p2.xlarge", "p2.8xlarge", "p2.16xlarge", "p3.2xlarge",
+                "inf1.24xlarge", "inf2.xlarge", "inf2.8xlarge", "inf2.24xlarge",
+        "inf2.48xlarge", "p2.xlarge", "p2.8xlarge", "p2.16xlarge", "p3.2xlarge",
    "p3.8xlarge", "p3.16xlarge", "p3dn.24xlarge", "p4d.24xlarge", "trn1.2xlarge",
-                  "trn1.32xlarge", "vt1.3xlarge", "vt1.6xlarge", "vt1.24xlarge"]
+                "trn1.32xlarge", "trn1n.32xlarge", "vt1.3xlarge", "vt1.6xlarge",
+                                                                 "vt1.24xlarge"]
       --family              EC2 instance family.
                     [array] [choices: "general", "compute", "memory", "storage",
                                                          "acceleratedComputing"]
@@ -664,7 +669,8 @@ Options:
     "r6gd", "r6i", "r6id", "r6idn", "r6in", "r7g", "x1", "x1e", "x2gd", "x2idn",
         "x2iedn", "x2iezn", "z1d", "d2", "d3", "d3en", "h1", "i2", "i3", "i3en",
        "i4i", "im4gn", "is4gen", "dl1", "f1", "g2", "g3", "g3s", "g4ad", "g4dn",
-                  "g5", "g5g", "inf1", "p2", "p3", "p3dn", "p4d", "trn1", "vt1"]
+        "g5", "g5g", "inf1", "inf2", "p2", "p3", "p3dn", "p4d", "trn1", "trn1n",
+                                                                          "vt1"]
   -s, --size                EC2 instance sizes.
         [array] [choices: "nano", "micro", "small", "medium", "large", "xlarge",
    "2xlarge", "3xlarge", "4xlarge", "6xlarge", "8xlarge", "9xlarge", "10xlarge",