[CI] Tiny enhance the dp attention load blance benchmark (#19194)

2026-02-23 14:33:32 -08:00
parent 672b666057
commit 9fac90d85d
2 changed files with 23 additions and 1 deletions
--- a/python/sglang/test/test_utils.py
+++ b/python/sglang/test/test_utils.py
@@ -1041,6 +1041,7 @@ def get_benchmark_args(
    gsp_output_len=32,
    gsp_num_turns=1,
    header=None,
+    max_concurrency=None,
 ):
    return SimpleNamespace(
        backend=backend,
@@ -1082,6 +1083,7 @@ def get_benchmark_args(
        gsp_output_len=gsp_output_len,
        gsp_num_turns=gsp_num_turns,
        header=header,
+        max_concurrency=max_concurrency,
    )