[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:
ashwini rathi
2026-06-09 09:26:11 +08:00
committed by GitHub
co-authored by Claude Opus 4.7
parent d7c8b9ab9f
commit 009a0ceefa
7 changed files with 63 additions and 63 deletions
@@ -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(
+1 -1
View File
@@ -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",
)
+5 -1
View File
@@ -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(
+1 -1
View File
@@ -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