Skip to content

Commit 98e4afb

Browse files
committed
s390x: Fix PyArrow build and add CPU test script for Buildkite CI
Signed-off-by: Nishan Acharya <[email protected]>
1 parent 95d63f3 commit 98e4afb

File tree

2 files changed

+16
-2
lines changed

2 files changed

+16
-2
lines changed

Diff for: .buildkite/run-cpu-test-s390x.sh

+13
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
#!/bin/bash
2+
3+
# This script build the CPU docker image and run the offline inference inside the container.
4+
# It serves a sanity check for compilation and basic model usage.
5+
set -ex
6+
7+
# Setup cleanup
8+
remove_docker_container() { docker rm -f cpu-test || true; docker system prune -f; }
9+
trap remove_docker_container EXIT
10+
remove_docker_container
11+
12+
# Try building the docker image
13+
docker build -t cpu-test -f docker/Dockerfile.s390x .

Diff for: docker/Dockerfile.s390x

+3-2
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@ RUN --mount=type=cache,target=/root/.cache/uv \
5858
cd ../../python && \
5959
export PYARROW_PARALLEL=4 && \
6060
export ARROW_BUILD_TYPE=release && \
61-
uv pip install -r requirements/build.txt && \
61+
uv pip install -r requirements-build.txt && \
6262
python setup.py build_ext --build-type=$ARROW_BUILD_TYPE --bundle-arrow-cpp bdist_wheel
6363

6464
FROM python-install AS numa-build
@@ -149,4 +149,5 @@ USER 2000
149149
WORKDIR /home/vllm
150150

151151
# Set the default entrypoint
152-
ENTRYPOINT ["python", "-m", "vllm.entrypoints.openai.api_server"]
152+
ENTRYPOINT ["python", "-m", "vllm.entrypoints.openai.api_server"]
153+

0 commit comments

Comments
 (0)