From d6fa63bc1a5efc98d4271cc7fc9070f270ebcc93 Mon Sep 17 00:00:00 2001
From: Feroze Mohideen <feroze@porter.run>
Date: Tue, 26 Mar 2024 13:41:46 -0400
Subject: [PATCH] add r7a instance types (#4448)

---
 dashboard/src/lib/clusters/constants.ts | 72 +++++++++++++++++++++++++
 dashboard/src/lib/clusters/types.ts     |  9 ++++
 2 files changed, 81 insertions(+)

diff --git a/dashboard/src/lib/clusters/constants.ts b/dashboard/src/lib/clusters/constants.ts
index 1aa60a06f1..8aa972e447 100644
--- a/dashboard/src/lib/clusters/constants.ts
+++ b/dashboard/src/lib/clusters/constants.ts
@@ -336,6 +336,78 @@ const SUPPORTED_AWS_MACHINE_TYPES: ClientMachineType[] = [
     cpuCores: 128,
     ramMegabytes: 1048576,
   },
+  {
+    name: "r7a.large",
+    displayName: "r7a.large",
+    supportedRegions: SUPPORTED_AWS_REGIONS.map((r) => r.name),
+    isGPU: false,
+    cpuCores: 2,
+    ramMegabytes: 16384,
+  },
+  {
+    name: "r7a.xlarge",
+    displayName: "r7a.xlarge",
+    supportedRegions: SUPPORTED_AWS_REGIONS.map((r) => r.name),
+    isGPU: false,
+    cpuCores: 4,
+    ramMegabytes: 32768,
+  },
+  {
+    name: "r7a.2xlarge",
+    displayName: "r7a.2xlarge",
+    supportedRegions: SUPPORTED_AWS_REGIONS.map((r) => r.name),
+    isGPU: false,
+    cpuCores: 8,
+    ramMegabytes: 65536,
+  },
+  {
+    name: "r7a.4xlarge",
+    displayName: "r7a.4xlarge",
+    supportedRegions: SUPPORTED_AWS_REGIONS.map((r) => r.name),
+    isGPU: false,
+    cpuCores: 16,
+    ramMegabytes: 131072,
+  },
+  {
+    name: "r7a.8xlarge",
+    displayName: "r7a.8xlarge",
+    supportedRegions: SUPPORTED_AWS_REGIONS.map((r) => r.name),
+    isGPU: false,
+    cpuCores: 32,
+    ramMegabytes: 262144,
+  },
+  {
+    name: "r7a.12xlarge",
+    displayName: "r7a.12xlarge",
+    supportedRegions: SUPPORTED_AWS_REGIONS.map((r) => r.name),
+    isGPU: false,
+    cpuCores: 48,
+    ramMegabytes: 393216,
+  },
+  {
+    name: "r7a.16xlarge",
+    displayName: "r7a.16xlarge",
+    supportedRegions: SUPPORTED_AWS_REGIONS.map((r) => r.name),
+    isGPU: false,
+    cpuCores: 64,
+    ramMegabytes: 524288,
+  },
+  {
+    name: "r7a.24xlarge",
+    displayName: "r7a.24xlarge",
+    supportedRegions: SUPPORTED_AWS_REGIONS.map((r) => r.name),
+    isGPU: false,
+    cpuCores: 96,
+    ramMegabytes: 786432,
+  },
+  {
+    name: "r7a.32xlarge",
+    displayName: "r7a.32xlarge",
+    supportedRegions: SUPPORTED_AWS_REGIONS.map((r) => r.name),
+    isGPU: false,
+    cpuCores: 128,
+    ramMegabytes: 1048576,
+  },
   {
     name: "m5n.large",
     displayName: "m5n.large",
diff --git a/dashboard/src/lib/clusters/types.ts b/dashboard/src/lib/clusters/types.ts
index 6f7f31e478..e104c8a742 100644
--- a/dashboard/src/lib/clusters/types.ts
+++ b/dashboard/src/lib/clusters/types.ts
@@ -143,6 +143,15 @@ const awsMachineTypeValidator = z.enum([
   "r6i.16xlarge",
   "r6i.24xlarge",
   "r6i.32xlarge",
+  "r7a.large",
+  "r7a.xlarge",
+  "r7a.2xlarge",
+  "r7a.4xlarge",
+  "r7a.8xlarge",
+  "r7a.12xlarge",
+  "r7a.16xlarge",
+  "r7a.24xlarge",
+  "r7a.32xlarge",
   "m5n.large",
   "m5n.xlarge",
   "m5n.2xlarge",