"lib/bindings/python/vscode:/vscode.git/clone" did not exist on "b5e762b2b875551bff51a7b09dba47d2d79fc8a8"
Unverified Commit 04e8a9c3 authored by GuanLuo's avatar GuanLuo Committed by GitHub
Browse files

fix: WAR DeepGemm JIT compilation errors (#2937)


Signed-off-by: default avatarGuanLuo <gluo@nvidia.com>
Co-authored-by: default avatarHarrison Saturley-Hall <hsaturleyhal@nvidia.com>
parent 729b5fd5
......@@ -199,7 +199,9 @@ RUN apt-get update && \
ninja-build \
g++ \
# prometheus dependencies
ca-certificates && \
ca-certificates \
# DeepGemm uses 'cuobjdump' which does not come with CUDA image
cuda-command-line-tools-12-8 && \
rm -rf /var/lib/apt/lists/*
# Copy CUDA development tools (nvcc, headers, dependencies, etc.) from base devel image
......@@ -249,6 +251,10 @@ $NIXL_PLUGIN_DIR:\
/usr/local/ucx/lib/ucx:\
$LD_LIBRARY_PATH
# DeepGemm runs nvcc for JIT kernel compilation, however the CUDA include path
# is not properly set for complilation. Set CPATH to help nvcc find the headers.
ENV CPATH=/usr/local/cuda/include:$CPATH
### VIRTUAL ENVIRONMENT SETUP ###
# Copy uv and entire virtual environment from framework container
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment