diff --git a/test/registered/chunked_prefill/test_scripted_core_1gpu.py b/test/registered/chunked_prefill/test_scripted_core_1gpu.py index c6f36d29a..4929b858c 100644 --- a/test/registered/chunked_prefill/test_scripted_core_1gpu.py +++ b/test/registered/chunked_prefill/test_scripted_core_1gpu.py @@ -1,6 +1,6 @@ import unittest -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.test_case import ScriptedTestCase from sglang.test.scripted_runtime_chunked_helpers import ( @@ -11,6 +11,7 @@ from sglang.test.scripted_runtime_chunked_helpers import ( ) register_cuda_ci(est_time=300, stage="extra-a", runner_config="1-gpu-small") +register_amd_ci(est_time=198, stage="extra-a", runner_config="1-gpu-small-amd") _CHUNK_SIZE = 64