[CI] fix amd 1 gpu basic test (#13551)

test/per_commit/test_srt_backend.py

@@ -1,7 +1,7 @@
 import unittest
 
 import sglang as sgl
-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_programs import (
     test_decode_int,
     test_decode_json_regex,
@@ -20,6 +20,7 @@ from sglang.test.test_programs import (
 from sglang.test.test_utils import DEFAULT_MODEL_NAME_FOR_TEST, CustomTestCase
 
 register_cuda_ci(est_time=80, suite="stage-a-test-1")
+register_amd_ci(est_time=120, suite="stage-a-test-1")
 
 
 class TestSRTBackend(CustomTestCase):

test/run_suite.py

@@ -8,11 +8,13 @@ from sglang.test.ci.ci_utils import TestFile, run_unittest_files
 HW_MAPPING = {
     "cpu": HWBackend.CPU,
     "cuda": HWBackend.CUDA,
+    "amd": HWBackend.AMD,
 }
 
 LABEL_MAPPING = {
-    HWBackend.CUDA: ["stage-a-test-1"],
     HWBackend.CPU: ["default"],
+    HWBackend.AMD: ["stage-a-test-1"],
+    HWBackend.CUDA: ["stage-a-test-1"],
 }
 
 
@@ -45,7 +47,7 @@ def main():
     parser.add_argument(
         "--hw",
         type=str,
-        choices=["cpu", "cuda"],
+        choices=["cpu", "cuda", "amd"],
         required=True,
         help="Hardware backend to run tests on.",
     )