[PD] Add simple unit test for disaggregation feature (#5654)

Signed-off-by: Shangming Cai <caishangming@linux.alibaba.com>
2025-05-11 13:35:27 +08:00
parent a823c6e834
commit 31d1f6e7f4
5 changed files with 241 additions and 0 deletions
--- a/test/srt/run_suite.py
+++ b/test/srt/run_suite.py
@@ -85,6 +85,9 @@ suites = {
        TestFile("test_w8a8_quantization.py", 46),
        TestFile("models/lora/test_lora_cuda_graph.py", 250),
    ],
+    "per-commit-pd": [
+        TestFile("test_disaggregation.py", 90),
+    ],
    "per-commit-2-gpu": [
        TestFile("models/lora/test_lora_tp.py", 116),
        TestFile("test_data_parallelism.py", 73),