From 32af545499c2cb28cbb5586de946dae8b84b4cc7 Mon Sep 17 00:00:00 2001
From: d-g-town <66391417+d-g-town@users.noreply.github.com>
Date: Fri, 17 May 2024 11:02:42 -0400
Subject: [PATCH] add c7gn (#4654)

---
 .../src/components/ProvisionerSettings.tsx    |  7 ++
 dashboard/src/lib/clusters/constants.ts       | 64 +++++++++++++++++++
 dashboard/src/lib/clusters/types.ts           |  8 +++
 3 files changed, 79 insertions(+)

diff --git a/dashboard/src/components/ProvisionerSettings.tsx b/dashboard/src/components/ProvisionerSettings.tsx
index 2d066dacad..b1d6a261c4 100644
--- a/dashboard/src/components/ProvisionerSettings.tsx
+++ b/dashboard/src/components/ProvisionerSettings.tsx
@@ -148,6 +148,13 @@ const machineTypeOptions = [
   { value: "c7g.8xlarge", label: "c7g.8xlarge" },
   { value: "c7g.12xlarge", label: "c7g.12xlarge" },
   { value: "c7g.16xlarge", label: "c7g.16xlarge" },
+  { value: "c7gn.large", label: "c7gn.large" },
+  { value: "c7gn.xlarge", label: "c7gn.xlarge" },
+  { value: "c7gn.2xlarge", label: "c7gn.2xlarge" },
+  { value: "c7gn.4xlarge", label: "c7gn.4xlarge" },
+  { value: "c7gn.8xlarge", label: "c7gn.8xlarge" },
+  { value: "c7gn.12xlarge", label: "c7gn.12xlarge" },
+  { value: "c7gn.16xlarge", label: "c7gn.16xlarge" },
 ];
 
 const defaultCidrVpc = "10.78.0.0/16";
diff --git a/dashboard/src/lib/clusters/constants.ts b/dashboard/src/lib/clusters/constants.ts
index 52cd836042..e94df0adde 100644
--- a/dashboard/src/lib/clusters/constants.ts
+++ b/dashboard/src/lib/clusters/constants.ts
@@ -801,6 +801,70 @@ const SUPPORTED_AWS_MACHINE_TYPES: ClientMachineType[] = [
     cpuCores: 64,
     ramMegabytes: 131072,
   },
+  {
+    name: "c7gn.medium",
+    displayName: "c7gn.medium",
+    supportedRegions: SUPPORTED_AWS_REGIONS.map((r) => r.name),
+    isGPU: false,
+    cpuCores: 1,
+    ramMegabytes: 2048,
+  },
+  {
+    name: "c7gn.large",
+    displayName: "c7gn.large",
+    supportedRegions: SUPPORTED_AWS_REGIONS.map((r) => r.name),
+    isGPU: false,
+    cpuCores: 2,
+    ramMegabytes: 4096,
+  },
+  {
+    name: "c7gn.xlarge",
+    displayName: "c7gn.xlarge",
+    supportedRegions: SUPPORTED_AWS_REGIONS.map((r) => r.name),
+    isGPU: false,
+    cpuCores: 4,
+    ramMegabytes: 8192,
+  },
+  {
+    name: "c7gn.2xlarge",
+    displayName: "c7gn.2xlarge",
+    supportedRegions: SUPPORTED_AWS_REGIONS.map((r) => r.name),
+    isGPU: false,
+    cpuCores: 8,
+    ramMegabytes: 16384,
+  },
+  {
+    name: "c7gn.4xlarge",
+    displayName: "c7gn.4xlarge",
+    supportedRegions: SUPPORTED_AWS_REGIONS.map((r) => r.name),
+    isGPU: false,
+    cpuCores: 16,
+    ramMegabytes: 32768,
+  },
+  {
+    name: "c7gn.8xlarge",
+    displayName: "c7gn.8xlarge",
+    supportedRegions: SUPPORTED_AWS_REGIONS.map((r) => r.name),
+    isGPU: false,
+    cpuCores: 32,
+    ramMegabytes: 65536,
+  },
+  {
+    name: "c7gn.12xlarge",
+    displayName: "c7gn.12xlarge",
+    supportedRegions: SUPPORTED_AWS_REGIONS.map((r) => r.name),
+    isGPU: false,
+    cpuCores: 48,
+    ramMegabytes: 98304,
+  },
+  {
+    name: "c7gn.16xlarge",
+    displayName: "c7gn.16xlarge",
+    supportedRegions: SUPPORTED_AWS_REGIONS.map((r) => r.name),
+    isGPU: false,
+    cpuCores: 64,
+    ramMegabytes: 131072,
+  },
   {
     name: "g4dn.xlarge",
     displayName: "g4dn.xlarge",
diff --git a/dashboard/src/lib/clusters/types.ts b/dashboard/src/lib/clusters/types.ts
index 4ad4b09225..cfa2c69c4d 100644
--- a/dashboard/src/lib/clusters/types.ts
+++ b/dashboard/src/lib/clusters/types.ts
@@ -202,6 +202,14 @@ const awsMachineTypeValidator = z.enum([
   "c7g.8xlarge",
   "c7g.12xlarge",
   "c7g.16xlarge",
+  "c7gn.medium",
+  "c7gn.large",
+  "c7gn.xlarge",
+  "c7gn.2xlarge",
+  "c7gn.4xlarge",
+  "c7gn.8xlarge",
+  "c7gn.12xlarge",
+  "c7gn.16xlarge",
   // gpu types
   "g4dn.xlarge",
   "g4dn.2xlarge",