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

This commit is contained in:
Liangsheng Yin
2025-11-19 11:47:41 +08:00
committed by GitHub
parent 9a1a9a4209
commit f7be98e113
3 changed files with 11 additions and 8 deletions
+4 -2
View File
@@ -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.",
)