From 3adfd0f34b4844f7e91d91956de7e9712df5c53c Mon Sep 17 00:00:00 2001 From: Michael <13900043+michaelzhang-ai@users.noreply.github.com> Date: Wed, 1 Jul 2026 12:57:48 -0700 Subject: [PATCH] [AMD] Register 3 unit mem_cache + utils tests for stage-b-test-1-gpu-small-amd (#29782) --- test/registered/unit/mem_cache/test_hiradix_cache_unit.py | 3 ++- test/registered/unit/mem_cache/test_triton_kernel_layout.py | 3 ++- test/registered/unit/utils/test_common.py | 3 ++- 3 files changed, 6 insertions(+), 3 deletions(-) diff --git a/test/registered/unit/mem_cache/test_hiradix_cache_unit.py b/test/registered/unit/mem_cache/test_hiradix_cache_unit.py index a83c59dfd..21c6e0851 100644 --- a/test/registered/unit/mem_cache/test_hiradix_cache_unit.py +++ b/test/registered/unit/mem_cache/test_hiradix_cache_unit.py @@ -14,10 +14,11 @@ from sglang.srt.mem_cache.hiradix_cache import HiRadixCache from sglang.srt.mem_cache.memory_pool import MHATokenToKVPool, ReqToTokenPool from sglang.srt.mem_cache.radix_cache import RadixKey from sglang.srt.server_args import ServerArgs, set_global_server_args_for_scheduler -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=15, stage="base-b", runner_config="1-gpu-small") +register_amd_ci(est_time=15, stage="stage-b", runner_config="1-gpu-small-amd") PAGE_SIZE = 2 diff --git a/test/registered/unit/mem_cache/test_triton_kernel_layout.py b/test/registered/unit/mem_cache/test_triton_kernel_layout.py index 224005692..f881c6ca8 100644 --- a/test/registered/unit/mem_cache/test_triton_kernel_layout.py +++ b/test/registered/unit/mem_cache/test_triton_kernel_layout.py @@ -23,11 +23,12 @@ import unittest import torch -from sglang.test.ci.ci_register import register_cuda_ci +from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci _HAS_CUDA = torch.cuda.is_available() register_cuda_ci(est_time=30, stage="base-b", runner_config="1-gpu-small") +register_amd_ci(est_time=5, stage="stage-b", runner_config="1-gpu-small-amd") @unittest.skipUnless(_HAS_CUDA, "Triton kernels require CUDA") diff --git a/test/registered/unit/utils/test_common.py b/test/registered/unit/utils/test_common.py index f38047601..f7a1209ea 100644 --- a/test/registered/unit/utils/test_common.py +++ b/test/registered/unit/utils/test_common.py @@ -8,10 +8,11 @@ from sglang.srt.utils.common import ( get_device_sm_nvidia_smi, get_nvidia_driver_version_str, ) -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=5, stage="base-b", runner_config="1-gpu-small") +register_amd_ci(est_time=5, stage="stage-b", runner_config="1-gpu-small-amd") @unittest.skipUnless(torch.cuda.is_available(), "requires CUDA")