From 91b17029971478e68319105343979f981e261809 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Daniele=20Trifir=C3=B2?= Date: Fri, 27 Sep 2024 11:46:54 +0200 Subject: [PATCH] Dockerfile.ubi: use COPY . . to make repo available when building wheel --- Dockerfile.ubi | 13 +------------ 1 file changed, 1 insertion(+), 12 deletions(-) diff --git a/Dockerfile.ubi b/Dockerfile.ubi index 679b9a9737449..15f01ec860862 100644 --- a/Dockerfile.ubi +++ b/Dockerfile.ubi @@ -93,16 +93,8 @@ RUN --mount=type=cache,target=/root/.cache/pip \ # install compiler cache to speed up compilation leveraging local or remote caching # git is required for the cutlass kernels RUN rpm -ivh https://dl.fedoraproject.org/pub/epel/epel-release-latest-9.noarch.rpm && rpm -ql epel-release && microdnf install -y git ccache && microdnf clean all -# install build dependencies -# copy input files -COPY csrc csrc -COPY setup.py setup.py -COPY cmake cmake -COPY CMakeLists.txt CMakeLists.txt -COPY requirements-common.txt requirements-common.txt -COPY requirements-cuda.txt requirements-cuda.txt -COPY pyproject.toml pyproject.toml +COPY . . ARG TORCH_CUDA_ARCH_LIST ENV TORCH_CUDA_ARCH_LIST=$TORCH_CUDA_ARCH_LIST @@ -121,9 +113,6 @@ ENV VLLM_INSTALL_PUNICA_KERNELS=1 # Make sure the cuda environment is in the PATH ENV PATH=/usr/local/cuda/bin:$PATH -# Copy the entire directory before building wheel -COPY vllm vllm - ENV CCACHE_DIR=/root/.cache/ccache RUN --mount=type=cache,target=/root/.cache/ccache \ --mount=type=cache,target=/root/.cache/pip \