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