Skip to content

Commit

Permalink
Dockerfile.ubi: use COPY . . to make repo available when building wheel
Browse files Browse the repository at this point in the history
  • Loading branch information
dtrifiro committed Sep 27, 2024
1 parent d151278 commit 78a09a7
Showing 1 changed file with 1 addition and 12 deletions.
13 changes: 1 addition & 12 deletions Dockerfile.ubi
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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 \
Expand Down

0 comments on commit 78a09a7

Please sign in to comment.