From 2210155a45105c6a08ca04f7efcd77f547c8c1e0 Mon Sep 17 00:00:00 2001
From: Alison Shao <54658187+alisonshao@users.noreply.github.com>
Date: Tue, 6 Jan 2026 09:44:26 -0800
Subject: [PATCH] ci: migrate DLLM tests to test/registered/dllm/ (#16420)

---
 test/{srt => registered}/dllm/test_llada2_mini.py     | 4 ++++
 test/{srt => registered}/dllm/test_llada2_mini_amd.py | 4 ++++
 test/srt/run_suite.py                                 | 2 --
 3 files changed, 8 insertions(+), 2 deletions(-)
 rename test/{srt => registered}/dllm/test_llada2_mini.py (94%)
 rename test/{srt => registered}/dllm/test_llada2_mini_amd.py (95%)

diff --git a/test/srt/dllm/test_llada2_mini.py b/test/registered/dllm/test_llada2_mini.py
similarity index 94%
rename from test/srt/dllm/test_llada2_mini.py
rename to test/registered/dllm/test_llada2_mini.py
index 0763977cb..ee72329a3 100644
--- a/test/srt/dllm/test_llada2_mini.py
+++ b/test/registered/dllm/test_llada2_mini.py
@@ -1,3 +1,7 @@
+from sglang.test.ci.ci_register import register_cuda_ci
+
+register_cuda_ci(est_time=520, suite="stage-b-test-small-1-gpu")
+
 import unittest
 from types import SimpleNamespace
 
diff --git a/test/srt/dllm/test_llada2_mini_amd.py b/test/registered/dllm/test_llada2_mini_amd.py
similarity index 95%
rename from test/srt/dllm/test_llada2_mini_amd.py
rename to test/registered/dllm/test_llada2_mini_amd.py
index fc0392392..50f378229 100644
--- a/test/srt/dllm/test_llada2_mini_amd.py
+++ b/test/registered/dllm/test_llada2_mini_amd.py
@@ -1,3 +1,7 @@
+from sglang.test.ci.ci_register import register_amd_ci
+
+register_amd_ci(est_time=520, suite="stage-b-test-small-1-gpu")
+
 """
 Test LLaDA2 (Diffusion Language Model) on AMD GPUs.
 
diff --git a/test/srt/run_suite.py b/test/srt/run_suite.py
index 9e990555b..9eb08c616 100644
--- a/test/srt/run_suite.py
+++ b/test/srt/run_suite.py
@@ -28,7 +28,6 @@ suites = {
         TestFile("openai_server/validation/test_openai_server_ignore_eos.py", 6),
         TestFile("openai_server/validation/test_request_length_validation.py", 38),
         TestFile("ops/test_repeat_interleave.py", 60),
-        TestFile("dllm/test_llada2_mini.py", 520),
         TestFile("test_abort.py", 131),
         TestFile("test_chunked_prefill.py", 312),
         TestFile("test_deterministic.py", 228),
@@ -156,7 +155,6 @@ suite_amd = {
         # TestFile("lora/test_lora_backend.py", 99), # Disabled temporarily, see https://github.com/sgl-project/sglang/issues/13107
         # TestFile("lora/test_lora_cuda_graph.py", 250), # Disabled temporarily, see https://github.com/sgl-project/sglang/issues/13107
         # TestFile("lora/test_lora_qwen3.py", 97), # Disabled temporarily, see https://github.com/sgl-project/sglang/issues/13107
-        TestFile("dllm/test_llada2_mini_amd.py", 520),
         TestFile("openai_server/basic/test_openai_embedding.py", 141),
         TestFile("openai_server/basic/test_openai_server.py", 149),
         TestFile("openai_server/basic/test_protocol.py", 10),