mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
cca34be584
Updating XNNPACK Version to 4ea82e595b36106653175dcb04b2aa532660d0d8 submodule update Pull Request resolved: https://github.com/pytorch/pytorch/pull/139913 Approved by: https://github.com/digantdesai, https://github.com/huydhn
76 lines
3.0 KiB
Bash
76 lines
3.0 KiB
Bash
#!/bin/bash
|
|
set -ex -o pipefail
|
|
|
|
echo ""
|
|
echo "DIR: $(pwd)"
|
|
WORKSPACE=/Users/distiller/workspace
|
|
PROJ_ROOT=/Users/distiller/project
|
|
ARTIFACTS_DIR=${WORKSPACE}/ios
|
|
ls ${ARTIFACTS_DIR}
|
|
ZIP_DIR=${WORKSPACE}/zip
|
|
mkdir -p ${ZIP_DIR}/install/lib
|
|
mkdir -p ${ZIP_DIR}/src
|
|
# copy header files
|
|
cp -R ${ARTIFACTS_DIR}/arm64/include ${ZIP_DIR}/install/
|
|
# build a FAT bianry
|
|
cd ${ZIP_DIR}/install/lib
|
|
target_libs=(libc10.a libclog.a libcpuinfo.a libeigen_blas.a libpthreadpool.a libpytorch_qnnpack.a libtorch_cpu.a libtorch.a libXNNPACK.a libmicrokernels-prod.a)
|
|
for lib in ${target_libs[*]}
|
|
do
|
|
if [ -f "${ARTIFACTS_DIR}/x86_64/lib/${lib}" ] && [ -f "${ARTIFACTS_DIR}/arm64/lib/${lib}" ]; then
|
|
libs=("${ARTIFACTS_DIR}/x86_64/lib/${lib}" "${ARTIFACTS_DIR}/arm64/lib/${lib}")
|
|
lipo -create "${libs[@]}" -o ${ZIP_DIR}/install/lib/${lib}
|
|
fi
|
|
done
|
|
lipo -i ${ZIP_DIR}/install/lib/*.a
|
|
echo "BUILD_LITE_INTERPRETER: ${BUILD_LITE_INTERPRETER}"
|
|
# copy the umbrella header and license
|
|
if [ "${BUILD_LITE_INTERPRETER}" == "1" ]; then
|
|
cp ${PROJ_ROOT}/ios/LibTorch-Lite.h ${ZIP_DIR}/src/
|
|
else
|
|
cp ${PROJ_ROOT}/ios/LibTorch.h ${ZIP_DIR}/src/
|
|
fi
|
|
cp ${PROJ_ROOT}/LICENSE ${ZIP_DIR}/
|
|
# zip the library
|
|
export DATE="$(date -u +%Y%m%d)"
|
|
export IOS_NIGHTLY_BUILD_VERSION="2.2.0.${DATE}"
|
|
if [ "${BUILD_LITE_INTERPRETER}" == "1" ]; then
|
|
# libtorch_lite_ios_nightly_1.11.0.20210810.zip
|
|
ZIPFILE="libtorch_lite_ios_nightly_${IOS_NIGHTLY_BUILD_VERSION}.zip"
|
|
else
|
|
ZIPFILE="libtorch_ios_nightly_build.zip"
|
|
fi
|
|
cd ${ZIP_DIR}
|
|
#for testing
|
|
touch version.txt
|
|
echo "${IOS_NIGHTLY_BUILD_VERSION}" > version.txt
|
|
zip -r ${ZIPFILE} install src version.txt LICENSE
|
|
# upload to aws
|
|
# Install conda then 'conda install' awscli
|
|
curl --retry 3 -o ~/conda.sh https://repo.anaconda.com/miniconda/Miniconda3-py39_4.12.0-MacOSX-x86_64.sh
|
|
chmod +x ~/conda.sh
|
|
/bin/bash ~/conda.sh -b -p ~/anaconda
|
|
export PATH="~/anaconda/bin:${PATH}"
|
|
source ~/anaconda/bin/activate
|
|
conda install -c conda-forge awscli --yes
|
|
set +x
|
|
export AWS_ACCESS_KEY_ID=${AWS_S3_ACCESS_KEY_FOR_PYTORCH_BINARY_UPLOAD}
|
|
export AWS_SECRET_ACCESS_KEY=${AWS_S3_ACCESS_SECRET_FOR_PYTORCH_BINARY_UPLOAD}
|
|
set +x
|
|
# echo "AWS KEY: ${AWS_ACCESS_KEY_ID}"
|
|
# echo "AWS SECRET: ${AWS_SECRET_ACCESS_KEY}"
|
|
aws s3 cp ${ZIPFILE} s3://ossci-ios-build/ --acl public-read
|
|
|
|
if [ "${BUILD_LITE_INTERPRETER}" == "1" ]; then
|
|
# create a new LibTorch-Lite-Nightly.podspec from the template
|
|
echo "cp ${PROJ_ROOT}/ios/LibTorch-Lite-Nightly.podspec.template ${PROJ_ROOT}/ios/LibTorch-Lite-Nightly.podspec"
|
|
cp ${PROJ_ROOT}/ios/LibTorch-Lite-Nightly.podspec.template ${PROJ_ROOT}/ios/LibTorch-Lite-Nightly.podspec
|
|
|
|
# update pod version
|
|
sed -i '' -e "s/IOS_NIGHTLY_BUILD_VERSION/${IOS_NIGHTLY_BUILD_VERSION}/g" ${PROJ_ROOT}/ios/LibTorch-Lite-Nightly.podspec
|
|
cat ${PROJ_ROOT}/ios/LibTorch-Lite-Nightly.podspec
|
|
|
|
# push the new LibTorch-Lite-Nightly.podspec to CocoaPods
|
|
pod trunk push --verbose --allow-warnings --use-libraries --skip-import-validation ${PROJ_ROOT}/ios/LibTorch-Lite-Nightly.podspec
|
|
fi
|