Install DeepEP from release wheels (#33932)
This commit is contained in:
@@ -15,7 +15,7 @@ from sglang.test.test_utils import (
|
||||
popen_launch_server,
|
||||
)
|
||||
|
||||
register_cuda_ci(est_time=528, stage="extra-b", runner_config="deepep-8-gpu-h200")
|
||||
register_cuda_ci(est_time=528, stage="extra-b", runner_config="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, stage="base-c", runner_config="deepep-4-gpu-h100")
|
||||
register_cuda_ci(est_time=478, stage="base-c", runner_config="4-gpu-h100")
|
||||
|
||||
|
||||
class TestPureDP(CustomTestCase):
|
||||
|
||||
@@ -21,7 +21,7 @@ from sglang.test.test_utils import (
|
||||
register_cuda_ci(
|
||||
est_time=189,
|
||||
stage="base-c",
|
||||
runner_config="deepep-4-gpu-h100",
|
||||
runner_config="4-gpu-h100",
|
||||
disabled="Temporarily disabled until the next Mooncake release includes the PyTorch 2.13 collective forwarding fix.",
|
||||
)
|
||||
|
||||
|
||||
@@ -12,7 +12,7 @@ from sglang.test.test_utils import (
|
||||
popen_launch_server,
|
||||
)
|
||||
|
||||
register_cuda_ci(est_time=900, stage="extra-b", runner_config="deepep-8-gpu-h200")
|
||||
register_cuda_ci(est_time=900, stage="extra-b", runner_config="8-gpu-h200")
|
||||
|
||||
DEEPSEEK_V3_MODEL_PATH = "deepseek-ai/DeepSeek-V3-0324"
|
||||
|
||||
|
||||
Reference in New Issue
Block a user