From bd22221cc7da9233b1afd5d1c0e316a02196d3b7 Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Tue, 5 Nov 2024 11:19:37 +0100
Subject: [PATCH] Adding A100 + H100

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

diff --git a/launcher/src/main.rs b/launcher/src/main.rs
index 6e054185..c6fd8574 100644
--- a/launcher/src/main.rs
+++ b/launcher/src/main.rs
@@ -1486,6 +1486,10 @@ impl ComputeType {
             "nvidia-l4" => Some(121 * 10u64.pow(12)),
             // https://www.nvidia.com/en-us/data-center/products/a10-gpu/
             "nvidia-a10g" => Some(125 * 10u64.pow(12)),
+            // https://www.nvidia.com/en-us/data-center/h100/
+            "nvidia-h100" => Some(1900 * 10u64.pow(12)),
+            // https://www.nvidia.com/content/dam/en-zz/Solutions/Data-Center/a100/pdf/nvidia-a100-datasheet-us-nvidia-1758950-r4-web.pdf
+            "nvidia-a100" => Some(312 * 10u64.pow(12)),
             card => {
                 tracing::warn!("Unkown compute for card {card}");
                 None