[PP] Add pipeline parallelism (#5724)

2025-04-30 18:18:07 -07:00
parent e97e57e699
commit 11383cec3c
25 changed files with 1150 additions and 308 deletions
--- a/test/srt/run_suite.py
+++ b/test/srt/run_suite.py
@@ -96,6 +96,8 @@ suites = {
    "per-commit-8-gpu": [
        TestFile("test_local_attn.py", 250),
        TestFile("test_full_deepseek_v3.py", 250),
+        TestFile("test_fa3.py", 30),
+        TestFile("test_pp_single_node.py", 150),
    ],
    "nightly": [
        TestFile("test_nightly_gsm8k_eval.py"),