|
|
@ -570,16 +570,17 @@ function Run_x86() {
|
|
|
|
input_num=${input_params%%;*}
|
|
|
|
input_num=${input_params%%;*}
|
|
|
|
input_files=''
|
|
|
|
input_files=''
|
|
|
|
output_file=''
|
|
|
|
output_file=''
|
|
|
|
if [[ -z "$input_files" || $input_files == 1 ]] && [ -e ${ms_models_path}/${model_name}'.ms.bin' ]; then
|
|
|
|
data_path="/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/"
|
|
|
|
input_files=$model_name'.ms.bin'
|
|
|
|
if [[ -z "$input_num" || $input_num == 1 ]] && [ -e ${data_path}'input/'${model_name}'.ms.bin' ]; then
|
|
|
|
elif [[ ! -z "$input_files" && $input_files -gt 1 ]]; 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)
|
|
|
|
for i in $(seq 1 $input_num)
|
|
|
|
do
|
|
|
|
do
|
|
|
|
input_files=$input_files$model_name'.ms.bin_'$i','
|
|
|
|
input_files=$input_files${data_path}'input/'$model_name'.ms.bin_'$i','
|
|
|
|
done
|
|
|
|
done
|
|
|
|
fi
|
|
|
|
fi
|
|
|
|
if [ -e ${ms_models_path}/${model_name}'.ms.out' ]; then
|
|
|
|
if [ -e ${data_path}'output/'${model_name}'.ms.out' ]; then
|
|
|
|
output_file=${ms_models_path}/${model_name}'.ms.out'
|
|
|
|
output_file=${data_path}'output/'${model_name}'.ms.out'
|
|
|
|
fi
|
|
|
|
fi
|
|
|
|
if [[ ${model_name##*.} == "caffemodel" ]]; then
|
|
|
|
if [[ ${model_name##*.} == "caffemodel" ]]; then
|
|
|
|
model_name=${model_name%.*}
|
|
|
|
model_name=${model_name%.*}
|
|
|
@ -835,16 +836,17 @@ function Run_x86_sse() {
|
|
|
|
input_num=${input_params%%;*}
|
|
|
|
input_num=${input_params%%;*}
|
|
|
|
input_files=''
|
|
|
|
input_files=''
|
|
|
|
output_file=''
|
|
|
|
output_file=''
|
|
|
|
if [[ -z "$input_files" || $input_files == 1 ]] && [ -e ${ms_models_path}/${model_name}'.ms.bin' ]; then
|
|
|
|
data_path="/home/workspace/mindspore_dataset/mslite/models/hiai/input_output/"
|
|
|
|
input_files=$model_name'.ms.bin'
|
|
|
|
if [[ -z "$input_num" || $input_num == 1 ]] && [ -e ${data_path}'input/'${model_name}'.ms.bin' ]; then
|
|
|
|
elif [[ ! -z "$input_files" && $input_files -gt 1 ]]; 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)
|
|
|
|
for i in $(seq 1 $input_num)
|
|
|
|
do
|
|
|
|
do
|
|
|
|
input_files=$input_files$model_name'.ms.bin_'$i','
|
|
|
|
input_files=$input_files${data_path}'input/'$model_name'.ms.bin_'$i','
|
|
|
|
done
|
|
|
|
done
|
|
|
|
fi
|
|
|
|
fi
|
|
|
|
if [ -e ${ms_models_path}/${model_name}'.ms.out' ]; then
|
|
|
|
if [ -e ${data_path}'output/'${model_name}'.ms.out' ]; then
|
|
|
|
output_file=${ms_models_path}/${model_name}'.ms.out'
|
|
|
|
output_file=${data_path}'output/'${model_name}'.ms.out'
|
|
|
|
fi
|
|
|
|
fi
|
|
|
|
if [[ ${model_name##*.} == "caffemodel" ]]; then
|
|
|
|
if [[ ${model_name##*.} == "caffemodel" ]]; then
|
|
|
|
model_name=${model_name%.*}
|
|
|
|
model_name=${model_name%.*}
|
|
|
@ -1323,16 +1325,17 @@ function Run_arm64() {
|
|
|
|
input_num=${input_params%%;*}
|
|
|
|
input_num=${input_params%%;*}
|
|
|
|
input_files=''
|
|
|
|
input_files=''
|
|
|
|
output_file=''
|
|
|
|
output_file=''
|
|
|
|
if [[ -z "$input_files" || $input_files == 1 ]] && [ -e ${ms_models_path}/${model_name}'.ms.bin' ]; then
|
|
|
|
data_path="/data/local/tmp/input_output/"
|
|
|
|
input_files=$model_name'.ms.bin'
|
|
|
|
if [[ -z "$input_num" || $input_num == 1 ]] && [ -e ${data_path}'input/'${model_name}'.ms.bin' ]; then
|
|
|
|
elif [[ ! -z "$input_files" && $input_files -gt 1 ]]; 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)
|
|
|
|
for i in $(seq 1 $input_num)
|
|
|
|
do
|
|
|
|
do
|
|
|
|
input_files=$input_files$model_name'.ms.bin_'$i','
|
|
|
|
input_files=$input_files${data_path}'input/'$model_name'.ms.bin_'$i','
|
|
|
|
done
|
|
|
|
done
|
|
|
|
fi
|
|
|
|
fi
|
|
|
|
if [ -e ${ms_models_path}/${model_name}'.ms.out' ]; then
|
|
|
|
if [ -e ${data_path}'output/'${model_name}'.ms.out' ]; then
|
|
|
|
output_file=${ms_models_path}/${model_name}'.ms.out'
|
|
|
|
output_file=${data_path}'output/'${model_name}'.ms.out'
|
|
|
|
fi
|
|
|
|
fi
|
|
|
|
if [[ ${model_name##*.} == "caffemodel" ]]; then
|
|
|
|
if [[ ${model_name##*.} == "caffemodel" ]]; then
|
|
|
|
model_name=${model_name%.*}
|
|
|
|
model_name=${model_name%.*}
|
|
|
|