From 8a8a608af99434c1239f6346e5986cd9880b453b Mon Sep 17 00:00:00 2001
From: Liangsheng Yin <hnyls2002@gmail.com>
Date: Tue, 7 Oct 2025 14:20:04 +0800
Subject: [PATCH] [ci] fix pp test (#11294)

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
---
 test/srt/test_bench_serving.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/test/srt/test_bench_serving.py b/test/srt/test_bench_serving.py
index 978033e3b..747794609 100644
--- a/test/srt/test_bench_serving.py
+++ b/test/srt/test_bench_serving.py
@@ -406,7 +406,7 @@ class TestBenchServing(CustomTestCase):
             request_rate=float("inf"),
             random_input_len=1,
             random_output_len=1024,
-            other_server_args=["--pp", "2"],
+            other_server_args=["--pp-size", "2"],
             need_warmup=True,
             seed=42,
         )
@@ -429,8 +429,8 @@ class TestBenchServing(CustomTestCase):
             other_server_args=[
                 "--quantization",
                 "fp8",
-                "--pp",
-                2,
+                "--pp-size",
+                "2",
             ],
             need_warmup=False,
             seed=42,