Skip to content

Commit 59c30ba

Browse files
committed
build(vllm-tensorizer): Specify nvcc wrapper via CMAKE_ARGS env var
1 parent e3b4475 commit 59c30ba

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

vllm-tensorizer/Dockerfile

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -130,6 +130,7 @@ RUN --mount=type=bind,from=vllm-downloader,source=/git/vllm,target=/workspace,rw
130130
--mount=type=tmpfs,target=/tmp \
131131
. /opt/sccache-start.sh && \
132132
. /opt/arch_flags.sh && \
133+
export CMAKE_ARGS='-DCMAKE_CUDA_COMPILER=/opt/nvcc-wrapper.py' && \
133134
if [ -z "$MAX_JOBS" ]; then unset MAX_JOBS; fi && \
134135
python3 -m pip install --no-cache-dir py-cpuinfo 'cmake>=3.26.1,<4' grpcio-tools && \
135136
if [ -f 'use_existing_torch.py' ]; then \

0 commit comments

Comments
 (0)