[AMD] Register 3 unit mem_cache + utils tests for stage-b-test-1-gpu-small-amd (#29782)

2026-07-01 12:57:48 -07:00
parent 8361561c24
commit 3adfd0f34b
3 changed files with 6 additions and 3 deletions
@@ -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")