File tree Expand file tree Collapse file tree 1 file changed +3
-0
lines changed Expand file tree Collapse file tree 1 file changed +3
-0
lines changed Original file line number Diff line number Diff line change @@ -149,13 +149,16 @@ if [ $((${MODE} & 0x02)) -ne 0 ]; then
149
149
sed -i " s/VER_IPEX=.*/VER_IPEX=/" compile_bundle.sh
150
150
bash compile_bundle.sh ${DPCPP_ROOT} ${ONEMKL_ROOT} ${ONECCL_ROOT} ${MPI_ROOT} ${PTI_ROOT} ${AOT} 9
151
151
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 \;
152
154
rm -rf compile_bundle.sh llvm-project llvm-release torch-ccl patchelf
153
155
if [ -d pytorch ]; then
154
156
rm -rf pytorch
155
157
else
156
158
echo " python -m pip install torch==${VER_TORCH} --index-url https://download.pytorch.org/whl/xpu" >> ${AUX_INSTALL_SCRIPT}
157
159
fi
158
160
echo " python -m pip install ./wheels/*.whl" >> ${AUX_INSTALL_SCRIPT}
161
+ echo " python -m pip install ./wheels/ipex_wheel/*.whl" >> ${AUX_INSTALL_SCRIPT}
159
162
echo " rm -rf wheels" >> ${AUX_INSTALL_SCRIPT}
160
163
fi
161
164
echo " python -m pip install -r ./requirements.txt" >> ${AUX_INSTALL_SCRIPT}
You can’t perform that action at this time.
0 commit comments