minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 13 Packages Projects Releases Wiki Activity

[CP V1 Deprecation 1/5] Migrate tests to strategy-based prefill CP (#36222)

Browse Source
This commit is contained in:
Baizhou Zhang
2026-08-24 20:15:34 -07:00
committed by GitHub
parent a0f52eca98
commit 833be86c15
20 changed files with 37 additions and 558 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/manual/dsv4/test_b200_flash.py
+3 -3
View File
@@ -87,9 +87,9 @@ class TestB200FlashCP(DSV4FlashAime25TestBase):
"4",
"--moe-a2a-backend",
"deepep",
"--enable-dsa-prefill-context-parallel",
"--dsa-prefill-cp-mode",
"round-robin-split",
"--enable-prefill-cp",
"--cp-strategy",
"interleave",
"--chunked-prefill-size",
"16384",
"--mem-fraction-static",
test/manual/dsv4/test_b200_pro.py
+3 -3
View File
@@ -101,9 +101,9 @@ class TestB200ProCP(DSV4ProAime25TestBase):
"8",
"--moe-a2a-backend",
"deepep",
"--enable-dsa-prefill-context-parallel",
"--dsa-prefill-cp-mode",
"round-robin-split",
"--enable-prefill-cp",
"--cp-strategy",
"interleave",
"--chunked-prefill-size",
"16384",
"--mem-fraction-static",
test/manual/dsv4/test_b300_flash.py
+3 -3
View File
@@ -89,9 +89,9 @@ class TestB300FlashCP(DSV4FlashAime25TestBase):
"4",
"--moe-a2a-backend",
"deepep",
"--enable-dsa-prefill-context-parallel",
"--dsa-prefill-cp-mode",
"round-robin-split",
"--enable-prefill-cp",
"--cp-strategy",
"interleave",
"--chunked-prefill-size",
"16384",
"--mem-fraction-static",
test/manual/dsv4/test_b300_pro.py
+3 -3
View File
@@ -103,9 +103,9 @@ class TestB300ProCP(DSV4ProAime25TestBase):
"8",
"--moe-a2a-backend",
"deepep",
"--enable-dsa-prefill-context-parallel",
"--dsa-prefill-cp-mode",
"round-robin-split",
"--enable-prefill-cp",
"--cp-strategy",
"interleave",
"--chunked-prefill-size",
"16384",
"--mem-fraction-static",
test/manual/dsv4/test_gb300_flash.py
+3 -3
View File
@@ -87,9 +87,9 @@ class TestGB300FlashCP(DSV4FlashAime25TestBase):
"4",
"--moe-a2a-backend",
"deepep",
"--enable-dsa-prefill-context-parallel",
"--dsa-prefill-cp-mode",
"round-robin-split",
"--enable-prefill-cp",
"--cp-strategy",
"interleave",
"--chunked-prefill-size",
"16384",
"--mem-fraction-static",
test/manual/dsv4/test_gb300_pro.py
+3 -3
View File
@@ -103,9 +103,9 @@ class TestGB300ProCP(DSV4ProAime25TestBase):
"4",
"--moe-a2a-backend",
"deepep",
"--enable-dsa-prefill-context-parallel",
"--dsa-prefill-cp-mode",
"round-robin-split",
"--enable-prefill-cp",
"--cp-strategy",
"interleave",
"--chunked-prefill-size",
"16384",
"--mem-fraction-static",
test/manual/dsv4/test_h200_fp8_flash.py
+3 -3
View File
@@ -98,9 +98,9 @@ class TestH200Fp8FlashCP(DSV4FlashAime25TestBase):
"4",
"--moe-a2a-backend",
"deepep",
"--enable-dsa-prefill-context-parallel",
"--dsa-prefill-cp-mode",
"round-robin-split",
"--enable-prefill-cp",
"--cp-strategy",
"interleave",
"--chunked-prefill-size",
"16384",
"--mem-fraction-static",
test/manual/test_kv_events.py
+3 -1
View File
@@ -329,7 +329,9 @@ class TestKvEvents(CustomTestCase):
2,
"--moe-dp-size",
2,
"--enable-prefill-context-parallel",
"--enable-prefill-cp",
"--cp-strategy",
"zigzag",
"--trust-remote-code",
"--max-total-tokens",
4096,
test/manual/test_qwen3_235b.py
+2 -1
View File
@@ -80,7 +80,8 @@ class TestQwen3235BFP8(unittest.TestCase):
"--moe-dp-size=2",
"--attn-cp-size=2",
"--ep-size=4",
"--enable-prefill-context-parallel",
"--enable-prefill-cp",
"--cp-strategy=zigzag",
]
MTP_ARGS = [