Skip to content

Commit 0428d5d

Browse files
authored
[Benchmark CI] Simplify h100 display name (#713)
1 parent 66cbf6a commit 0428d5d

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

benchmarks/run.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -906,7 +906,11 @@ def helion_method(
906906
@functools.cache
907907
def get_device_name() -> str:
908908
if torch.cuda.is_available():
909-
return torch.cuda.get_device_name(0)
909+
name = torch.cuda.get_device_name(0)
910+
# Inconsistent name reporting, so lets fix H100 to report simple name
911+
if name.startswith("cuda (NVIDIA H100"):
912+
return "cuda (NVIDIA H100)"
913+
return name
910914
return "unknown"
911915

912916

0 commit comments

Comments
 (0)