We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e950b15 commit 258d2d3Copy full SHA for 258d2d3
docker/Dockerfile.rocm
@@ -1,6 +1,5 @@
1
# default base image
2
ARG REMOTE_VLLM="0"
3
-ARG USE_CYTHON="0"
4
ARG BUILD_RPD="1"
5
ARG COMMON_WORKDIR=/app
6
ARG BASE_IMAGE=rocm/vllm-dev:base
@@ -36,12 +35,10 @@ FROM fetch_vllm_${REMOTE_VLLM} AS fetch_vllm
36
35
# -----------------------
37
# vLLM build stages
38
FROM fetch_vllm AS build_vllm
39
-ARG USE_CYTHON
40
# Build vLLM
41
RUN cd vllm \
42
&& python3 -m pip install -r requirements/rocm.txt \
43
&& python3 setup.py clean --all \
44
- && if [ ${USE_CYTHON} -eq "1" ]; then python3 tests/build_cython.py build_ext --inplace; fi \
45
&& python3 setup.py bdist_wheel --dist-dir=dist
46
FROM scratch AS export_vllm
47
ARG COMMON_WORKDIR
0 commit comments