From 60f502a4fd28bc11ff8be8dbf5cfa09cb0205978 Mon Sep 17 00:00:00 2001
From: Michael <13900043+michaelzhang-ai@users.noreply.github.com>
Date: Tue, 7 Jul 2026 14:46:22 -0700
Subject: [PATCH] [AMD] Register 2 hardware-agnostic 1-GPU PR tests for AMD CI
 (#30207)

---
 test/registered/lora/test_moe_lora_info.py                     | 3 ++-
 .../registered/unit/managers/test_customized_info_streaming.py | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/test/registered/lora/test_moe_lora_info.py b/test/registered/lora/test_moe_lora_info.py
index 44dd871c2..6ed49ef5f 100644
--- a/test/registered/lora/test_moe_lora_info.py
+++ b/test/registered/lora/test_moe_lora_info.py
@@ -4,9 +4,10 @@ import pytest
 import torch
 
 from sglang.srt.lora.backend.base_backend import _compute_moe_lora_info
-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=5, stage="base-b", runner_config="1-gpu-small")
+register_amd_ci(est_time=5, stage="stage-b", runner_config="1-gpu-small-amd")
 
 
 def _expected_adapter_enabled(
diff --git a/test/registered/unit/managers/test_customized_info_streaming.py b/test/registered/unit/managers/test_customized_info_streaming.py
index fc6bf5a9a..71c814725 100644
--- a/test/registered/unit/managers/test_customized_info_streaming.py
+++ b/test/registered/unit/managers/test_customized_info_streaming.py
@@ -8,7 +8,7 @@ import torch
 from sglang.srt.entrypoints.engine import Engine
 from sglang.srt.layers.sampler import Sampler, register_sampler_backend
 from sglang.srt.managers.scheduler import run_scheduler_process
-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.mock_model.utils import MOCK_MODEL_PATH
 from sglang.test.test_utils import CustomTestCase
 
@@ -17,6 +17,7 @@ if TYPE_CHECKING:
     from sglang.srt.sampling.sampling_batch_info import SamplingBatchInfo
 
 register_cuda_ci(est_time=120, stage="base-b", runner_config="1-gpu-small")
+register_amd_ci(est_time=120, stage="stage-b", runner_config="1-gpu-small-amd")
 
 
 CUSTOMIZED_INFO_FIELD = "sampled_token_ids_copy"