[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:
@@ -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):
|
||||
|
||||
@@ -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",
|
||||
)
|
||||
|
||||
|
||||
@@ -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 = [
|
||||
|
||||
Reference in New Issue
Block a user