mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 13:44:15 +08:00
201ba13911
Summary: Updated file to correct shell code to test whether $ANDROID_HOME env variable is empty or not. Pull Request resolved: https://github.com/pytorch/pytorch/pull/37064 Differential Revision: D21181787 Pulled By: IvanKobzarev fbshipit-source-id: 40c1d79d0fb730c7f68aa7472ce9b2398e91f2a2
106 lines
2.8 KiB
Bash
Executable File
106 lines
2.8 KiB
Bash
Executable File
#!/bin/bash
|
|
set -eux
|
|
|
|
# Script's workflow:
|
|
# 1. Builds libtorch for android for specified android abisi (by default for all 4).
|
|
# Custom list of android abis can be specified as a bash argument as comma separated list.
|
|
# For example just for testing on android x86 emulator we need only x86 build.
|
|
# ./scipts/build_pytorch_androis.sh x86
|
|
# 2. Creates symbolic links to android/pytorch_android/src/main/jniLibs/${abi} for libtorch build output,
|
|
# android/pytorch_android/src/main/cpp/libtorch_include/${abi} for headers.
|
|
# 3. Runs pyotrch_android gradle build:
|
|
# gradle assembleRelease
|
|
|
|
PYTORCH_DIR="$(cd $(dirname $0)/..; pwd -P)"
|
|
|
|
PYTORCH_ANDROID_DIR=$PYTORCH_DIR/android
|
|
WORK_DIR=$PYTORCH_DIR
|
|
|
|
echo "PYTORCH_DIR:$PYTORCH_DIR"
|
|
echo "WORK_DIR:$WORK_DIR"
|
|
|
|
echo "ANDROID_HOME:$ANDROID_HOME"
|
|
if [ -z "$ANDROID_HOME" ]; then
|
|
echo "ANDROID_HOME not set; please set it to Android sdk directory"
|
|
fi
|
|
|
|
if [ ! -d $ANDROID_HOME ]; then
|
|
echo "ANDROID_HOME not a directory; did you install it under $ANDROID_HOME?"
|
|
exit 1
|
|
fi
|
|
|
|
GRADLE_PATH=gradle
|
|
GRADLE_NOT_FOUND_MSG="Unable to find gradle, please add it to PATH or set GRADLE_HOME"
|
|
|
|
if [ ! -x "$(command -v gradle)" ]; then
|
|
if [ -z "$GRADLE_HOME" ]; then
|
|
echo GRADLE_NOT_FOUND_MSG
|
|
exit 1
|
|
fi
|
|
GRADLE_PATH=$GRADLE_HOME/bin/gradle
|
|
if [ ! -f "$GRADLE_PATH" ]; then
|
|
echo GRADLE_NOT_FOUND_MSG
|
|
exit 1
|
|
fi
|
|
fi
|
|
echo "GRADLE_PATH:$GRADLE_PATH"
|
|
|
|
ABIS_LIST="armeabi-v7a,arm64-v8a,x86,x86_64"
|
|
CUSTOM_ABIS_LIST=false
|
|
if [ $# -gt 0 ]; then
|
|
ABIS_LIST=$1
|
|
CUSTOM_ABIS_LIST=true
|
|
fi
|
|
|
|
echo "ABIS_LIST:$ABIS_LIST"
|
|
|
|
LIB_DIR=$PYTORCH_ANDROID_DIR/pytorch_android/src/main/jniLibs
|
|
INCLUDE_DIR=$PYTORCH_ANDROID_DIR/pytorch_android/src/main/cpp/libtorch_include
|
|
mkdir -p $LIB_DIR
|
|
mkdir -p $INCLUDE_DIR
|
|
|
|
for abi in $(echo $ABIS_LIST | tr ',' '\n')
|
|
do
|
|
echo "abi:$abi"
|
|
|
|
OUT_DIR=$WORK_DIR/build_android_$abi
|
|
|
|
rm -rf $OUT_DIR
|
|
mkdir -p $OUT_DIR
|
|
|
|
pushd $PYTORCH_DIR
|
|
python $PYTORCH_DIR/setup.py clean
|
|
|
|
ANDROID_ABI=$abi $PYTORCH_DIR/scripts/build_android.sh -DANDROID_CCACHE=$(which ccache)
|
|
|
|
cp -R $PYTORCH_DIR/build_android/install/lib $OUT_DIR/
|
|
cp -R $PYTORCH_DIR/build_android/install/include $OUT_DIR/
|
|
|
|
echo "$abi build output lib,include copied to $OUT_DIR"
|
|
|
|
LIB_LINK_PATH=$LIB_DIR/$abi
|
|
INCLUDE_LINK_PATH=$INCLUDE_DIR/$abi
|
|
|
|
rm -f $LIB_LINK_PATH
|
|
rm -f $INCLUDE_LINK_PATH
|
|
|
|
ln -s $OUT_DIR/lib $LIB_LINK_PATH
|
|
ln -s $OUT_DIR/include $INCLUDE_LINK_PATH
|
|
|
|
done
|
|
|
|
# To set proxy for gradle add following lines to ./gradle/gradle.properties:
|
|
# systemProp.http.proxyHost=...
|
|
# systemProp.http.proxyPort=8080
|
|
# systemProp.https.proxyHost=...
|
|
# systemProp.https.proxyPort=8080
|
|
|
|
if [ "$CUSTOM_ABIS_LIST" = true ]; then
|
|
$GRADLE_PATH -PABI_FILTERS=$ABIS_LIST -p $PYTORCH_ANDROID_DIR clean assembleRelease
|
|
else
|
|
$GRADLE_PATH -p $PYTORCH_ANDROID_DIR clean assembleRelease
|
|
fi
|
|
|
|
find $PYTORCH_ANDROID_DIR -type f -name *aar | xargs ls -lah
|
|
popd
|