[CI] Make B200 base-b suites single-GPU as prep for 1-gpu B200 runners (#33605)
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user