[Benchmark] Add new H100 machine (#10547)

This commit is contained in:
Simon Mo
2024-11-21 18:27:20 -08:00
committed by GitHub
parent 9afa014552
commit aed074860a
2 changed files with 31 additions and 21 deletions

View File

@@ -157,10 +157,17 @@ if __name__ == "__main__":
throughput_results,
serving_results)
# Sort all dataframes by their respective "Test name" columns
for df in [latency_results, serving_results, throughput_results]:
if not df.empty:
df.sort_values(by="Test name", inplace=True)
if df.empty:
continue
# Sort all dataframes by their respective "Test name" columns
df.sort_values(by="Test name", inplace=True)
# The GPUs sometimes come in format of "GPUTYPE\nGPUTYPE\n...",
# we want to turn it into "8xGPUTYPE"
df["GPU"] = df["GPU"].apply(
lambda x: f"{len(x.split('\n'))}x{x.split('\n')[0]}")
# get markdown tables
latency_md_table = tabulate(latency_results,