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
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}
You can’t perform that action at this time.
0 commit comments