diff --git a/python/sglang/bench_serving.py b/python/sglang/bench_serving.py index e328b7094..eee5607cf 100644 --- a/python/sglang/bench_serving.py +++ b/python/sglang/bench_serving.py @@ -2033,6 +2033,7 @@ async def benchmark( ): result = { # Arguments + "tag": args.tag, "backend": args.backend, "dataset_name": args.dataset_name, "request_rate": "trace" if use_trace_timestamps else request_rate, @@ -2627,5 +2628,8 @@ if __name__ == "__main__": ], 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() run_benchmark(args)