From 2d33d36bb1c8becdc3ca39202aa55dc369521a29 Mon Sep 17 00:00:00 2001
From: wang_shaocong <wangshaocong1@huawei.com>
Date: Sat, 19 Dec 2020 14:08:14 +0800
Subject: [PATCH] fix bug of test file of ci.

---
 mindspore/lite/test/run_benchmark_nets.sh | 39 ++++++++++++-----------
 1 file changed, 21 insertions(+), 18 deletions(-)

diff --git a/mindspore/lite/test/run_benchmark_nets.sh b/mindspore/lite/test/run_benchmark_nets.sh
index 955c2fa509..7d3cfe01e3 100644
--- a/mindspore/lite/test/run_benchmark_nets.sh
+++ b/mindspore/lite/test/run_benchmark_nets.sh
@@ -570,16 +570,17 @@ function Run_x86() {
         input_num=${input_params%%;*}
         input_files=''
         output_file=''
-        if [[ -z "$input_files" || $input_files == 1 ]] && [ -e ${ms_models_path}/${model_name}'.ms.bin' ]; then
-          input_files=$model_name'.ms.bin'
-        elif [[ ! -z "$input_files" && $input_files -gt 1 ]]; then
+        data_path="/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/"
+        if [[ -z "$input_num" || $input_num == 1 ]] && [ -e ${data_path}'input/'${model_name}'.ms.bin' ]; then
+          input_files=${data_path}'input/'$model_name'.ms.bin'
+        elif [[ ! -z "$input_num" && $input_num -gt 1 ]]; then
           for i in $(seq 1 $input_num)
           do
-            input_files=$input_files$model_name'.ms.bin_'$i','
+            input_files=$input_files${data_path}'input/'$model_name'.ms.bin_'$i','
           done
         fi
-        if [ -e ${ms_models_path}/${model_name}'.ms.out' ]; then
-          output_file=${ms_models_path}/${model_name}'.ms.out'
+        if [ -e ${data_path}'output/'${model_name}'.ms.out' ]; then
+          output_file=${data_path}'output/'${model_name}'.ms.out'
         fi
         if [[ ${model_name##*.} == "caffemodel" ]]; then
           model_name=${model_name%.*}
@@ -835,16 +836,17 @@ function Run_x86_sse() {
         input_num=${input_params%%;*}
         input_files=''
         output_file=''
-        if [[ -z "$input_files" || $input_files == 1 ]] && [ -e ${ms_models_path}/${model_name}'.ms.bin' ]; then
-          input_files=$model_name'.ms.bin'
-        elif [[ ! -z "$input_files" && $input_files -gt 1 ]]; then
+        data_path="/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/"
+        if [[ -z "$input_num" || $input_num == 1 ]] && [ -e ${data_path}'input/'${model_name}'.ms.bin' ]; then
+          input_files=${data_path}'input/'$model_name'.ms.bin'
+        elif [[ ! -z "$input_num" && $input_num -gt 1 ]]; then
           for i in $(seq 1 $input_num)
           do
-            input_files=$input_files$model_name'.ms.bin_'$i','
+            input_files=$input_files${data_path}'input/'$model_name'.ms.bin_'$i','
           done
         fi
-        if [ -e ${ms_models_path}/${model_name}'.ms.out' ]; then
-          output_file=${ms_models_path}/${model_name}'.ms.out'
+        if [ -e ${data_path}'output/'${model_name}'.ms.out' ]; then
+          output_file=${data_path}'output/'${model_name}'.ms.out'
         fi
         if [[ ${model_name##*.} == "caffemodel" ]]; then
           model_name=${model_name%.*}
@@ -1323,16 +1325,17 @@ function Run_arm64() {
         input_num=${input_params%%;*}
         input_files=''
         output_file=''
-        if [[ -z "$input_files" || $input_files == 1 ]] && [ -e ${ms_models_path}/${model_name}'.ms.bin' ]; then
-          input_files=$model_name'.ms.bin'
-        elif [[ ! -z "$input_files" && $input_files -gt 1 ]]; then
+        data_path="/data/local/tmp/input_output/"
+        if [[ -z "$input_num" || $input_num == 1 ]] && [ -e ${data_path}'input/'${model_name}'.ms.bin' ]; then
+          input_files=${data_path}'input/'$model_name'.ms.bin'
+        elif [[ ! -z "$input_num" && $input_num -gt 1 ]]; then
           for i in $(seq 1 $input_num)
           do
-            input_files=$input_files$model_name'.ms.bin_'$i','
+            input_files=$input_files${data_path}'input/'$model_name'.ms.bin_'$i','
           done
         fi
-        if [ -e ${ms_models_path}/${model_name}'.ms.out' ]; then
-          output_file=${ms_models_path}/${model_name}'.ms.out'
+        if [ -e ${data_path}'output/'${model_name}'.ms.out' ]; then
+          output_file=${data_path}'output/'${model_name}'.ms.out'
         fi
         if [[ ${model_name##*.} == "caffemodel" ]]; then
           model_name=${model_name%.*}