Skip to content

Commit 084659e

Browse files
authored
Update script to build and upload MiniBench artifacts
Build this app during CI. We can start trying it with ``` adb shell am start -n org.pytorch.minibench/org.pytorch.minibench.BenchmarkActivity --es model_path /data/local/tmp/model.pte adb shell run-as org.pytorch.minibench cat files/benchmark_results.txt ``` Pull Request resolved: #5017
1 parent e793795 commit 084659e

File tree

1 file changed

+12
-2
lines changed

1 file changed

+12
-2
lines changed

build/build_android_llm_demo.sh

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -91,12 +91,18 @@ build_aar() {
9191
popd
9292
}
9393

94-
build_android_llm_demo_app() {
94+
build_android_demo_apps() {
9595
mkdir -p examples/demo-apps/android/LlamaDemo/app/libs
9696
cp ${BUILD_AAR_DIR}/executorch-llama.aar examples/demo-apps/android/LlamaDemo/app/libs
9797
pushd examples/demo-apps/android/LlamaDemo
9898
ANDROID_HOME="${ANDROID_SDK:-/opt/android/sdk}" ./gradlew build assembleAndroidTest
9999
popd
100+
101+
mkdir -p extension/android/benchmark/app/libs
102+
cp ${BUILD_AAR_DIR}/executorch.aar extension/android/benchmark/app/libs
103+
pushd extension/android/benchmark
104+
ANDROID_HOME="${ANDROID_SDK:-/opt/android/sdk}" ./gradlew build
105+
popd
100106
}
101107

102108
collect_artifacts_to_be_uploaded() {
@@ -116,6 +122,10 @@ collect_artifacts_to_be_uploaded() {
116122
# Collect JAR and AAR
117123
cp extension/android/build/libs/executorch.jar "${DEMO_APP_DIR}"
118124
find "${BUILD_AAR_DIR}/" -name 'executorch*.aar' -exec cp {} "${DEMO_APP_DIR}" \;
125+
# Collect MiniBench APK
126+
MINIBENCH_APP_DIR="${ARTIFACTS_DIR_NAME}/minibench"
127+
mkdir -p "${MINIBENCH_APP_DIR}"
128+
cp extension/android/benchmark/app/build/outputs/apk/debug/*.apk "${MINIBENCH_APP_DIR}"
119129
}
120130

121131
BUILD_AAR_DIR="$(mktemp -d)"
@@ -130,5 +140,5 @@ for ANDROID_ABI in "${ANDROID_ABIS[@]}"; do
130140
build_android_native_library ${ANDROID_ABI}
131141
done
132142
build_aar
133-
build_android_llm_demo_app
143+
build_android_demo_apps
134144
collect_artifacts_to_be_uploaded ${ARTIFACTS_DIR_NAME}

0 commit comments

Comments
 (0)