Browse Source

!6938 【MSLite】Specify train converter

Merge pull request !6938 from lz/master
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
d15febdb91
2 changed files with 4 additions and 3 deletions
  1. +3
    -3
      mindspore/lite/CMakeLists.txt
  2. +1
    -0
      mindspore/lite/test/run_benchmark_nets.sh

+ 3
- 3
mindspore/lite/CMakeLists.txt View File

@@ -28,10 +28,10 @@ if (SUPPORT_TRAIN)
elseif (PLATFORM_ARM32)
set(COMPONENT_NAME runtime-arm32-${PROCESS_UNIT}-train)
elseif (WIN32)
set(PARSER_NAME libconverter-parser-win-${PROCESS_UNIT})
set(COMPONENT_NAME converter-win-${PROCESS_UNIT})
set(PARSER_NAME libconverter-parser-win-${PROCESS_UNIT}-train)
set(COMPONENT_NAME converter-win-${PROCESS_UNIT}-train)
else ()
set(COMPONENT_NAME converter-ubuntu)
set(COMPONENT_NAME converter-ubuntu-train)
endif()
set(RUN_X86_COMPONENT_NAME runtime-x86-${PROCESS_UNIT}-train)
else ()


+ 1
- 0
mindspore/lite/test/run_benchmark_nets.sh View File

@@ -655,6 +655,7 @@ IFS="." read -r -a suffix <<< "${file_name_array[-1]}"
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)
IFS="-" read -r -a file_name_array <<< "$file_name"
IFS="." read -r -a suffix <<< "${file_name_array[-1]}"


Loading…
Cancel
Save