From 6516a5c0de29e0a4642db952269b41277d81a27c Mon Sep 17 00:00:00 2001 From: lazariv Date: Wed, 18 Dec 2024 09:40:38 +0100 Subject: [PATCH] Update main.rs Add nvidia-a100-sxm4-40gb --- launcher/src/main.rs | 1 + 1 file changed, 1 insertion(+) diff --git a/launcher/src/main.rs b/launcher/src/main.rs index c89a2432..c092d745 100644 --- a/launcher/src/main.rs +++ b/launcher/src/main.rs @@ -1655,6 +1655,7 @@ impl From<&str> for Gpu { "nvidia-h100-nvl" => Gpu::H100, "nvidia-h100" => Gpu::H100, "nvidia-a100-sxm4-80gb" => Gpu::A100, + "nvidia-a100-sxm4-40gb" => Gpu::A100, "nvidia-a100-80gb-pcie" => Gpu::A100, "nvidia-a100" => Gpu::A100, card => Gpu::Unknown(card.to_string()),