| @@ -50,10 +50,10 @@ usage() | |||||
| echo " -Q Enable dump memory, default off" | echo " -Q Enable dump memory, default off" | ||||
| echo " -D Enable dumping of function graph ir, default on" | echo " -D Enable dumping of function graph ir, default on" | ||||
| echo " -z Compile dataset & mindrecord, default on" | echo " -z Compile dataset & mindrecord, default on" | ||||
| echo " -n Compile minddata with mindspore-lite, available: off, lite, full, default is lite" | |||||
| echo " -n Compile minddata with mindspore lite, available: off, lite, full, default is lite" | |||||
| echo " -M Enable MPI and NCCL for GPU training, gpu default on" | echo " -M Enable MPI and NCCL for GPU training, gpu default on" | ||||
| echo " -V Specify the minimum required cuda version, default CUDA 10.1" | echo " -V Specify the minimum required cuda version, default CUDA 10.1" | ||||
| echo " -I Compile mindspore-lite" | |||||
| echo " -I Enable compiling mindspore lite for arm64, arm32 or x86_64, default disable mindspore lite compiling" | |||||
| echo " -K Compile with AKG, default on" | echo " -K Compile with AKG, default on" | ||||
| echo " -s Enable serving module, default off" | echo " -s Enable serving module, default off" | ||||
| echo " -w Enable acl module, default off" | echo " -w Enable acl module, default off" | ||||
| @@ -29,7 +29,7 @@ elseif (WIN32) | |||||
| set(PARSER_NAME libconverter-parser-win-${PROCESS_UNIT}) | set(PARSER_NAME libconverter-parser-win-${PROCESS_UNIT}) | ||||
| set(COMPONENT_NAME converter-win-${PROCESS_UNIT}) | set(COMPONENT_NAME converter-win-${PROCESS_UNIT}) | ||||
| else () | else () | ||||
| set(COMPONENT_NAME convert-ubuntu) | |||||
| set(COMPONENT_NAME converter-ubuntu) | |||||
| endif() | endif() | ||||
| set(RUN_X86_COMPONENT_NAME runtime-x86-${PROCESS_UNIT}) | set(RUN_X86_COMPONENT_NAME runtime-x86-${PROCESS_UNIT}) | ||||
| @@ -374,13 +374,13 @@ tar -zxf mindspore-lite-${version}-runtime-arm64-${process_unit_arm}.tar.gz || e | |||||
| cd ${convertor_path} || exit 1 | cd ${convertor_path} || exit 1 | ||||
| tar -zxf mindspore-lite-${version}-runtime-x86-${process_unit_x86}.tar.gz || exit 1 | tar -zxf mindspore-lite-${version}-runtime-x86-${process_unit_x86}.tar.gz || exit 1 | ||||
| tar -zxf mindspore-lite-${version}-convert-ubuntu.tar.gz || exit 1 | |||||
| cd ${convertor_path}/mindspore-lite-${version}-convert-ubuntu || exit 1 | |||||
| tar -zxf mindspore-lite-${version}-converter-ubuntu.tar.gz || exit 1 | |||||
| cd ${convertor_path}/mindspore-lite-${version}-converter-ubuntu || exit 1 | |||||
| cp converter/converter_lite ./ || exit 1 | cp converter/converter_lite ./ || exit 1 | ||||
| export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:./lib/:./third_party/protobuf/lib | export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:./lib/:./third_party/protobuf/lib | ||||
| # Convert the models | # Convert the models | ||||
| cd ${convertor_path}/mindspore-lite-${version}-convert-ubuntu || exit 1 | |||||
| cd ${convertor_path}/mindspore-lite-${version}-converter-ubuntu || exit 1 | |||||
| # Write resulte to temp file | # Write resulte to temp file | ||||
| run_benchmark_result_file=${basepath}/run_benchmark_result.txt | run_benchmark_result_file=${basepath}/run_benchmark_result.txt | ||||