From 1bf2d55342467b9ebb774f93c956e611d635d16a Mon Sep 17 00:00:00 2001
From: wuchenghui <wuchenghui@xiaomi.com>
Date: Tue, 27 Feb 2018 19:18:36 +0800
Subject: [PATCH] add EMBED_MODE_DATA option

---
 build_mace_run.sh      | 3 +++
 config                 | 1 +
 generate_model_code.sh | 1 +
 tuning_run.sh          | 5 +++++
 4 files changed, 10 insertions(+)

diff --git a/build_mace_run.sh b/build_mace_run.sh
index 0d9408a2..ec5438f9 100644
--- a/build_mace_run.sh
+++ b/build_mace_run.sh
@@ -65,3 +65,6 @@ if [ -f "$MODEL_OUTPUT_DIR/mace_run" ]; then
   rm -rf $MODEL_OUTPUT_DIR/mace_run
 fi
 cp bazel-bin/examples/mace_run $MODEL_OUTPUT_DIR
+if [ "$EMBED_MODEL_DATA" = 0 ]; then
+  cp codegen/models/${MODEL_TAG}/${MODEL_TAG}.data $MODEL_OUTPUT_DIR
+fi
\ No newline at end of file
diff --git a/config b/config
index 9f76f97a..f47bb827 100644
--- a/config
+++ b/config
@@ -14,3 +14,4 @@ TARGET_ABI=armeabi-v7a
 LIMIT_OPENCL_KERNEL_TIME=0
 DSP_MODE=0
 BENCHMARK_FLAG=0
+EMBED_MODEL_DATA=1
diff --git a/generate_model_code.sh b/generate_model_code.sh
index 2427696b..8184b871 100644
--- a/generate_model_code.sh
+++ b/generate_model_code.sh
@@ -26,4 +26,5 @@ bazel-bin/lib/python/tools/tf_converter --input=${TF_MODEL_FILE_PATH} \
                                         --model_tag=${MODEL_TAG} \
                                         --input_shape=${INPUT_SHAPE} \
                                         ${DSP_MODE_FLAG} \
+                                        --embed_model_data=${EMBED_MODEL_DATA} \
                                         --obfuscate=${OBFUSCATE} || exit 1
diff --git a/tuning_run.sh b/tuning_run.sh
index 222390d3..d7bfa4a9 100644
--- a/tuning_run.sh
+++ b/tuning_run.sh
@@ -24,6 +24,7 @@ if [ x"$TARGET_ABI" = x"host" ]; then
       --output_shape="${OUTPUT_SHAPE}"\
       --input_file=${MODEL_OUTPUT_DIR}/${INPUT_FILE_NAME} \
       --output_file=${MODEL_OUTPUT_DIR}/${OUTPUT_FILE_NAME} \
+      --model_data_file=${MODEL_OUTPUT_DIR}/${MODEL_TAG}.data \
       --device=${DEVICE_TYPE}   \
       --round=1 || exit 1
 else
@@ -39,6 +40,9 @@ else
   fi
   adb push ${MODEL_OUTPUT_DIR}/${INPUT_FILE_NAME} ${PHONE_DATA_DIR} || exit 1
   adb push ${MODEL_OUTPUT_DIR}/mace_run ${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 push lib/hexagon/libhexagon_controller.so ${PHONE_DATA_DIR} || exit 1
   
   adb </dev/null shell \
@@ -53,6 +57,7 @@ else
     --output_shape="${OUTPUT_SHAPE}"\
     --input_file=${PHONE_DATA_DIR}/${INPUT_FILE_NAME} \
     --output_file=${PHONE_DATA_DIR}/${OUTPUT_FILE_NAME} \
+    --model_data_file=${PHONE_DATA_DIR}/${MODEL_TAG}.data \
     --device=${DEVICE_TYPE}   \
     --round=$ROUND || exit 1
 fi
-- 
GitLab