add testing

SkafteNicki · SkafteNicki · commit 854f07ab3395 · 2025-08-27T06:34:55.000+02:00
diff --git a/tests/tests_fabric/utilities/test_throughput.py b/tests/tests_fabric/utilities/test_throughput.py
@@ -68,6 +68,8 @@ def test_get_available_flops(xla_available):
     "device_name",
     [
         # Hopper
+        "NVIDIA H200 SXM1",
+        "NVIDIA H200 NVL1",
         "h100-nvl",  # TODO: switch with `torch.cuda.get_device_name()` result
         "h100-hbm3",  # TODO: switch with `torch.cuda.get_device_name()` result
         "NVIDIA H100 PCIe",

Original file line number	Diff line number	Diff line change
`@@ -68,6 +68,8 @@ def test_get_available_flops(xla_available):`
`68`	`68`	`"device_name",`
`69`	`69`	`[`
`70`	`70`	`# Hopper`
	`71`	`+ "NVIDIA H200 SXM1",`
	`72`	`+ "NVIDIA H200 NVL1",`
`71`	`73`	"h100-nvl", # TODO: switch with `torch.cuda.get_device_name()` result
`72`	`74`	"h100-hbm3", # TODO: switch with `torch.cuda.get_device_name()` result
`73`	`75`	`"NVIDIA H100 PCIe",`