From 25c5049870ff9e7f0029454faf5a70f17d461317 Mon Sep 17 00:00:00 2001
From: fzyzcjy <5236035+fzyzcjy@users.noreply.github.com>
Date: Thu, 30 Oct 2025 11:19:14 +0800
Subject: [PATCH] Super tiny add tag for benchmark scripts (#12340)

---
 python/sglang/bench_serving.py | 4 ++++
 1 file changed, 4 insertions(+)

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)