Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "ebaa4d954aa0b3d70f14b2fd6fec63df1599b8ce"
Unverified Commit 25c50498 authored by fzyzcjy's avatar fzyzcjy Committed by GitHub
Browse files

Super tiny add tag for benchmark scripts (#12340)

parent 8e2ac2e6
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
python/sglang/bench_serving.py
View file @ 25c50498
...@@ -2033,6 +2033,7 @@ async def benchmark( ...@@ -2033,6 +2033,7 @@ async def benchmark(
): ):
result = { result = {
# Arguments # Arguments
"tag": args.tag,
"backend": args.backend, "backend": args.backend,
"dataset_name": args.dataset_name, "dataset_name": args.dataset_name,
"request_rate": "trace" if use_trace_timestamps else request_rate, "request_rate": "trace" if use_trace_timestamps else request_rate,
...@@ -2627,5 +2628,8 @@ if __name__ == "__main__": ...@@ -2627,5 +2628,8 @@ if __name__ == "__main__":
], ],
help="Underlying workload for the mooncake dataset.", help="Underlying workload for the mooncake dataset.",
) )
parser.add_argument(
"--tag", type=str, default=None, help="The tag to be dumped to output."
)
args = parser.parse_args() args = parser.parse_args()
run_benchmark(args) run_benchmark(args)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment