Fix hybrid attention PD Disaggregation test (#17099)
Signed-off-by: Shangming Cai <csmthu@gmail.com>
This commit is contained in:
@@ -19,7 +19,7 @@ suites = {
|
||||
"per-commit-8-gpu-h200": [
|
||||
TestFile("test_deepseek_v3_basic.py", 275),
|
||||
TestFile("test_deepseek_v3_mtp.py", 275),
|
||||
TestFile("test_disaggregation_hybrid_attention.py", 600),
|
||||
TestFile("test_disaggregation_hybrid_attention.py", 400),
|
||||
TestFile("models/test_kimi_k2_models.py", 200),
|
||||
TestFile("test_deepseek_v32_basic.py", 360),
|
||||
TestFile("test_deepseek_v32_mtp.py", 360),
|
||||
|
||||
@@ -7,10 +7,12 @@ from sglang.test.server_fixtures.disaggregation_fixture import (
|
||||
)
|
||||
from sglang.test.test_utils import (
|
||||
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
||||
is_in_ci,
|
||||
popen_launch_pd_server,
|
||||
)
|
||||
|
||||
|
||||
@unittest.skipIf(is_in_ci(), "Temporarily disable the flaky test.")
|
||||
class TestDisaggregationHybridAttentionMamba(PDDisaggregationServerBase):
|
||||
@classmethod
|
||||
def setUpClass(cls):
|
||||
|
||||
Reference in New Issue
Block a user