EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

misc: parse bench_serving result as markdown table (#9377)

Browse Source
This commit is contained in:
Mick
2025-08-21 07:44:20 +08:00
committed by GitHub
parent 84719b527a
commit ef3004d90a
1 changed files with 11 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
test/srt/parse_results.py
View File

@@ -8,6 +8,11 @@ from tabulate import tabulate
# Parse command-line arguments # Parse command-line arguments
parser = argparse.ArgumentParser(description="Parse JSONL benchmark and summarize.") parser = argparse.ArgumentParser(description="Parse JSONL benchmark and summarize.")
parser.add_argument("input_file", type=str, help="Path to input JSONL file") parser.add_argument("input_file", type=str, help="Path to input JSONL file")
parser.add_argument(
"--md",
action="store_true",
help="If set, print the summary table in Markdown format (GitHub style)",
)
args = parser.parse_args() args = parser.parse_args()
input_file = args.input_file input_file = args.input_file
@@ -44,5 +49,9 @@ df = pd.DataFrame(results)
df.to_csv(output_file, index=False) df.to_csv(output_file, index=False)
print(f"\nSaved summary to: {output_file}\n") print(f"\nSaved summary to: {output_file}\n")
# Print ASCII table if args.md:
print(tabulate(df, headers="keys", tablefmt="grid", floatfmt=".3f")) # Print Markdown table
print(tabulate(df, headers="keys", tablefmt="github", floatfmt=".3f"))
else:
# Print ASCII table
print(tabulate(df, headers="keys", tablefmt="grid", floatfmt=".3f"))