[AMD] Fix CI - unit-test-backend-1-gpu-amd-mi35x and unit-test-backend-2-gpu-amd, stage-b-test-small-1-gpu-amd (#16675)

Co-authored-by: Bingxu Chen <bingxche@amd.com>
This commit is contained in:
YC Tseng
2026-01-08 10:16:27 -08:00
committed by GitHub
co-authored by Bingxu Chen
parent fbc24886dd
commit 4e999404c3
60 changed files with 106 additions and 143 deletions
+1 -1
View File
@@ -25,7 +25,7 @@ from sglang.test.lora_utils import (
from sglang.test.test_utils import CustomTestCase, is_in_ci
register_cuda_ci(est_time=82, suite="stage-b-test-small-1-gpu")
register_amd_ci(est_time=82, suite="stage-b-test-small-1-gpu")
register_amd_ci(est_time=82, suite="stage-b-test-small-1-gpu-amd")
class TestLoRA(CustomTestCase):
+1 -1
View File
@@ -32,7 +32,7 @@ from sglang.test.test_utils import CustomTestCase, is_in_ci
register_cuda_ci(est_time=200, suite="stage-b-test-small-1-gpu")
register_amd_ci(
est_time=200,
suite="stage-b-test-small-1-gpu",
suite="stage-b-test-small-1-gpu-amd",
disabled="see https://github.com/sgl-project/sglang/issues/13107",
)
+1 -1
View File
@@ -24,7 +24,7 @@ from sglang.test.runners import SRTRunner
from sglang.test.test_utils import CustomTestCase
register_cuda_ci(est_time=224, suite="stage-b-test-small-1-gpu")
register_amd_ci(est_time=224, suite="stage-b-test-small-1-gpu")
register_amd_ci(est_time=224, suite="stage-b-test-small-1-gpu-amd")
PROMPTS = [
"AI is a field of computer science focused on",
@@ -25,7 +25,7 @@ from sglang.test.lora_utils import (
from sglang.test.test_utils import CustomTestCase, is_in_ci
register_cuda_ci(est_time=60, suite="stage-b-test-small-1-gpu")
register_amd_ci(est_time=60, suite="stage-b-test-small-1-gpu")
register_amd_ci(est_time=60, suite="stage-b-test-small-1-gpu-amd")
# All prompts are used at once in a batch.
PROMPTS = [