xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-04-09 07:17:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

Patch DeepEP when building docker image with CUDA 13 (#29154)

Browse Source 
Signed-off-by: Qidong Su <soodoshll@gmail.com>
This commit is contained in:
Qidong Su 2025-11-22 18:25:13 -05:00 committed by GitHub
parent 472fdee974
commit 4587063267
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tools/ep_kernels/install_python_libraries.sh
View File

@ -96,6 +96,13 @@ clone_repo() {
fi
}
deepep_cuda13_patch() {
cuda_version_major=$(${CUDA_HOME}/bin/nvcc --version | egrep -o "release [0-9]+" | cut -d ' ' -f 2)
if [ ${cuda_version_major} -ge 13 ]; then
sed -i "s|f'{nvshmem_dir}/include']|f'{nvshmem_dir}/include', '${CUDA_HOME}/include/cccl']|" "setup.py"
fi
}
do_build() {
local repo=$1
local name=$2
@ -107,6 +114,10 @@ do_build() {
clone_repo "$repo" "$name" "$key" "$commit"
cd "$name"
if [ "$name" == "DeepEP" ]; then
deepep_cuda13_patch
fi
if [ "$MODE" = "install" ]; then
echo "Installing $name into environment"
eval "$extra_env" uv pip install --no-build-isolation -vvv .