fix: increasing H200 test timeout (#15600)
This commit is contained in:
@@ -6,7 +6,7 @@ from sglang.test.performance_test_runner import PerformanceTestParams
|
||||
from sglang.test.run_combined_tests import run_combined_tests
|
||||
from sglang.test.test_utils import ModelLaunchSettings, is_blackwell_system
|
||||
|
||||
register_cuda_ci(est_time=8000, suite="nightly-8-gpu-common", nightly=True)
|
||||
register_cuda_ci(est_time=18000, suite="nightly-8-gpu-common", nightly=True)
|
||||
|
||||
DEEPSEEK_V32_MODEL_PATH = "deepseek-ai/DeepSeek-V3.2"
|
||||
|
||||
@@ -142,13 +142,13 @@ class TestDeepseekV32Unified(unittest.TestCase):
|
||||
accuracy_params=AccuracyTestParams(
|
||||
dataset="gsm8k", baseline_accuracy=GSM8K_BASELINE
|
||||
),
|
||||
performance_params=PerformanceTestParams(
|
||||
batch_sizes=[1, 8, 16, 64],
|
||||
profile_dir="performance_profiles_deepseek_v32_nsa",
|
||||
),
|
||||
performance_params=None,
|
||||
)
|
||||
|
||||
@unittest.skipIf(not is_blackwell_system(), "Requires B200")
|
||||
@unittest.skipIf(
|
||||
not is_blackwell_system(),
|
||||
"Hardware agnostic - just using B200 for efficiency reasons",
|
||||
)
|
||||
def test_deepseek_v32_b200(self):
|
||||
"""Test DeepSeek V3.2 with GPQA evaluation using thinking mode (B200 only).
|
||||
|
||||
|
||||
Reference in New Issue
Block a user