From 62ab09a47886c05a664113c7f080a913b20a2924 Mon Sep 17 00:00:00 2001 From: Michael <13900043+michaelzhang-ai@users.noreply.github.com> Date: Thu, 18 Jun 2026 16:33:23 -0700 Subject: [PATCH] [AMD] register 2 spec tests to stage-b-test-1-gpu-large-amd (batch-5) (#28558) --- test/registered/spec/eagle/test_adaptive_speculative.py | 3 ++- test/registered/spec/test_frozen_kv_mtp.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/test/registered/spec/eagle/test_adaptive_speculative.py b/test/registered/spec/eagle/test_adaptive_speculative.py index 1c33e0aa1..15add0fa0 100644 --- a/test/registered/spec/eagle/test_adaptive_speculative.py +++ b/test/registered/spec/eagle/test_adaptive_speculative.py @@ -7,7 +7,7 @@ from types import SimpleNamespace import requests from sglang.srt.utils import kill_process_tree -from sglang.test.ci.ci_register import register_cuda_ci +from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci from sglang.test.run_eval import run_eval from sglang.test.test_utils import ( DEFAULT_DRAFT_MODEL_EAGLE, @@ -19,6 +19,7 @@ from sglang.test.test_utils import ( ) register_cuda_ci(est_time=160, stage="base-b", runner_config="1-gpu-large") +register_amd_ci(est_time=240, suite="stage-b-test-1-gpu-large-amd") HIGH_ACCEPT_PROMPT = ( "Output exactly 128 new lines. " diff --git a/test/registered/spec/test_frozen_kv_mtp.py b/test/registered/spec/test_frozen_kv_mtp.py index c0d1705cb..3f9bbd035 100644 --- a/test/registered/spec/test_frozen_kv_mtp.py +++ b/test/registered/spec/test_frozen_kv_mtp.py @@ -5,7 +5,7 @@ from typing import Optional import requests from sglang.srt.utils import kill_process_tree -from sglang.test.ci.ci_register import register_cuda_ci +from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci from sglang.test.run_eval import run_eval from sglang.test.test_utils import ( DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH, @@ -17,6 +17,7 @@ from sglang.test.test_utils import ( ) register_cuda_ci(est_time=300, stage="base-b", runner_config="1-gpu-large") +register_amd_ci(est_time=450, suite="stage-b-test-1-gpu-large-amd") def get_server_info(base_url: str) -> dict: