[UnifiedTree]: move some kl test from base to extra stage (#28627)

Co-authored-by: ispobock <ispobaoke@gmail.com>
This commit is contained in:
Zhangheng
2026-06-18 01:39:46 -07:00
committed by GitHub
co-authored by ispobock
parent fa71064147
commit 867707f1f2
4 changed files with 4 additions and 4 deletions
@@ -13,7 +13,7 @@ from sglang.test.test_utils import (
popen_launch_server,
)
register_cuda_ci(est_time=400, stage="base-c", runner_config="4-gpu-h100")
register_cuda_ci(est_time=900, stage="base-c", runner_config="4-gpu-h100")
QWEN3_32B_MODEL = "Qwen/Qwen3-32B"
@@ -11,7 +11,7 @@ from sglang.test.test_utils import (
popen_launch_server,
)
register_cuda_ci(est_time=400, stage="base-c", runner_config="4-gpu-h100")
register_cuda_ci(est_time=950, stage="extra-b", runner_config="4-gpu-h100")
QWEN3_30B_MODEL = "Qwen/Qwen3-30B-A3B-FP8"
@@ -21,7 +21,7 @@ from sglang.test.test_utils import (
DSV4_FLASH_MODEL = "sgl-project/DeepSeek-V4-Flash-FP8"
DSV4_FLASH_LAUNCH_TIMEOUT = 3600
register_cuda_ci(est_time=1200, stage="base-c", runner_config="4-gpu-h100")
register_cuda_ci(est_time=1000, stage="extra-b", runner_config="4-gpu-h100")
def _assert_dsv4_decode_cached_tokens(result, history_len, output_len, label):
@@ -20,7 +20,7 @@ from sglang.test.test_utils import (
popen_launch_server,
)
register_cuda_ci(est_time=768, stage="base-c", runner_config="4-gpu-h100")
register_cuda_ci(est_time=800, stage="extra-b", runner_config="4-gpu-h100")
MAMBA_MODEL = "Qwen/Qwen3-Next-80B-A3B-Instruct-FP8"
MAMBA_CHUNK_SIZE = 64