From 3da49f3fb9c1d1d036ec662a7fe600563a29ccc2 Mon Sep 17 00:00:00 2001
From: lazariv <lazariv.taras@gmail.com>
Date: Fri, 13 Dec 2024 09:53:06 +0100
Subject: [PATCH] Update main.rs with A100 and H100 variants

---
 launcher/src/main.rs | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/launcher/src/main.rs b/launcher/src/main.rs
index fb6ba2b2..2a98ce81 100644
--- a/launcher/src/main.rs
+++ b/launcher/src/main.rs
@@ -1652,7 +1652,9 @@ impl From<&str> for Gpu {
             "nvidia-l40s" => Gpu::L40S,
             "nvidia-a10g" => Gpu::A10G,
             "nvidia-h100-80gb-hbm3" => Gpu::H100,
+            "nvidia-h100" => Gpu::H100,
             "nvidia-a100-sxm4-80gb" => Gpu::A100,
+            "nvidia-a100-80gb-pcie" => Gpu::A100,
             "nvidia-a100" => Gpu::A100,
             card => Gpu::Unknown(card.to_string()),
         }