ci: decouple stage and runner for cuda registry (#25197)
This commit is contained in:
@@ -16,7 +16,7 @@ from sglang.test.test_utils import (
|
||||
popen_launch_server,
|
||||
)
|
||||
|
||||
register_cuda_ci(est_time=528, suite="stage-c-test-deepep-8-gpu-h200")
|
||||
register_cuda_ci(est_time=528, stage="stage-c", runner_config="deepep-8-gpu-h200")
|
||||
|
||||
DEEPSEEK_V32_MODEL_PATH = "deepseek-ai/DeepSeek-V3.2"
|
||||
|
||||
|
||||
@@ -16,7 +16,7 @@ from sglang.test.test_utils import (
|
||||
popen_launch_server,
|
||||
)
|
||||
|
||||
register_cuda_ci(est_time=478, suite="stage-c-test-deepep-4-gpu-h100")
|
||||
register_cuda_ci(est_time=478, stage="stage-c", runner_config="deepep-4-gpu-h100")
|
||||
|
||||
|
||||
class TestPureDP(CustomTestCase):
|
||||
|
||||
@@ -15,7 +15,7 @@ from sglang.test.test_utils import (
|
||||
popen_launch_server,
|
||||
)
|
||||
|
||||
register_cuda_ci(est_time=82, suite="stage-c-test-deepep-4-gpu-h100")
|
||||
register_cuda_ci(est_time=82, stage="stage-c", runner_config="deepep-4-gpu-h100")
|
||||
|
||||
ib_devices = get_rdma_devices_args()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user