From 167285408be0b181a6d9d1b1fd0b6890b12f5d60 Mon Sep 17 00:00:00 2001
From: liuqi <liuqi10@xiaomi.com>
Date: Fri, 16 Mar 2018 15:39:47 +0800
Subject: [PATCH] Fix benchmark gpu bug.

---
 mace/benchmark/BUILD |  3 ++-
 tools/benchmark.sh   | 18 ++++++++++++------
 2 files changed, 14 insertions(+), 7 deletions(-)

diff --git a/mace/benchmark/BUILD b/mace/benchmark/BUILD
index 2cdc00e1..9334e747 100644
--- a/mace/benchmark/BUILD
+++ b/mace/benchmark/BUILD
@@ -16,7 +16,8 @@ cc_library(
     hdrs = ["stat_summarizer.h"],
     linkstatic = 1,
     deps = [
-        "//mace/core",
+        "//mace/public",
+        "//mace/utils",
     ],
 )
 
diff --git a/tools/benchmark.sh b/tools/benchmark.sh
index a37627c4..d081c7e4 100644
--- a/tools/benchmark.sh
+++ b/tools/benchmark.sh
@@ -26,7 +26,8 @@ if [ "$EMBED_MODEL_DATA" = 0 ]; then
 fi
 
 if [ x"$TARGET_ABI" == x"host" ]; then
-  bazel build --verbose_failures -c opt --strip always //mace/benchmark:benchmark_model \
+  bazel build --verbose_failures -c opt --strip always \
+    //mace/benchmark:benchmark_model \
     --copt="-std=c++11" \
     --copt="-D_GLIBCXX_USE_C99_MATH_TR1" \
     --copt="-Werror=return-type" \
@@ -49,16 +50,18 @@ if [ x"$TARGET_ABI" == x"host" ]; then
       $OPTION_ARGS || exit 1
 
 else
-  bazel build --verbose_failures -c opt --strip always //mace/benchmark:benchmark_model \
+  bazel build --verbose_failures -c opt --strip always \
+    //mace/benchmark:benchmark_model \
     --crosstool_top=//external:android/crosstool \
     --host_crosstool_top=@bazel_tools//tools/cpp:toolchain \
     --cpu=${TARGET_ABI} \
     --copt="-std=c++11" \
     --copt="-D_GLIBCXX_USE_C99_MATH_TR1" \
     --copt="-Werror=return-type" \
+    --copt="-DMACE_OBFUSCATE_LITERALS" \
     --copt="-DMACE_MODEL_TAG=${MODEL_TAG}" \
-    --copt="-O3" \
     --define openmp=true \
+    --copt="-O3" \
     --define production=true || exit 1
 
   cp bazel-bin/mace/benchmark/benchmark_model $MODEL_OUTPUT_DIR
@@ -67,11 +70,14 @@ else
   IFS=',' read -r -a INPUT_NAMES <<< "${INPUT_NODES}"
   for NAME in "${INPUT_NAMES[@]}";do
     FORMATTED_NAME=$(sed s/[^[:alnum:]]/_/g <<< ${NAME})
-    adb push ${MODEL_OUTPUT_DIR}/${INPUT_FILE_NAME}_${FORMATTED_NAME} ${PHONE_DATA_DIR} || exit 1
+    adb push ${MODEL_OUTPUT_DIR}/${INPUT_FILE_NAME}_${FORMATTED_NAME} \
+        ${PHONE_DATA_DIR} > /dev/null || exit 1
   done
-  adb push ${MODEL_OUTPUT_DIR}/benchmark_model ${PHONE_DATA_DIR} || exit 1
+  adb push ${MODEL_OUTPUT_DIR}/benchmark_model \
+      ${PHONE_DATA_DIR} > /dev/null || exit 1
   if [ "$EMBED_MODEL_DATA" = 0 ]; then
-    adb push ${MODEL_OUTPUT_DIR}/${MODEL_TAG}.data ${PHONE_DATA_DIR} || exit 1
+    adb push ${MODEL_OUTPUT_DIR}/${MODEL_TAG}.data
+        ${PHONE_DATA_DIR} > /dev/null || exit 1
   fi
 
   adb </dev/null shell \
-- 
GitLab