Merge performance/accuracy test suites into regular stage-b suites (#17609)
This commit is contained in:
@@ -19,8 +19,8 @@ from sglang.test.test_utils import (
|
||||
write_github_step_summary,
|
||||
)
|
||||
|
||||
register_cuda_ci(est_time=300, suite="stage-b-test-small-1-gpu-accuracy")
|
||||
register_amd_ci(est_time=300, suite="stage-b-test-small-1-gpu-accuracy-amd")
|
||||
register_cuda_ci(est_time=300, suite="stage-b-test-small-1-gpu")
|
||||
register_amd_ci(est_time=300, suite="stage-b-test-small-1-gpu-amd")
|
||||
|
||||
|
||||
class TestEvalAccuracyLarge(CustomTestCase):
|
||||
|
||||
@@ -19,8 +19,8 @@ from sglang.test.test_utils import (
|
||||
write_github_step_summary,
|
||||
)
|
||||
|
||||
register_cuda_ci(est_time=500, suite="stage-b-test-large-2-gpu-accuracy")
|
||||
register_amd_ci(est_time=500, suite="stage-b-test-large-2-gpu-accuracy-amd")
|
||||
register_cuda_ci(est_time=500, suite="stage-b-test-large-2-gpu")
|
||||
register_amd_ci(est_time=500, suite="stage-b-test-large-2-gpu-amd")
|
||||
|
||||
|
||||
class TestMoEEvalAccuracyLarge(CustomTestCase):
|
||||
|
||||
@@ -11,8 +11,8 @@ from sglang.test.test_utils import (
|
||||
write_github_step_summary,
|
||||
)
|
||||
|
||||
register_cuda_ci(est_time=120, suite="stage-b-test-large-1-gpu-performance")
|
||||
register_amd_ci(est_time=120, suite="stage-b-test-large-1-gpu-performance-amd")
|
||||
register_cuda_ci(est_time=120, suite="stage-b-test-large-1-gpu")
|
||||
register_amd_ci(est_time=120, suite="stage-b-test-large-1-gpu-amd")
|
||||
|
||||
|
||||
class TestBenchOneBatch1GPU(CustomTestCase):
|
||||
|
||||
@@ -11,8 +11,8 @@ from sglang.test.test_utils import (
|
||||
write_github_step_summary,
|
||||
)
|
||||
|
||||
register_cuda_ci(est_time=180, suite="stage-b-test-large-2-gpu-performance")
|
||||
register_amd_ci(est_time=630, suite="stage-b-test-large-2-gpu-performance-amd")
|
||||
register_cuda_ci(est_time=180, suite="stage-b-test-large-2-gpu")
|
||||
register_amd_ci(est_time=630, suite="stage-b-test-large-2-gpu-amd")
|
||||
|
||||
|
||||
class TestBenchOneBatch2GPU(CustomTestCase):
|
||||
|
||||
@@ -17,8 +17,8 @@ from sglang.test.test_utils import (
|
||||
write_github_step_summary,
|
||||
)
|
||||
|
||||
register_cuda_ci(est_time=300, suite="stage-b-test-large-1-gpu-performance")
|
||||
register_amd_ci(est_time=300, suite="stage-b-test-large-1-gpu-performance-amd")
|
||||
register_cuda_ci(est_time=300, suite="stage-b-test-large-1-gpu")
|
||||
register_amd_ci(est_time=300, suite="stage-b-test-large-1-gpu-amd")
|
||||
|
||||
|
||||
class TestBenchServing1GPULarge(CustomTestCase):
|
||||
|
||||
@@ -19,8 +19,8 @@ from sglang.test.test_utils import (
|
||||
write_github_step_summary,
|
||||
)
|
||||
|
||||
register_cuda_ci(est_time=1000, suite="stage-b-test-large-1-gpu-performance")
|
||||
register_amd_ci(est_time=1100, suite="stage-b-test-large-1-gpu-performance-amd")
|
||||
register_cuda_ci(est_time=1000, suite="stage-b-test-large-1-gpu")
|
||||
register_amd_ci(est_time=1100, suite="stage-b-test-large-1-gpu-amd")
|
||||
|
||||
|
||||
class TestBenchServing1GPUPart1(CustomTestCase):
|
||||
|
||||
@@ -19,8 +19,8 @@ from sglang.test.test_utils import (
|
||||
write_github_step_summary,
|
||||
)
|
||||
|
||||
register_cuda_ci(est_time=900, suite="stage-b-test-large-1-gpu-performance")
|
||||
register_amd_ci(est_time=900, suite="stage-b-test-large-1-gpu-performance-amd")
|
||||
register_cuda_ci(est_time=900, suite="stage-b-test-large-1-gpu")
|
||||
register_amd_ci(est_time=900, suite="stage-b-test-large-1-gpu-amd")
|
||||
|
||||
|
||||
class TestBenchServing1GPUPart2(CustomTestCase):
|
||||
|
||||
@@ -14,8 +14,8 @@ from sglang.test.test_utils import (
|
||||
write_github_step_summary,
|
||||
)
|
||||
|
||||
register_cuda_ci(est_time=600, suite="stage-b-test-large-2-gpu-performance")
|
||||
register_amd_ci(est_time=600, suite="stage-b-test-large-2-gpu-performance-amd")
|
||||
register_cuda_ci(est_time=600, suite="stage-b-test-large-2-gpu")
|
||||
register_amd_ci(est_time=600, suite="stage-b-test-large-2-gpu-amd")
|
||||
|
||||
|
||||
class TestBenchServing2GPU(CustomTestCase):
|
||||
|
||||
@@ -13,8 +13,8 @@ from sglang.test.test_utils import (
|
||||
write_github_step_summary,
|
||||
)
|
||||
|
||||
register_cuda_ci(est_time=600, suite="stage-b-test-small-1-gpu-performance")
|
||||
register_amd_ci(est_time=500, suite="stage-b-test-small-1-gpu-performance-amd")
|
||||
register_cuda_ci(est_time=600, suite="stage-b-test-small-1-gpu")
|
||||
register_amd_ci(est_time=500, suite="stage-b-test-small-1-gpu-amd")
|
||||
|
||||
|
||||
class TestVLMPerf5090(CustomTestCase):
|
||||
|
||||
Reference in New Issue
Block a user