From 5cfa901b5739fdbb54840abd05150f3294fdca41 Mon Sep 17 00:00:00 2001
From: Alison Shao <54658187+alisonshao@users.noreply.github.com>
Date: Mon, 5 Jan 2026 22:29:22 -0800
Subject: [PATCH] ci: migrate Constrained Decoding tests to
 test/registered/constrained/ (#16421)

---
 .../constrained_decoding}/test_constrained_decoding.py       | 5 +++++
 test/srt/run_suite.py                                        | 2 --
 2 files changed, 5 insertions(+), 2 deletions(-)
 rename test/{srt => registered/constrained_decoding}/test_constrained_decoding.py (89%)

diff --git a/test/srt/test_constrained_decoding.py b/test/registered/constrained_decoding/test_constrained_decoding.py
similarity index 89%
rename from test/srt/test_constrained_decoding.py
rename to test/registered/constrained_decoding/test_constrained_decoding.py
index 9b8b42a9e..8992e7074 100644
--- a/test/srt/test_constrained_decoding.py
+++ b/test/registered/constrained_decoding/test_constrained_decoding.py
@@ -1,3 +1,8 @@
+from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
+
+register_cuda_ci(est_time=111, suite="stage-b-test-small-1-gpu")
+register_amd_ci(est_time=179, suite="stage-b-test-small-1-gpu")
+
 import unittest
 
 from sglang.srt.utils import kill_process_tree
diff --git a/test/srt/run_suite.py b/test/srt/run_suite.py
index f293d0144..a0d108f7c 100644
--- a/test/srt/run_suite.py
+++ b/test/srt/run_suite.py
@@ -32,7 +32,6 @@ suites = {
         TestFile("test_abort.py", 131),
         TestFile("test_chunked_prefill.py", 312),
         TestFile("test_deterministic.py", 228),
-        TestFile("test_constrained_decoding.py", 111),
         TestFile("test_eval_fp8_accuracy.py", 250),
         TestFile("test_evs.py", 20),
         TestFile("test_external_models.py", 30),
@@ -182,7 +181,6 @@ suite_amd = {
         TestFile("test_abort.py", 51),
         TestFile("test_bench_typebaseddispatcher.py", 10),
         TestFile("test_chunked_prefill.py", 312),
-        TestFile("test_constrained_decoding.py", 120),
         TestFile("test_eval_fp8_accuracy.py", 303),
         TestFile("test_external_models.py", 45),
         TestFile("test_input_embeddings.py", 38),