From 20fc17ff38f78eb97c35455ac2be30f9ac9b02ef Mon Sep 17 00:00:00 2001
From: wuchenghui <wuchenghui@xiaomi.com>
Date: Mon, 5 Mar 2018 17:18:37 +0800
Subject: [PATCH] fix benchmark & support global_avg_pool convert

---
 benchmark.sh | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/benchmark.sh b/benchmark.sh
index 5f5d80e9..7154b55a 100644
--- a/benchmark.sh
+++ b/benchmark.sh
@@ -18,6 +18,10 @@ if [ -f "$MODEL_OUTPUT_DIR/benchmark_model" ]; then
   rm -rf $MODEL_OUTPUT_DIR/benchmark_model
 fi
 
+if [ "$EMBED_MODEL_DATA" = 0 ]; then
+  cp codegen/models/${MODEL_TAG}/${MODEL_TAG}.data $MODEL_OUTPUT_DIR
+fi
+
 if [ x"$RUNTIME" = x"host" ]; then
   bazel build --verbose_failures -c opt --strip always benchmark:benchmark_model \
     --copt="-std=c++11" \
@@ -26,6 +30,7 @@ if [ x"$RUNTIME" = x"host" ]; then
     --copt="-DMACE_MODEL_TAG=${MODEL_TAG}" \
     --copt="-O3" \
     --define openmp=true \
+    --model_data_file=${MODEL_OUTPUT_DIR}/${MODEL_TAG}.data \
     --define production=true || exit 1
 
   cp bazel-bin/benchmark/benchmark_model $MODEL_OUTPUT_DIR
@@ -55,6 +60,9 @@ else
   adb shell "mkdir -p ${PHONE_DATA_DIR}" || exit 1
   adb push ${MODEL_OUTPUT_DIR}/${INPUT_FILE_NAME} ${PHONE_DATA_DIR} || exit 1
   adb push ${MODEL_OUTPUT_DIR}/benchmark_model ${PHONE_DATA_DIR} || exit 1
+  if [ "$EMBED_MODEL_DATA" = 0 ]; then
+    adb push ${MODEL_OUTPUT_DIR}/${MODEL_TAG}.data ${PHONE_DATA_DIR} || exit 1
+  fi
 
   adb </dev/null shell \
     LD_LIBRARY_PATH=${PHONE_DATA_DIR} \
-- 
GitLab