From 4ba8634780d5d37ffd6aa0265ec5c32ac0a304ad Mon Sep 17 00:00:00 2001 From: Michael <13900043+michaelzhang-ai@users.noreply.github.com> Date: Thu, 25 Jun 2026 01:50:08 -0700 Subject: [PATCH] [AMD] Register scripted-core chunked-prefill test for AMD extra-a CI (#29058) --- test/registered/chunked_prefill/test_scripted_core_1gpu.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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