From 61612befaba81de806263eb2bc85edd0c30857b6 Mon Sep 17 00:00:00 2001 From: Joe Runde Date: Tue, 21 May 2024 10:12:16 -0600 Subject: [PATCH] :bug: fix kernel compilation Signed-off-by: Joe Runde --- Dockerfile.ubi | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/Dockerfile.ubi b/Dockerfile.ubi index d9e6b8a40..915191869 100644 --- a/Dockerfile.ubi +++ b/Dockerfile.ubi @@ -209,6 +209,7 @@ COPY --link vllm vllm ################################################## # Copy the prebuilt *.so files # COPY --from=prebuilt-wheel --link /workspace/vllm/*.so /workspace/vllm/ +# ENV VLLM_USE_PRECOMPILED=1 ################################################## # Copy over the generated *.pb2 files @@ -217,7 +218,7 @@ COPY --from=gen-protos --link /workspace/vllm/entrypoints/grpc/pb vllm/entrypoin ENV CCACHE_DIR=/root/.cache/ccache RUN --mount=type=cache,target=/root/.cache/ccache \ --mount=type=cache,target=/root/.cache/pip \ - VLLM_USE_PRECOMPILED=1 python3 setup.py bdist_wheel --dist-dir=dist + python3 setup.py bdist_wheel --dist-dir=dist ## Release ##################################################################### # Note from the non-UBI Dockerfile: