chenchenghao/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

[CI] Bring back CI test for Mamba PD Disaggregation (#20675)

Browse Source
This commit is contained in:
Shangming Cai
2026-03-17 01:04:21 +08:00
committed by GitHub
parent 289cbcf482
commit a4528a5737
1 changed files with 1 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
test/registered/distributed/test_disaggregation_hybrid_attention.py
View File

@@ -12,9 +12,7 @@ from sglang.test.test_utils import (
popen_launch_pd_server,
)
register_cuda_ci(
est_time=400, suite="stage-c-test-8-gpu-h200", disabled="TCP fallback flaky"
)
register_cuda_ci(est_time=500, suite="stage-c-test-8-gpu-h200")
@unittest.skipIf(is_in_ci(), "Temporarily disable the flaky test.")
@@ -158,10 +156,6 @@ class TestDisaggregationHybridAttentionMambaExtraBuffer(PDDisaggregationServerBa
self.assertGreater(metrics["accuracy"], 0.93)
@unittest.skipIf(
is_in_ci(),
"Temporarily disable the flaky test: tcp fallback is not stable currently.",
)
class TestDisaggregationHybridAttentionMambaDPDecode(PDDisaggregationServerBase):
"""Test with prefill tp=2 and decode tp=2/dp=2 with dp-attention enabled."""