fix: increasing H200 test timeout (#15600)

This commit is contained in:
Douglas Yang
2025-12-23 01:00:37 -08:00
committed by GitHub
parent 66902e0f1b
commit f9dd90ac35
5 changed files with 27 additions and 8 deletions
@@ -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).