[Misc] Make EP kernels install script support uv (#25785)
Signed-off-by: Lucas Wilkinson <lwilkins@redhat.com>
This commit is contained in:
@@ -10,8 +10,12 @@ if [ ! -d "$WORKSPACE" ]; then
|
||||
mkdir -p $WORKSPACE
|
||||
fi
|
||||
|
||||
# configurable pip command (default: pip3)
|
||||
PIP_CMD=${PIP_CMD:-pip3}
|
||||
CUDA_HOME=${CUDA_HOME:-/usr/local/cuda}
|
||||
|
||||
# install dependencies if not installed
|
||||
pip3 install cmake torch ninja
|
||||
$PIP_CMD install cmake torch ninja
|
||||
|
||||
# build nvshmem
|
||||
pushd $WORKSPACE
|
||||
@@ -110,9 +114,7 @@ clone_repo() {
|
||||
pushd $WORKSPACE
|
||||
clone_repo "https://github.com/ppl-ai/pplx-kernels" "pplx-kernels" "setup.py" "c336faf"
|
||||
cd pplx-kernels
|
||||
# see https://github.com/pypa/pip/issues/9955#issuecomment-838065925
|
||||
# PIP_NO_BUILD_ISOLATION=0 disables build isolation
|
||||
PIP_NO_BUILD_ISOLATION=0 pip install -vvv -e .
|
||||
$PIP_CMD install --no-build-isolation -vvv -e .
|
||||
popd
|
||||
|
||||
# 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"
|
||||
cd DeepEP
|
||||
export NVSHMEM_DIR=$WORKSPACE/nvshmem_install
|
||||
PIP_NO_BUILD_ISOLATION=0 pip install -vvv -e .
|
||||
$PIP_CMD install --no-build-isolation -vvv -e .
|
||||
popd
|
||||
|
||||
Reference in New Issue
Block a user