Browse Source

!8327 [MS][LITE][CPU SSE] add x86 new sse package and modify run_benchmark_nets.sh

From: @lzkcode
Reviewed-by: @xsmq
Signed-off-by:
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
507381b8c7
2 changed files with 10 additions and 4 deletions
  1. +5
    -1
      mindspore/lite/CMakeLists.txt
  2. +5
    -3
      mindspore/lite/test/run_benchmark_nets.sh

+ 5
- 1
mindspore/lite/CMakeLists.txt View File

@@ -69,7 +69,11 @@ else ()
else ()
set(COMPONENT_NAME converter-ubuntu)
endif()
set(RUN_X86_COMPONENT_NAME runtime-x86-${PROCESS_UNIT})
if ("${X86_64_SIMD}" STREQUAL "sse")
set(RUN_X86_COMPONENT_NAME runtime-x86-${X86_64_SIMD}-${PROCESS_UNIT})
else ()
set(RUN_X86_COMPONENT_NAME runtime-x86-${PROCESS_UNIT})
endif ()
endif()

string(REPLACE "/mindspore/lite" "" TOP_DIR ${CMAKE_CURRENT_SOURCE_DIR})


+ 5
- 3
mindspore/lite/test/run_benchmark_nets.sh View File

@@ -974,7 +974,7 @@ process_unit_arm32=${suffix[0]}

x86_path=${release_path}/ubuntu_x86
mv ${x86_path}/*runtime-*train* ./train
file_name=$(ls ${x86_path}/*runtime-x86*.tar.gz)
file_name=$(ls ${x86_path}/*runtime-x86-cpu*.tar.gz)
IFS="-" read -r -a file_name_array <<< "$file_name"
IFS="." read -r -a suffix <<< "${file_name_array[-1]}"
process_unit_x86=${suffix[0]}
@@ -1036,7 +1036,6 @@ else
exit 1
fi


# Write benchmark result to temp file
run_benchmark_result_file=${basepath}/run_benchmark_result.txt
echo ' ' > ${run_benchmark_result_file}
@@ -1044,6 +1043,9 @@ echo ' ' > ${run_benchmark_result_file}
run_x86_log_file=${basepath}/run_x86_log.txt
echo 'run x86 logs: ' > ${run_x86_log_file}

run_x86_sse_log_file=${basepath}/run_x86_sse_log.txt
echo 'run x86 sse logs: ' > ${run_x86_sse_log_file}

run_arm64_log_file=${basepath}/run_arm64_log.txt
echo 'run arm64 logs: ' > ${run_arm64_log_file}

@@ -1111,6 +1113,6 @@ if [[ ${Run_arm32_status} != 0 ]];then
exit 1
fi

echo "Run_x86 and Run_arm64 and Run_arm32 is ended"
echo "Run_x86 and Run_x86_sse and Run_arm64 and Run_arm32 is ended"
Print_Benchmark_Result
exit 0

Loading…
Cancel
Save