[AMD] Register 8 CPU-bound unit tests for AMD 1-GPU PR CI (#24569)

test/registered/unit/auto_benchmark/test_dataset_tools.py

@@ -13,9 +13,10 @@ from auto_benchmark import AutoBenchmarkTestCase
 
 from sglang.auto_benchmark_lib import infer_backend, prepare_dataset
 from sglang.benchmark.datasets.autobench import sample_autobench_requests
-from sglang.test.ci.ci_register import register_cuda_ci
+from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
 
 register_cuda_ci(est_time=6, suite="stage-b-test-1-gpu-small")
+register_amd_ci(est_time=6, suite="stage-b-test-1-gpu-small-amd")
 
 
 class TestAutoBenchmarkDatasetTools(AutoBenchmarkTestCase):

test/registered/unit/auto_benchmark/test_run_candidate.py

@@ -12,9 +12,10 @@ if str(PARENT_DIR) not in sys.path:
 from auto_benchmark import AutoBenchmarkTestCase
 
 from sglang.auto_benchmark_lib import SearchDeadlineExceeded, run_candidate
-from sglang.test.ci.ci_register import register_cuda_ci
+from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
 
 register_cuda_ci(est_time=6, suite="stage-b-test-1-gpu-small")
+register_amd_ci(est_time=6, suite="stage-b-test-1-gpu-small-amd")
 
 
 class TestAutoBenchmarkRunCandidate(AutoBenchmarkTestCase):

test/registered/unit/auto_benchmark/test_search_tools.py

@@ -26,9 +26,10 @@ from sglang.auto_benchmark_lib import (
     rendered_launch_command,
     resolve_max_candidates,
 )
-from sglang.test.ci.ci_register import register_cuda_ci
+from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
 
 register_cuda_ci(est_time=6, suite="stage-b-test-1-gpu-small")
+register_amd_ci(est_time=6, suite="stage-b-test-1-gpu-small-amd")
 
 
 class TestAutoBenchmarkSearchTools(AutoBenchmarkTestCase):

test/registered/unit/layers/test_conv_layer.py

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

test/registered/unit/lora/test_mem_pool_ep_unit.py

@@ -13,10 +13,11 @@ Usage:
     python -m pytest test/registered/unit/lora/test_mem_pool_ep_unit.py -v
 """
 
-from sglang.test.ci.ci_register import register_cuda_ci
+from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
 
 # CPU-only unit test; no CUDA/distributed dependencies.
 register_cuda_ci(est_time=9, suite="stage-b-test-1-gpu-small")
+register_amd_ci(est_time=9, suite="stage-b-test-1-gpu-small-amd")
 
 import types
 import unittest

test/registered/unit/mem_cache/test_decode_radix_lock_ref.py

@@ -20,9 +20,10 @@ Usage:
     python -m pytest test/registered/unit/mem_cache/test_decode_radix_lock_ref.py -v
 """
 
-from sglang.test.ci.ci_register import register_cuda_ci
+from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
 
 register_cuda_ci(est_time=10, suite="stage-b-test-1-gpu-small")
+register_amd_ci(est_time=10, suite="stage-b-test-1-gpu-small-amd")
 
 import unittest
 from unittest.mock import MagicMock

test/registered/unit/mem_cache/test_radix_cache_slru_accuracy.py

@@ -11,9 +11,10 @@ from sglang.srt.mem_cache.base_prefix_cache import (
 from sglang.srt.mem_cache.cache_init_params import CacheInitParams
 from sglang.srt.mem_cache.memory_pool import MHATokenToKVPool, ReqToTokenPool
 from sglang.srt.mem_cache.radix_cache import RadixCache, RadixKey
-from sglang.test.ci.ci_register import register_cuda_ci
+from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
 
 register_cuda_ci(est_time=8, suite="stage-b-test-1-gpu-small")
+register_amd_ci(est_time=8, suite="stage-b-test-1-gpu-small-amd")
 
 
 class TestSLRUAccuracy(unittest.TestCase):

test/registered/unit/models/test_llava.py

@@ -2,10 +2,11 @@ import unittest
 from unittest.mock import patch
 
 from sglang.srt.models.llava import AutoModel, LlavaForConditionalGeneration
-from sglang.test.ci.ci_register import register_cuda_ci
+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=9, suite="stage-b-test-1-gpu-small")
+register_amd_ci(est_time=9, suite="stage-b-test-1-gpu-small-amd")
 
 
 class PixtralVisionConfig: