提交 e936f8c1 编写于 作者: J juncaipeng 提交者: Yan Chunwei

modify benchmark.sh test=develop (#1871)

上级 b35e89d6
#!/bin/bash #!/bin/bash
set -e set -e
if [ $# -lt 2 ]; if [ $# -lt 3 ];
then then
echo "Input error" echo "Input error"
echo "USAGE:" echo "USAGE:"
echo " sh benchmark.sh benchmark_bin_path benchmark_models_path" echo " sh benchmark.sh benchmark_bin_path benchmark_models_path result_filename"
echo " sh benchmark.sh benchmark_bin_path benchmark_models_path is_run_model_optimize" echo " sh benchmark.sh benchmark_bin_path benchmark_models_path result_filename is_run_model_optimize"
exit exit
fi fi
ANDROID_DIR=/data/local/tmp ANDROID_DIR=/data/local/tmp
RESULT_FILENAME="result.txt"
WARMUP=10 WARMUP=10
REPEATS=30 REPEATS=30
BENCHMARK_BIN=$1 BENCHMARK_BIN=$1
MODELS_DIR=$2 MODELS_DIR=$2
RESULT_FILENAME=$3
IS_RUN_MODEL_OPTIMIZE=false IS_RUN_MODEL_OPTIMIZE=false
if [ $# -gt 2 ]; if [ $# -gt 3 ];
then then
IS_RUN_MODEL_OPTIMIZE=$3 IS_RUN_MODEL_OPTIMIZE=$4
fi fi
adb push $BENCHMARK_BIN $ANDROID_DIR/benchmark_bin adb push $BENCHMARK_BIN $ANDROID_DIR/benchmark_bin
...@@ -32,9 +32,12 @@ do ...@@ -32,9 +32,12 @@ do
adb shell "echo Threads=$threads Warmup=$WARMUP Repeats=$REPEATS >> $ANDROID_DIR/$RESULT_FILENAME" adb shell "echo Threads=$threads Warmup=$WARMUP Repeats=$REPEATS >> $ANDROID_DIR/$RESULT_FILENAME"
for model_name in `ls $MODELS_DIR` for model_name in `ls $MODELS_DIR`
do do
echo $model_name echo "Model=$model_name Threads=$threads"
adb shell "$ANDROID_DIR/benchmark_bin --model_dir=$ANDROID_DIR/${MODELS_DIR##*/}/$model_name --warmup=$WARMUP --repeats=$REPEATS --threads=$threads --result_filename=$ANDROID_DIR/$RESULT_FILENAME --run_model_optimize=$IS_RUN_MODEL_OPTIMIZE" adb shell "$ANDROID_DIR/benchmark_bin --model_dir=$ANDROID_DIR/${MODELS_DIR##*/}/$model_name --warmup=$WARMUP --repeats=$REPEATS --threads=$threads --result_filename=$ANDROID_DIR/$RESULT_FILENAME --run_model_optimize=$IS_RUN_MODEL_OPTIMIZE"
done done
adb shell "echo >> $ANDROID_DIR/$RESULT_FILENAME" adb shell "echo >> $ANDROID_DIR/$RESULT_FILENAME"
done done
adb pull $ANDROID_DIR/$RESULT_FILENAME . adb pull $ANDROID_DIR/$RESULT_FILENAME .
echo "\n--------------------------------------"
cat $RESULT_FILENAME
echo "--------------------------------------"
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册