From 0d99d91e493855cd74b50641630b39d6154df20d Mon Sep 17 00:00:00 2001 From: Liangsheng Yin Date: Tue, 4 Aug 2026 15:53:01 -0700 Subject: [PATCH] [CI] Make B200 base-b suites single-GPU as prep for 1-gpu B200 runners (#33605) --- test/registered/kernels/ops/kimi_k3/test_collectives.py | 2 +- test/registered/spec/eagle/test_deepseek_v3_fp4_mtp_small.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/test/registered/kernels/ops/kimi_k3/test_collectives.py b/test/registered/kernels/ops/kimi_k3/test_collectives.py index d88892523..dcaca8d5f 100644 --- a/test/registered/kernels/ops/kimi_k3/test_collectives.py +++ b/test/registered/kernels/ops/kimi_k3/test_collectives.py @@ -23,7 +23,7 @@ from sglang.srt.distributed.device_communicators.custom_all_reduce_v2 import ( from sglang.test.ci.ci_register import register_cuda_ci from sglang.test.kernels.utils import multigpu_pytest_main -register_cuda_ci(est_time=240, stage="base-b-kernel-unit", runner_config="4-gpu-b200") +register_cuda_ci(est_time=240, stage="base-c", runner_config="4-gpu-b200") register_cuda_ci(est_time=480, suite="nightly-8-gpu-b200", nightly=True) _HIDDEN_SIZE = 7168 diff --git a/test/registered/spec/eagle/test_deepseek_v3_fp4_mtp_small.py b/test/registered/spec/eagle/test_deepseek_v3_fp4_mtp_small.py index e28dcbc2c..346886491 100644 --- a/test/registered/spec/eagle/test_deepseek_v3_fp4_mtp_small.py +++ b/test/registered/spec/eagle/test_deepseek_v3_fp4_mtp_small.py @@ -15,7 +15,7 @@ from sglang.test.test_utils import ( write_github_step_summary, ) -register_cuda_ci(est_time=340, stage="base-b", runner_config="4-gpu-b200") +register_cuda_ci(est_time=340, stage="base-c", runner_config="4-gpu-b200") FULL_DEEPSEEK_V3_FP4_MODEL_PATH = "nvidia/DeepSeek-V3-0324-FP4" SERVER_LAUNCH_TIMEOUT = 1200