Merge pull request #436 from mystic-ai/matthew/pc-1053-aws-gpu-options

A10 GPU support

pipeline/cloud/compute_requirements.py

@@ -18,6 +18,8 @@ class Accelerator(str, Enum):
     nvidia_a100_80gb_20gb: str = "nvidia_a100_80gb_20gb"
     nvidia_a100_80gb_40gb: str = "nvidia_a100_80gb_40gb"
 
+    nvidia_a10: str = "nvidia_a10"
+
     @classmethod
     def from_str(cls, accelerator: str) -> "Accelerator":
         if "T4" in accelerator:
@@ -31,6 +33,9 @@ def from_str(cls, accelerator: str) -> "Accelerator":
             accelerator_type = Accelerator.nvidia_h100
         elif "L4" in accelerator:
             accelerator_type = Accelerator.nvidia_l4
+        # guess this works as long as its after A100
+        elif "A10" in accelerator:
+            accelerator_type = Accelerator.nvidia_a10
 
         else:
             raise Exception(f"Unknown GPU name: {accelerator}")
@@ -48,6 +53,7 @@ def valid_accelerator_config(cls, accelerators: list["Accelerator"]):
             [Accelerator.nvidia_a100_80gb] * 2,
             [Accelerator.nvidia_a100_80gb] * 4,
             [Accelerator.nvidia_l4],
+            [Accelerator.nvidia_a10],
             [Accelerator.cpu],
         ]
 
@@ -64,4 +70,5 @@ def valid_accelerator_config(cls, accelerators: list["Accelerator"]):
     Accelerator.nvidia_a100_80gb_10gb,
     Accelerator.nvidia_a100_80gb_20gb,
     Accelerator.nvidia_a100_80gb_40gb,
+    Accelerator.nvidia_a10,
 ]

pipeline/container/frontend/src/types.ts

@@ -56,6 +56,7 @@ export type Accelerator =
   | "nvidia_h100"
   | "nvidia_l4"
   | "nvidia_all"
+  | "nvidia_a10"
   | "cpu";
 
 export interface GetPipelineResponse {

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pipeline-ai"
-version = "2.0.23"
+version = "2.0.24"
 description = "Pipelines for machine learning workloads."
 authors = [
   "Paul Hetherington <ph@mystic.ai>",