[CI] Reorganize stage-b 1-GPU tests for 5090 compatibility (#16826)

test/registered/dllm/test_llada2_mini.py

@@ -1,6 +1,6 @@
 from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
 
-register_cuda_ci(est_time=181, suite="stage-b-test-small-1-gpu")
+register_cuda_ci(est_time=181, suite="stage-b-test-large-1-gpu")
 register_amd_ci(est_time=330, suite="stage-b-test-small-1-gpu-amd")
 
 import unittest

test/registered/dllm/test_llada2_mini_amd.py

@@ -1,7 +1,3 @@
-from sglang.test.ci.ci_register import register_amd_ci
-
-register_amd_ci(est_time=520, suite="stage-b-test-small-1-gpu-amd")
-
 """
 Test LLaDA2 (Diffusion Language Model) on AMD GPUs.
 
@@ -12,6 +8,7 @@ import unittest
 from types import SimpleNamespace
 
 from sglang.srt.utils import kill_process_tree
+from sglang.test.ci.ci_register import register_amd_ci
 from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
 from sglang.test.send_one import BenchArgs, send_one_prompt
 from sglang.test.test_utils import (
@@ -23,6 +20,8 @@ from sglang.test.test_utils import (
     write_github_step_summary,
 )
 
+register_amd_ci(est_time=520, suite="stage-b-test-small-1-gpu-amd")
+
 
 class TestLLaDA2MiniAMD(CustomTestCase):
     @classmethod