h100 -- support correctly restricting machine types

sagemathinc · Jan 28, 2025 · b7c07eb · b7c07eb
1 parent 20a75e5
commit b7c07eb
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/src/packages/frontend/compute/google-cloud-config.tsx b/src/packages/frontend/compute/google-cloud-config.tsx
@@ -976,18 +976,14 @@ function MachineType({ priceData, setConfig, configuration, disabled, state }) {
           return false;
         }
       } else {
-        if (
-          acceleratorType == "nvidia-tesla-a100" ||
-          acceleratorType == "nvidia-a100-80gb" ||
-          acceleratorType == "nvidia-l4"
-        ) {
+        if (acceleratorType == "nvidia-tesla-t4") {
+          return machineType.startsWith("n1-");
+        } else {
           const machines =
             priceData.accelerators[acceleratorType].machineType[
               configuration.acceleratorCount ?? 1
             ] ?? [];
           return machines.includes(machineType);
-        } else {
-          return machineType.startsWith("n1-");
         }
       }