Skip to content

Commit d7e8e8c

Browse files
Small changes to support releases (vllm-project#37)
These are some changes we've been adding onto the last couple releases. Would be nice to have on main for the next round, and hopefully this is all moot once we hop over to the ODH fork anyway --------- Signed-off-by: Travis Johnson <[email protected]> Signed-off-by: Joe Runde <[email protected]> Co-authored-by: Travis Johnson <[email protected]>
1 parent 24f4ff0 commit d7e8e8c

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

Dockerfile.ubi

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -193,7 +193,7 @@ ENV TORCH_CUDA_ARCH_LIST=$TORCH_CUDA_ARCH_LIST
193193
ARG max_jobs=2
194194
ENV MAX_JOBS=${max_jobs}
195195
# number of threads used by nvcc
196-
ARG nvcc_threads=8
196+
ARG nvcc_threads=2
197197
ENV NVCC_THREADS=$nvcc_threads
198198
# make sure punica kernels are built (for LoRA)
199199
ENV VLLM_INSTALL_PUNICA_KERNELS=1
@@ -208,7 +208,12 @@ COPY --link vllm vllm
208208
# Comment if building *.so files from scratch
209209
##################################################
210210
# Copy the prebuilt *.so files
211-
COPY --from=prebuilt-wheel --link /workspace/vllm/*.so /workspace/vllm/
211+
# COPY --from=prebuilt-wheel --link /workspace/vllm/*.so /workspace/vllm/
212+
# ENV VLLM_USE_PRECOMPILED=1
213+
##################################################
214+
# Comment if not building .so files from scratch
215+
RUN microdnf install -y git \
216+
&& microdnf clean all
212217
##################################################
213218

214219
# Copy over the generated *.pb2 files
@@ -217,7 +222,7 @@ COPY --from=gen-protos --link /workspace/vllm/entrypoints/grpc/pb vllm/entrypoin
217222
ENV CCACHE_DIR=/root/.cache/ccache
218223
RUN --mount=type=cache,target=/root/.cache/ccache \
219224
--mount=type=cache,target=/root/.cache/pip \
220-
VLLM_USE_PRECOMPILED=1 python3 setup.py bdist_wheel --dist-dir=dist
225+
CMAKE_BUILD_TYPE=Release python3 setup.py bdist_wheel --dist-dir=dist
221226

222227
## Release #####################################################################
223228
# Note from the non-UBI Dockerfile:

0 commit comments

Comments
 (0)