diff --git a/mace/benchmark/BUILD b/mace/benchmark/BUILD index 2cdc00e180175be1237545a8b1d4e7cbfb876633..9334e7473acb0e706cdd34840228d16cc38578b3 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 a37627c4f091c76658959a0a793788f2f7db102f..d081c7e40d0e660b2e89fc9ac2fd5fffac022406 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