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),