From e8a326cb6588304b1f4b15c8729155e75a6095cd Mon Sep 17 00:00:00 2001 From: Matthew Cherry Date: Tue, 26 Mar 2024 16:10:20 +0000 Subject: [PATCH] Add A10 fractional accelerators --- pipeline/cloud/compute_requirements.py | 6 ++++++ pyproject.toml | 2 +- 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/pipeline/cloud/compute_requirements.py b/pipeline/cloud/compute_requirements.py index 8f95b071..9eec048b 100644 --- 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 index 45cb3619..0d834f2d 100644 --- 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 ",