Skip to content

Commit 13e4ee1

Browse files
[XPU][UT] increase intel xpu CI test scope (#21492)
Signed-off-by: Ma, Liangliang <[email protected]>
1 parent 772ce5a commit 13e4ee1

File tree

3 files changed

+13
-3
lines changed

3 files changed

+13
-3
lines changed

.buildkite/scripts/hardware_ci/run-xpu-test.sh

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,4 +31,13 @@ docker run \
3131
VLLM_USE_V1=1 python3 examples/offline_inference/basic/generate.py --model facebook/opt-125m --block-size 64 --enforce-eager -tp 2 --distributed-executor-backend mp
3232
cd tests
3333
pytest -v -s v1/core
34+
pytest -v -s v1/engine
35+
pytest -v -s v1/sample --ignore=v1/sample/test_logprobs.py --ignore=v1/sample/test_logprobs_e2e.py
36+
pytest -v -s v1/worker --ignore=v1/worker/test_gpu_model_runner.py
37+
pytest -v -s v1/structured_output
38+
pytest -v -s v1/spec_decode --ignore=v1/spec_decode/test_max_len.py --ignore=v1/spec_decode/test_eagle.py
39+
pytest -v -s v1/kv_connector/unit --ignore=v1/kv_connector/unit/test_multi_connector.py --ignore=v1/kv_connector/unit/test_nixl_connector.py
40+
pytest -v -s v1/test_serial_utils.py
41+
pytest -v -s v1/test_utils.py
42+
pytest -v -s v1/test_metrics_reader.py
3443
'

docker/Dockerfile.xpu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ FROM vllm-base AS vllm-openai
4747

4848
# install additional dependencies for openai api server
4949
RUN --mount=type=cache,target=/root/.cache/pip \
50-
pip install accelerate hf_transfer pytest modelscope
50+
pip install accelerate hf_transfer pytest pytest_asyncio lm_eval[api] modelscope
5151

5252
ENV VLLM_USAGE_SOURCE production-docker-image \
5353
TRITON_XPU_PROFILE 1

tests/entrypoints/openai/correctness/test_lmeval.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -69,8 +69,9 @@ def run_test(more_args):
6969

7070

7171
@pytest.mark.skipif(not current_platform.is_cuda()
72-
and not current_platform.is_tpu(),
73-
reason="V1 currently only supported on CUDA and TPU")
72+
and not current_platform.is_tpu()
73+
and not current_platform.is_xpu(),
74+
reason="V1 currently only supported on CUDA, XPU and TPU")
7475
def test_lm_eval_accuracy_v1_engine(monkeypatch: pytest.MonkeyPatch):
7576
"""Run with the V1 Engine."""
7677

0 commit comments

Comments
 (0)