[XPU CI] Re-enable stage B with docker-pull flow and split tests (#27526)
Co-authored-by: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
This commit is contained in:
co-authored by
Claude Opus 4.7
parent
d7c8b9ab9f
commit
009a0ceefa
@@ -10,7 +10,7 @@ from sglang.srt.utils import get_device
|
||||
from sglang.test.ci.ci_register import register_cuda_ci, register_xpu_ci
|
||||
|
||||
register_cuda_ci(est_time=11, stage="base-b", runner_config="1-gpu-large")
|
||||
register_xpu_ci(est_time=900, suite="stage-a-test-1-gpu-xpu")
|
||||
register_xpu_ci(est_time=900, suite="stage-b-test-1-gpu-xpu")
|
||||
|
||||
|
||||
@unittest.skipIf(
|
||||
|
||||
@@ -19,7 +19,7 @@ from sglang.test.test_utils import (
|
||||
popen_launch_server,
|
||||
)
|
||||
|
||||
register_xpu_ci(est_time=360, suite="stage-a-test-1-gpu-xpu")
|
||||
register_xpu_ci(est_time=360, suite="stage-b-test-1-gpu-xpu")
|
||||
|
||||
|
||||
class TestDeepSeekOCR(CustomTestCase):
|
||||
|
||||
@@ -18,7 +18,7 @@ from sglang.test.test_utils import (
|
||||
|
||||
register_xpu_ci(
|
||||
est_time=360,
|
||||
suite="stage-a-test-1-gpu-xpu",
|
||||
suite="stage-b-test-1-gpu-xpu",
|
||||
disabled="Temporarily disabled until Triton-XPU upgrade",
|
||||
)
|
||||
|
||||
|
||||
@@ -130,7 +130,11 @@ class TestGemma4E2BXPU(CustomTestCase):
|
||||
from sglang.test.ci.ci_register import register_xpu_ci
|
||||
|
||||
# Single e2e test: boot + a short Q&A.
|
||||
register_xpu_ci(est_time=240, suite="stage-b-test-1-gpu-xpu")
|
||||
register_xpu_ci(
|
||||
est_time=240,
|
||||
suite="stage-b-test-1-gpu-xpu",
|
||||
disabled="OOM on stage-b XPU runner (server launch fails with --mem-fraction-static)",
|
||||
)
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
|
||||
@@ -16,7 +16,7 @@ from sglang.test.test_utils import (
|
||||
run_bench_one_batch,
|
||||
)
|
||||
|
||||
register_xpu_ci(est_time=600, suite="stage-a-test-1-gpu-xpu")
|
||||
register_xpu_ci(est_time=600, suite="stage-b-test-1-gpu-xpu")
|
||||
|
||||
|
||||
def intel_xpu_benchmark(
|
||||
|
||||
@@ -11,7 +11,7 @@ from sglang.srt.layers.moe.topk import (
|
||||
from sglang.test.ci.ci_register import register_xpu_ci
|
||||
from sglang.test.test_utils import CustomTestCase
|
||||
|
||||
register_xpu_ci(est_time=5, suite="stage-a-test-1-gpu-xpu")
|
||||
register_xpu_ci(est_time=5, suite="stage-b-test-1-gpu-xpu")
|
||||
|
||||
|
||||
# Nemotron-3 uses biased_grouped_topk
|
||||
|
||||
Reference in New Issue
Block a user