Add 5090 dry run stage to PR test workflow (#17022)
This commit is contained in:
@@ -30,6 +30,7 @@ from sglang.test.lora_utils import (
|
||||
from sglang.test.test_utils import CustomTestCase, is_in_ci
|
||||
|
||||
register_cuda_ci(est_time=200, suite="stage-b-test-small-1-gpu")
|
||||
register_cuda_ci(est_time=137, suite="stage-b-test-small-1-gpu-5090")
|
||||
register_amd_ci(
|
||||
est_time=200,
|
||||
suite="stage-b-test-small-1-gpu-amd",
|
||||
|
||||
@@ -24,6 +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_cuda_ci(est_time=215, suite="stage-b-test-small-1-gpu-5090")
|
||||
register_amd_ci(est_time=224, suite="stage-b-test-small-1-gpu-amd")
|
||||
|
||||
PROMPTS = [
|
||||
|
||||
@@ -41,6 +41,10 @@ register_cuda_ci(
|
||||
est_time=150,
|
||||
suite="stage-b-test-small-1-gpu",
|
||||
)
|
||||
register_cuda_ci(
|
||||
est_time=76,
|
||||
suite="stage-b-test-small-1-gpu-5090",
|
||||
)
|
||||
|
||||
from sglang.test.test_utils import DEFAULT_PORT_FOR_SRT_TEST_RUNNER, CustomTestCase
|
||||
|
||||
|
||||
Reference in New Issue
Block a user