diff --git a/test/registered/scripted_runtime/test_scripted_runtime_core.py b/test/registered/scripted_runtime/test_scripted_runtime_core.py index 220864cb3..b30521973 100644 --- a/test/registered/scripted_runtime/test_scripted_runtime_core.py +++ b/test/registered/scripted_runtime/test_scripted_runtime_core.py @@ -1,7 +1,7 @@ import unittest from sglang.srt.managers.schedule_batch import FINISH_ABORT -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.scripted_runtime.context import ScriptedContext from sglang.test.scripted_runtime.http_server import ScriptedHttpServer from sglang.test.scripted_runtime.req_handle import ScriptedReqHandle @@ -17,6 +17,7 @@ from sglang.test.scripted_runtime_chunked_helpers import ( from sglang.test.test_utils import CustomTestCase register_cuda_ci(est_time=460, stage="base-b", runner_config="1-gpu-small") +register_amd_ci(est_time=460, stage="stage-b", runner_config="1-gpu-small-amd") _CHUNK_SIZE = 64 diff --git a/test/registered/utils/test_phase_checker.py b/test/registered/utils/test_phase_checker.py index 03de94d33..d7ee8da65 100644 --- a/test/registered/utils/test_phase_checker.py +++ b/test/registered/utils/test_phase_checker.py @@ -11,10 +11,11 @@ from enum import IntEnum import torch from sglang.srt.utils.phase_checker import SimplePhaseChecker -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.test_utils import CustomTestCase register_cuda_ci(est_time=120, stage="base-b", runner_config="1-gpu-small") +register_amd_ci(est_time=120, stage="stage-b", runner_config="1-gpu-small-amd") class _Phase(IntEnum):