[PD] Add support for different TP sizes per DP rank (#5922)

Signed-off-by: Shangming Cai <caishangming@linux.alibaba.com>
2025-05-13 04:55:42 +08:00
parent 983c663de6
commit f1c896007a
8 changed files with 396 additions and 119 deletions
--- a/test/srt/run_suite.py
+++ b/test/srt/run_suite.py
@@ -103,6 +103,7 @@ suites = {
        # TestFile("test_moe_deepep_eval_accuracy_large.py", 250),
        TestFile("test_disaggregation.py", 210),
        TestFile("test_local_attn.py", 250),
+        TestFile("test_disaggregation_different_tp.py", 210),
        TestFile("test_full_deepseek_v3.py", 250),
        TestFile("test_pp_single_node.py", 150),
    ],