diff --git a/test/registered/8-gpu-models/test_deepseek_v31.py b/test/registered/8-gpu-models/test_deepseek_v31.py index 3ab39a185..56757c88f 100644 --- a/test/registered/8-gpu-models/test_deepseek_v31.py +++ b/test/registered/8-gpu-models/test_deepseek_v31.py @@ -4,7 +4,7 @@ from sglang.test.accuracy_test_runner import AccuracyTestParams from sglang.test.ci.ci_register import register_cuda_ci 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 +from sglang.test.test_utils import ModelLaunchSettings # Runs on both H200 and B200 via nightly-8-gpu-common suite register_cuda_ci(est_time=12000, suite="nightly-8-gpu-common", nightly=True) @@ -12,7 +12,6 @@ register_cuda_ci(est_time=12000, suite="nightly-8-gpu-common", nightly=True) DEEPSEEK_V31_MODEL_PATH = "deepseek-ai/DeepSeek-V3.1" -@unittest.skipIf(not is_blackwell_system(), "Requires B200") class TestDeepseekV31Unified(unittest.TestCase): """Unified test class for DeepSeek-V3.1 performance and accuracy.