Test for H200

cmdr2 · cmdr2 · commit 7f6e70a82fbd · 2025-10-04T16:24:27.000+05:30
diff --git a/tests/test_gpu_db.py b/tests/test_gpu_db.py
@@ -42,7 +42,7 @@ def test_ampere_detection(self):
         assert get_nvidia_arch(device_names) == 8.6
 
     def test_hopper_detection(self):
-        device_names = ["NVIDIA GH100", "H100 PCIe"]
+        device_names = ["NVIDIA GH100", "H100 PCIe", "H200 SXM 141GB"]
         assert get_nvidia_arch(device_names) == 9.0
 
     def test_ada_lovelace_detection(self):
