Merge pull request #109 from depot/fly-hardware

depot · Sep 12, 2024 · 068612c · 068612c
2 parents db78fda + 1fcfd89
commit 068612c
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 2 deletions.
diff --git a/proto/depot/cloud/v4/cloud.proto b/proto/depot/cloud/v4/cloud.proto
@@ -77,6 +77,7 @@ message GetDesiredStateResponse {
     KIND_BUILDKIT_8X16 = 4;
     KIND_BUILDKIT_4X8 = 5;
     KIND_BUILDKIT_4X4 = 6;
+    KIND_BUILDKIT_8X8 = 7;
   }
 
   enum MachineState {

diff --git a/src/proto/depot/cloud/v4/cloud_pb.ts b/src/proto/depot/cloud/v4/cloud_pb.ts
@@ -175,6 +175,11 @@ export enum GetDesiredStateResponse_Kind {
    * @generated from enum value: KIND_BUILDKIT_4X4 = 6;
    */
   BUILDKIT_4X4 = 6,
+
+  /**
+   * @generated from enum value: KIND_BUILDKIT_8X8 = 7;
+   */
+  BUILDKIT_8X8 = 7,
 }
 // Retrieve enum metadata with: proto3.getEnumType(GetDesiredStateResponse_Kind)
 proto3.util.setEnumType(GetDesiredStateResponse_Kind, 'depot.cloud.v4.GetDesiredStateResponse.Kind', [
@@ -185,6 +190,7 @@ proto3.util.setEnumType(GetDesiredStateResponse_Kind, 'depot.cloud.v4.GetDesired
   {no: 4, name: 'KIND_BUILDKIT_8X16'},
   {no: 5, name: 'KIND_BUILDKIT_4X8'},
   {no: 6, name: 'KIND_BUILDKIT_4X4'},
+  {no: 7, name: 'KIND_BUILDKIT_8X8'},
 ])
 
 /**

diff --git a/src/utils/fly/reconcile.ts b/src/utils/fly/reconcile.ts
@@ -248,9 +248,11 @@ function machineKind(kind: GetDesiredStateResponse_Kind): MachineKind {
     case GetDesiredStateResponse_Kind.BUILDKIT_4X4:
       return {cpuKind: 'shared', cpus: 4, memGBs: 4, needsGPU: false}
     case GetDesiredStateResponse_Kind.BUILDKIT_4X8:
-      return {cpuKind: 'performance', cpus: 4, memGBs: 8, needsGPU: false}
+      return {cpuKind: 'shared', cpus: 4, memGBs: 8, needsGPU: false}
+    case GetDesiredStateResponse_Kind.BUILDKIT_8X8:
+      return {cpuKind: 'shared', cpus: 8, memGBs: 8, needsGPU: false}
     case GetDesiredStateResponse_Kind.BUILDKIT_8X16:
-      return {cpuKind: 'performance', cpus: 8, memGBs: 16, needsGPU: false}
+      return {cpuKind: 'shared', cpus: 8, memGBs: 16, needsGPU: false}
     case GetDesiredStateResponse_Kind.BUILDKIT_16X32:
       return {cpuKind: 'performance', cpus: 16, memGBs: 32, needsGPU: false}
     case GetDesiredStateResponse_Kind.BUILDKIT_16X32_GPU: