Feat: enable new modal gpus 🚀

Author: S1ro1

src/discord-cluster-manager/consts.py

@@ -29,6 +29,7 @@ class ModalGPU(Enum):
     A100 = "A100"
     H100 = "H100"
     B200 = "B200"
+    H200 = "H200"
 
 
 @dataclasses.dataclass
@@ -115,6 +116,7 @@ class RankCriterion(Enum):
     "L4": "80",
     "A100": "80",
     "H100": "90a",
+    "H200": "90a",
     "B200": "100",
     "NVIDIA": None,
     "MI300": None,

src/discord-cluster-manager/modal_runner_archs.py

@@ -4,7 +4,7 @@
 from modal_utils import deserialize_full_result
 from run_eval import FullResult, SystemInfo
 
-gpus = ["T4", "L4", "A100-80GB", "H100!"]
+gpus = ["T4", "L4", "A100-80GB", "H100!", "B200", "H200"]
 for gpu in gpus:
     gpu_slug = gpu.lower().split("-")[0].strip("!")
     app.function(gpu=gpu, image=cuda_image, name=f"run_cuda_script_{gpu_slug}", serialized=True)(