Merge pull request #445 from mystic-ai/matthew/a10s

Add A10 fractional accelerators
mystic-ai · Mar 26, 2024 · 3372a21 · 3372a21
2 parents 9adf1db + e8a326c
commit 3372a21
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/pipeline/cloud/compute_requirements.py b/pipeline/cloud/compute_requirements.py
@@ -19,6 +19,9 @@ class Accelerator(str, Enum):
     nvidia_a100_80gb_40gb: str = "nvidia_a100_80gb_40gb"
 
     nvidia_a10: str = "nvidia_a10"
+    nvidia_a10_12gb: str = "nvidia_a10_12gb"
+    nvidia_a10_8gb: str = "nvidia_a10_8gb"
+    nvidia_a10_4gb: str = "nvidia_a10_4gb"
 
     @classmethod
     def from_str(cls, accelerator: str) -> "Accelerator":
@@ -71,4 +74,7 @@ def valid_accelerator_config(cls, accelerators: list["Accelerator"]):
     Accelerator.nvidia_a100_80gb_20gb,
     Accelerator.nvidia_a100_80gb_40gb,
     Accelerator.nvidia_a10,
+    Accelerator.nvidia_a10_12gb,
+    Accelerator.nvidia_a10_8gb,
+    Accelerator.nvidia_a10_4gb,
 ]
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pipeline-ai"
-version = "2.1.2"
+version = "2.1.3"
 description = "Pipelines for machine learning workloads."
 authors = [
   "Paul Hetherington <ph@mystic.ai>",