Skip to content

Commit b95ea42

Browse files
authored
fix intel compiler dependency issue in env_setup (#5407)
1 parent 2cec7af commit b95ea42

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

examples/gpu/llm/tools/env_setup.sh

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -149,13 +149,16 @@ if [ $((${MODE} & 0x02)) -ne 0 ]; then
149149
sed -i "s/VER_IPEX=.*/VER_IPEX=/" compile_bundle.sh
150150
bash compile_bundle.sh ${DPCPP_ROOT} ${ONEMKL_ROOT} ${ONECCL_ROOT} ${MPI_ROOT} ${PTI_ROOT} ${AOT} 9
151151
find . -name "dist" -exec bash -c "cp {}/*.whl ${WHEELFOLDER}" \;
152+
mkdir -p ${WHEELFOLDER}/ipex_wheel
153+
find ${WHEELFOLDER} -name "intel_extension_for_pytorch*.whl" -exec mv {} ${WHEELFOLDER}/ipex_wheel \;
152154
rm -rf compile_bundle.sh llvm-project llvm-release torch-ccl patchelf
153155
if [ -d pytorch ]; then
154156
rm -rf pytorch
155157
else
156158
echo "python -m pip install torch==${VER_TORCH} --index-url https://download.pytorch.org/whl/xpu" >> ${AUX_INSTALL_SCRIPT}
157159
fi
158160
echo "python -m pip install ./wheels/*.whl" >> ${AUX_INSTALL_SCRIPT}
161+
echo "python -m pip install ./wheels/ipex_wheel/*.whl" >> ${AUX_INSTALL_SCRIPT}
159162
echo "rm -rf wheels" >> ${AUX_INSTALL_SCRIPT}
160163
fi
161164
echo "python -m pip install -r ./requirements.txt" >> ${AUX_INSTALL_SCRIPT}

0 commit comments

Comments
 (0)