diff --git a/src/gpuhunt/_internal/constraints.py b/src/gpuhunt/_internal/constraints.py index e4199f8..c33e758 100644 --- a/src/gpuhunt/_internal/constraints.py +++ b/src/gpuhunt/_internal/constraints.py @@ -223,6 +223,7 @@ def is_nvidia_superchip(gpu_name: str) -> bool: NvidiaGPUInfo(name="RTX2000Ada", memory=16, compute_capability=(8, 9)), NvidiaGPUInfo(name="RTX4000Ada", memory=20, compute_capability=(8, 9)), NvidiaGPUInfo(name="RTX6000Ada", memory=48, compute_capability=(8, 9)), + NvidiaGPUInfo(name="RTXPRO4500", memory=32, compute_capability=(12, 0)), NvidiaGPUInfo(name="RTXPRO6000", memory=96, compute_capability=(12, 0)), NvidiaGPUInfo(name="T4", memory=16, compute_capability=(7, 5)), NvidiaGPUInfo(name="V100", memory=16, compute_capability=(7, 0)), diff --git a/src/gpuhunt/providers/aws.py b/src/gpuhunt/providers/aws.py index 5d3725f..a7e00c0 100644 --- a/src/gpuhunt/providers/aws.py +++ b/src/gpuhunt/providers/aws.py @@ -86,6 +86,7 @@ "me-south-1", } GPU_NAME_MAPPING = { + "RTX PRO 4500": "RTXPRO4500", "RTX PRO Server 6000": "RTXPRO6000", } @@ -384,6 +385,7 @@ def filter(cls, offers: list[RawCatalogItem]) -> list[RawCatalogItem]: "p5e.", "p4d.", "p4de.", + "g7.", "g7e.", "g6.", "g6e.", diff --git a/src/integrity_tests/test_aws.py b/src/integrity_tests/test_aws.py index 38bb085..a6183c6 100644 --- a/src/integrity_tests/test_aws.py +++ b/src/integrity_tests/test_aws.py @@ -59,7 +59,9 @@ def test_gpu_presented(self, data: str): "A10G", "T4", "L40S", + "RTXPRO4500", "RTXPRO6000", "L4", ] - assert all(f",{i}," in data for i in gpus) + for gpu in gpus: + assert f",{gpu}," in data