Skip to content

Commit 96ebcaa

Browse files
[Misc] Make EP kernels install script support uv (#25785)
Signed-off-by: Lucas Wilkinson <[email protected]>
1 parent 5db1870 commit 96ebcaa

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

tools/ep_kernels/install_python_libraries.sh

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -10,8 +10,12 @@ if [ ! -d "$WORKSPACE" ]; then
1010
mkdir -p $WORKSPACE
1111
fi
1212

13+
# configurable pip command (default: pip3)
14+
PIP_CMD=${PIP_CMD:-pip3}
15+
CUDA_HOME=${CUDA_HOME:-/usr/local/cuda}
16+
1317
# install dependencies if not installed
14-
pip3 install cmake torch ninja
18+
$PIP_CMD install cmake torch ninja
1519

1620
# build nvshmem
1721
pushd $WORKSPACE
@@ -110,15 +114,13 @@ clone_repo() {
110114
pushd $WORKSPACE
111115
clone_repo "https://github.com/ppl-ai/pplx-kernels" "pplx-kernels" "setup.py" "c336faf"
112116
cd pplx-kernels
113-
# see https://github.com/pypa/pip/issues/9955#issuecomment-838065925
114-
# PIP_NO_BUILD_ISOLATION=0 disables build isolation
115-
PIP_NO_BUILD_ISOLATION=0 pip install -vvv -e .
117+
$PIP_CMD install --no-build-isolation -vvv -e .
116118
popd
117119

118120
# build and install deepep, require pytorch installed
119121
pushd $WORKSPACE
120122
clone_repo "https://github.com/deepseek-ai/DeepEP" "DeepEP" "setup.py" "e3908bf"
121123
cd DeepEP
122124
export NVSHMEM_DIR=$WORKSPACE/nvshmem_install
123-
PIP_NO_BUILD_ISOLATION=0 pip install -vvv -e .
125+
$PIP_CMD install --no-build-isolation -vvv -e .
124126
popd

0 commit comments

Comments
 (0)