From 83624a07be8894dcdb9d35c8519201b23ccb07c9 Mon Sep 17 00:00:00 2001
From: lazariv <lazariv.taras@gmail.com>
Date: Fri, 10 Jan 2025 16:12:02 +0100
Subject: [PATCH] Add possible variants for A100 and H100 GPUs for
 auto-detecting flops (#2837)

* Update main.rs with A100 and H100 variants

* Add another variant "nvidia-h100-nvl"

* Update main.rs

Add nvidia-a100-sxm4-40gb
---
 launcher/src/main.rs | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/launcher/src/main.rs b/launcher/src/main.rs
index fb6ba2b25..c092d7451 100644
--- a/launcher/src/main.rs
+++ b/launcher/src/main.rs
@@ -1652,7 +1652,11 @@ impl From<&str> for Gpu {
             "nvidia-l40s" => Gpu::L40S,
             "nvidia-a10g" => Gpu::A10G,
             "nvidia-h100-80gb-hbm3" => Gpu::H100,
+            "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()),
         }