From 46745e2a211e95610dd4159818f508dc4a91727c Mon Sep 17 00:00:00 2001 From: yefeng Date: Thu, 25 Mar 2021 09:31:33 +0800 Subject: [PATCH] fix_fp16_door --- mindspore/lite/test/models_tf_fp16.cfg | 2 +- mindspore/lite/test/run_benchmark_nets.sh | 6 ++++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/mindspore/lite/test/models_tf_fp16.cfg b/mindspore/lite/test/models_tf_fp16.cfg index ec302d8f74..605b4b2971 100644 --- a/mindspore/lite/test/models_tf_fp16.cfg +++ b/mindspore/lite/test/models_tf_fp16.cfg @@ -1 +1 @@ -decoder_step_201217_modified.pb 5 \ No newline at end of file +decoder_step_201217_modified.pb 5 diff --git a/mindspore/lite/test/run_benchmark_nets.sh b/mindspore/lite/test/run_benchmark_nets.sh index b62eeffac0..7c899a56fb 100755 --- a/mindspore/lite/test/run_benchmark_nets.sh +++ b/mindspore/lite/test/run_benchmark_nets.sh @@ -220,6 +220,7 @@ function Run_Converter() { continue fi model_name=`echo ${fp16_line_info}|awk -F ' ' '{print $1}'` + input_num=`echo ${fp16_line_info}|awk -F ' ' '{print $2}'` echo 'cp '${ms_models_path}'/'${model_name}'.ms' ${ms_models_path}'/'${model_name}'.fp16.ms' cp ${ms_models_path}/${model_name}.ms ${ms_models_path}/${model_name}.fp16.ms if [ $? = 0 ]; then @@ -1866,8 +1867,9 @@ function Run_arm64_fp16() { done echo ${model_name} >> "${run_arm64_fp16_log_file}" echo 'cd /data/local/tmp/benchmark_test' > adb_run_cmd.txt - echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/data/local/tmp/benchmark_test;./benchmark --inputShapes='${input_shapes}' --modelFile='${model_name}'.ms --inDataFile='${input_files}' --benchmarkDataFile=/data/local/tmp/input_output/output/'${model_name}'.ms.out --enableFp16=true' >> "${run_arm64_fp16_log_file}" - echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/data/local/tmp/benchmark_test;./benchmark --inputShapes='${input_shapes}' --modelFile='${model_name}'.ms --inDataFile='${input_files}' --benchmarkDataFile=/data/local/tmp/input_output/output/'${model_name}'.ms.out --enableFp16=true' >> adb_run_cmd.txt + echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/data/local/tmp/benchmark_test;./benchmark --inputShapes='${input_shapes}' --modelFile='${model_name}'.fp16.ms --inDataFile='${input_files}' --benchmarkDataFile=/data/local/tmp/input_output/output/'${model_name}'.ms.out --enableFp16=true' >> adb_run_cmd.txt + echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/data/local/tmp/benchmark_test;./benchmark --inputShapes='${input_shapes}' --modelFile='${model_name}'.fp16.ms --inDataFile='${input_files}' --benchmarkDataFile=/data/local/tmp/input_output/output/'${model_name}'.ms.out --enableFp16=true' >> adb_run_cmd.txt + cat adb_run_cmd.txt >> "${run_arm64_fp16_log_file}" adb -s ${device_id} shell < adb_run_cmd.txt >> "${run_arm64_fp16_log_file}" if [ $? = 0 ]; then run_result='arm64_fp16: '${model_name}' pass'; echo ${run_result} >> ${run_benchmark_result_file}