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

[Misc] Make EP kernels install script support uv (#25785)

Browse Source 
Signed-off-by: Lucas Wilkinson <lwilkins@redhat.com>
This commit is contained in:
Lucas Wilkinson 2025-09-30 19:38:34 -04:00 committed by GitHub
parent 5db1870bb9
commit 96ebcaa3ad
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tools/ep_kernels/install_python_libraries.sh
View File

@ -10,8 +10,12 @@ if [ ! -d "$WORKSPACE" ]; then
mkdir -p $WORKSPACE mkdir -p $WORKSPACE
fi fi
# configurable pip command (default: pip3)
PIP_CMD=${PIP_CMD:-pip3}
CUDA_HOME=${CUDA_HOME:-/usr/local/cuda}
# install dependencies if not installed # install dependencies if not installed
pip3 install cmake torch ninja $PIP_CMD install cmake torch ninja
# build nvshmem # build nvshmem
pushd $WORKSPACE pushd $WORKSPACE
@ -110,9 +114,7 @@ clone_repo() {
pushd $WORKSPACE pushd $WORKSPACE
clone_repo "https://github.com/ppl-ai/pplx-kernels" "pplx-kernels" "setup.py" "c336faf" clone_repo "https://github.com/ppl-ai/pplx-kernels" "pplx-kernels" "setup.py" "c336faf"
cd pplx-kernels cd pplx-kernels
# see https://github.com/pypa/pip/issues/9955#issuecomment-838065925 $PIP_CMD install --no-build-isolation -vvv -e .
# PIP_NO_BUILD_ISOLATION=0 disables build isolation
PIP_NO_BUILD_ISOLATION=0 pip install -vvv -e .
popd popd
# build and install deepep, require pytorch installed # build and install deepep, require pytorch installed
@ -120,5 +122,5 @@ pushd $WORKSPACE
clone_repo "https://github.com/deepseek-ai/DeepEP" "DeepEP" "setup.py" "e3908bf" clone_repo "https://github.com/deepseek-ai/DeepEP" "DeepEP" "setup.py" "e3908bf"
cd DeepEP cd DeepEP
export NVSHMEM_DIR=$WORKSPACE/nvshmem_install export NVSHMEM_DIR=$WORKSPACE/nvshmem_install
PIP_NO_BUILD_ISOLATION=0 pip install -vvv -e . $PIP_CMD install --no-build-isolation -vvv -e .
popd popd