diff --git a/test/registered/spec/eagle/test_eagle_constrained_decoding.py b/test/registered/spec/eagle/test_eagle_constrained_decoding.py index 6478ce0b3..85e7d75e7 100644 --- a/test/registered/spec/eagle/test_eagle_constrained_decoding.py +++ b/test/registered/spec/eagle/test_eagle_constrained_decoding.py @@ -2,7 +2,7 @@ import unittest from sglang.srt.environ import envs 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.kits.json_constrained_kit import JSONConstrainedMixin from sglang.test.kits.regex_constrained_kit import RegexConstrainedMixin from sglang.test.test_utils import ( @@ -15,6 +15,7 @@ from sglang.test.test_utils import ( ) register_cuda_ci(est_time=116, stage="base-b", runner_config="1-gpu-large") +register_amd_ci(est_time=165, stage="stage-b", runner_config="1-gpu-large-amd") class TestEagleConstrainedDecoding(