minke.yu/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
test/registered/distributed/test_disaggregation_hybrid_attention.py
+1 -7
View File
@@ -12,9 +12,7 @@ from sglang.test.test_utils import (
popen_launch_pd_server, popen_launch_pd_server,
) )
register_cuda_ci( register_cuda_ci(est_time=500, suite="stage-c-test-8-gpu-h200")
est_time=400, suite="stage-c-test-8-gpu-h200", disabled="TCP fallback flaky"
)
@unittest.skipIf(is_in_ci(), "Temporarily disable the flaky test.") @unittest.skipIf(is_in_ci(), "Temporarily disable the flaky test.")
@@ -158,10 +156,6 @@ class TestDisaggregationHybridAttentionMambaExtraBuffer(PDDisaggregationServerBa
self.assertGreater(metrics["accuracy"], 0.93) 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): class TestDisaggregationHybridAttentionMambaDPDecode(PDDisaggregationServerBase):
"""Test with prefill tp=2 and decode tp=2/dp=2 with dp-attention enabled.""" """Test with prefill tp=2 and decode tp=2/dp=2 with dp-attention enabled."""