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

test/registered/model_loading/test_external_models.py

@@ -6,7 +6,6 @@ from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
 from sglang.test.test_utils import CustomTestCase
 
 register_cuda_ci(est_time=30, suite="stage-b-test-small-1-gpu")
-register_cuda_ci(est_time=26, suite="stage-b-test-small-1-gpu-5090")
 register_amd_ci(est_time=45, suite="stage-b-test-small-1-gpu-amd")
 
 

test/registered/model_loading/test_modelopt_export.py

@@ -19,7 +19,7 @@ from sglang.srt.configs.model_config import ModelConfig
 from sglang.srt.model_loader.loader import ModelOptModelLoader
 from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
 
-register_cuda_ci(est_time=9, suite="stage-b-test-small-1-gpu")
+register_cuda_ci(est_time=9, suite="stage-b-test-large-1-gpu")
 register_amd_ci(est_time=9, suite="stage-b-test-small-1-gpu-amd")
 
 # Note: PYTHONPATH=python should be set when running tests

test/registered/model_loading/test_modelopt_loader.py

@@ -26,7 +26,6 @@ CALIBRATION_NUM_SAMPLES = 512
 DEFAULT_DEVICE = "cuda:0"
 
 register_cuda_ci(est_time=11, suite="stage-b-test-small-1-gpu")
-register_cuda_ci(est_time=8, suite="stage-b-test-small-1-gpu-5090")
 
 
 class TestModelOptModelLoader(CustomTestCase):

test/registered/model_loading/test_utils_update_weights.py

@@ -12,7 +12,7 @@ from sglang.srt.weight_sync.utils import update_weights
 from sglang.test.ci.ci_register import register_cuda_ci
 from sglang.test.test_utils import DEFAULT_SMALL_MODEL_NAME_FOR_TEST
 
-register_cuda_ci(est_time=29, suite="stage-b-test-small-1-gpu")
+register_cuda_ci(est_time=29, suite="stage-b-test-large-1-gpu")
 
 
 class AsyncEngine(Engine):