From 190b15c8fe2a9399f0956fc8bdb86e17644c5b83 Mon Sep 17 00:00:00 2001 From: Michael <13900043+michaelzhang-ai@users.noreply.github.com> Date: Sat, 9 May 2026 07:01:58 +0800 Subject: [PATCH] [AMD] Register 8 CPU-bound unit tests for AMD 1-GPU PR CI (#24569) --- test/registered/unit/auto_benchmark/test_dataset_tools.py | 3 ++- test/registered/unit/auto_benchmark/test_run_candidate.py | 3 ++- test/registered/unit/auto_benchmark/test_search_tools.py | 3 ++- test/registered/unit/layers/test_conv_layer.py | 3 ++- test/registered/unit/lora/test_mem_pool_ep_unit.py | 3 ++- test/registered/unit/mem_cache/test_decode_radix_lock_ref.py | 3 ++- .../unit/mem_cache/test_radix_cache_slru_accuracy.py | 3 ++- test/registered/unit/models/test_llava.py | 3 ++- 8 files changed, 16 insertions(+), 8 deletions(-) diff --git a/test/registered/unit/auto_benchmark/test_dataset_tools.py b/test/registered/unit/auto_benchmark/test_dataset_tools.py index cab0aaee3..08a52ce13 100644 --- a/test/registered/unit/auto_benchmark/test_dataset_tools.py +++ b/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): diff --git a/test/registered/unit/auto_benchmark/test_run_candidate.py b/test/registered/unit/auto_benchmark/test_run_candidate.py index e172fac3e..99e7a81fa 100644 --- a/test/registered/unit/auto_benchmark/test_run_candidate.py +++ b/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): diff --git a/test/registered/unit/auto_benchmark/test_search_tools.py b/test/registered/unit/auto_benchmark/test_search_tools.py index 24af47d0b..3e8c0ae88 100644 --- a/test/registered/unit/auto_benchmark/test_search_tools.py +++ b/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): diff --git a/test/registered/unit/layers/test_conv_layer.py b/test/registered/unit/layers/test_conv_layer.py index 2dedbc3b5..c892b4fcd 100644 --- a/test/registered/unit/layers/test_conv_layer.py +++ b/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 diff --git a/test/registered/unit/lora/test_mem_pool_ep_unit.py b/test/registered/unit/lora/test_mem_pool_ep_unit.py index 56a68dc98..a0a3de607 100644 --- a/test/registered/unit/lora/test_mem_pool_ep_unit.py +++ b/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 diff --git a/test/registered/unit/mem_cache/test_decode_radix_lock_ref.py b/test/registered/unit/mem_cache/test_decode_radix_lock_ref.py index 9415e7258..4dca23453 100644 --- a/test/registered/unit/mem_cache/test_decode_radix_lock_ref.py +++ b/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 diff --git a/test/registered/unit/mem_cache/test_radix_cache_slru_accuracy.py b/test/registered/unit/mem_cache/test_radix_cache_slru_accuracy.py index 46dc341a3..95531c984 100644 --- a/test/registered/unit/mem_cache/test_radix_cache_slru_accuracy.py +++ b/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): diff --git a/test/registered/unit/models/test_llava.py b/test/registered/unit/models/test_llava.py index 4ebaecf8e..91f7f34f4 100644 --- a/test/registered/unit/models/test_llava.py +++ b/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: