rename net_train to benchmark_train

pull/10364/head
xutianchun 5 years ago
parent 3fb0bc399a
commit fa0129435f

@ -18,7 +18,7 @@ mindspore_add_pkg(flatbuffers
EXE flatc EXE flatc
URL ${REQ_URL} URL ${REQ_URL}
MD5 ${MD5} MD5 ${MD5}
CMAKE_OPTION -DFLATBUFFERS_BUILD_TESTS=OFF ) CMAKE_OPTION -DFLATBUFFERS_BUILD_TESTS=OFF -DCMAKE_INSTALL_LIBDIR=lib)
include_directories(${flatbuffers_INC}) include_directories(${flatbuffers_INC})
add_library(mindspore::flatbuffers ALIAS flatbuffers::flatbuffers) add_library(mindspore::flatbuffers ALIAS flatbuffers::flatbuffers)

@ -230,7 +230,7 @@ add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/nnacl)
if (ENABLE_TOOLS) if (ENABLE_TOOLS)
add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/tools/benchmark) add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/tools/benchmark)
if (SUPPORT_TRAIN) if (SUPPORT_TRAIN)
add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/tools/net_train) add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/tools/benchmark_train)
endif () endif ()
endif () endif ()
if (NOT WIN32) if (NOT WIN32)

@ -17,6 +17,7 @@
#include "src/net_runner.h" #include "src/net_runner.h"
#include <math.h> #include <math.h>
#include <getopt.h> #include <getopt.h>
#include <cstring>
#include <iostream> #include <iostream>
#include <fstream> #include <fstream>
#include "include/context.h" #include "include/context.h"
@ -104,7 +105,7 @@ std::vector<int> NetRunner::FillInputData(const std::vector<DataLabelTuple> &dat
int label = 0; int label = 0;
char *data = nullptr; char *data = nullptr;
std::tie(data, label) = dataset[idx]; std::tie(data, label) = dataset[idx];
memcpy(input_data + i * data_size_, data, data_size_); std::memcpy(input_data + i * data_size_, data, data_size_);
labels[i * num_of_classes_ + label] = 1.0; // Model expects labels in onehot representation labels[i * num_of_classes_ + label] = 1.0; // Model expects labels in onehot representation
labels_vec.push_back(label); labels_vec.push_back(label);
} }

@ -17,6 +17,10 @@
#ifndef MINDSPORE_LITE_EXAMPLES_TRAIN_LENET_SRC_UTILS_H_ #ifndef MINDSPORE_LITE_EXAMPLES_TRAIN_LENET_SRC_UTILS_H_
#define MINDSPORE_LITE_EXAMPLES_TRAIN_LENET_SRC_UTILS_H_ #define MINDSPORE_LITE_EXAMPLES_TRAIN_LENET_SRC_UTILS_H_
// DEBUG should be defined because the source code use assert to test exception
// but the Code Analysis Tool not allow us to use assert directly
#define DEBUG TRUE
#ifdef DEBUG #ifdef DEBUG
#include <cassert> #include <cassert>
#define MS_ASSERT(f) assert(f) #define MS_ASSERT(f) assert(f)

@ -17,6 +17,7 @@
#include "src/net_runner.h" #include "src/net_runner.h"
#include <math.h> #include <math.h>
#include <getopt.h> #include <getopt.h>
#include <cstring>
#include <iostream> #include <iostream>
#include <fstream> #include <fstream>
#include "include/context.h" #include "include/context.h"
@ -112,7 +113,7 @@ std::vector<int> NetRunner::FillInputData(const std::vector<DataLabelTuple> &dat
int label = 0; int label = 0;
char *data = nullptr; char *data = nullptr;
std::tie(data, label) = dataset[idx]; std::tie(data, label) = dataset[idx];
memcpy(input_data + i * data_size_, data, data_size_); std::memcpy(input_data + i * data_size_, data, data_size_);
labels[i * num_of_classes_ + label] = 1.0; // Model expects labels in onehot representation labels[i * num_of_classes_ + label] = 1.0; // Model expects labels in onehot representation
labels_vec.push_back(label); labels_vec.push_back(label);
} }

@ -17,6 +17,11 @@
#ifndef MINDSPORE_LITE_EXAMPLES_TRANSFER_LEARNING_SRC_UTILS_H_ #ifndef MINDSPORE_LITE_EXAMPLES_TRANSFER_LEARNING_SRC_UTILS_H_
#define MINDSPORE_LITE_EXAMPLES_TRANSFER_LEARNING_SRC_UTILS_H_ #define MINDSPORE_LITE_EXAMPLES_TRANSFER_LEARNING_SRC_UTILS_H_
// DEBUG should be defined because the source code use assert to test exception
// but the Code Analysis Tool not allow us to use assert directly
#define DEBUG TRUE
#ifdef DEBUG #ifdef DEBUG
#include <cassert> #include <cassert>
#define MS_ASSERT(f) assert(f) #define MS_ASSERT(f) assert(f)

@ -75,19 +75,19 @@ function Run_x86() {
echo ${model_name}'_train' >> "${run_x86_log_file}" echo ${model_name}'_train' >> "${run_x86_log_file}"
echo 'cd '${x86_path}'/mindspore-lite-'${version}'-train-linux-x64' >> "${run_x86_log_file}" echo 'cd '${x86_path}'/mindspore-lite-'${version}'-train-linux-x64' >> "${run_x86_log_file}"
cd ${x86_path}/mindspore-lite-${version}-train-linux-x64 || return 1 cd ${x86_path}/mindspore-lite-${version}-train-linux-x64 || return 1
echo 'LD_LIBRARY_PATH='${LD_LIBRARY_PATH}':./lib:./third_party/libjpeg-turbo/lib:./third_party/opencv/lib;./net_train/net_train --epochs='${epoch_num}' --modelFile='${ms_models_path}'/'${model_name}'_train.ms --inDataFile='${train_io_path}/${model_name}_input1.bin,${train_io_path}/${model_name}_input2.bin' --expectedDataFile='${train_io_path}'/'${model_name}'_outputs.bin --exportFile='${ms_models_path}'/'${model_name}'_train_exported.ms' >> "${run_x86_log_file}" echo 'LD_LIBRARY_PATH='${LD_LIBRARY_PATH}':./lib:./third_party/libjpeg-turbo/lib:./third_party/opencv/lib;./benchmark_train/benchmark_train --epochs='${epoch_num}' --modelFile='${ms_models_path}'/'${model_name}'_train.ms --inDataFile='${train_io_path}/${model_name}_input1.bin,${train_io_path}/${model_name}_input2.bin' --expectedDataFile='${train_io_path}'/'${model_name}'_outputs.bin --exportFile='${ms_models_path}'/'${model_name}'_train_exported.ms' >> "${run_x86_log_file}"
echo '-------------------------------------------------------------------------------' >> "${run_x86_log_file}" echo '-------------------------------------------------------------------------------' >> "${run_x86_log_file}"
LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:./lib:./third_party/libjpeg-turbo/lib:./third_party/opencv/lib \ LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:./lib:./third_party/libjpeg-turbo/lib:./third_party/opencv/lib \
${run_valgrind}./net_train/net_train \ ${run_valgrind}./benchmark_train/benchmark_train \
--modelFile=${ms_models_path}/${model_name}_train.ms \ --modelFile=${ms_models_path}/${model_name}_train.ms \
--inDataFile=${train_io_path}/${model_name}_input1.bin,${train_io_path}/${model_name}_input2.bin \ --inDataFile=${train_io_path}/${model_name}_input1.bin,${train_io_path}/${model_name}_input2.bin \
--expectedDataFile=${train_io_path}/${model_name}_outputs.bin \ --expectedDataFile=${train_io_path}/${model_name}_outputs.bin \
--exportFile=${ms_models_path}/${model_name}_train_exported.ms >> "${run_x86_log_file}" \ --exportFile=${ms_models_path}/${model_name}_train_exported.ms >> "${run_x86_log_file}" \
--epochs=${epoch_num} --epochs=${epoch_num}
if [ $? = 0 ]; then if [ $? = 0 ]; then
run_result='x86: '${model_name}'_train pass'; echo ${run_result} >> ${run_net_train_result_file} run_result='x86: '${model_name}'_train pass'; echo ${run_result} >> ${run_benchmark_train_result_file}
else else
run_result='x86: '${model_name}'_train failed'; echo ${run_result} >> ${run_net_train_result_file} run_result='x86: '${model_name}'_train failed'; echo ${run_result} >> ${run_benchmark_train_result_file}
fi fi
done < ${models_mindspore_train_config} done < ${models_mindspore_train_config}
} }
@ -95,7 +95,7 @@ function Run_x86() {
# Run on arm platform: # Run on arm platform:
# Gets a parameter - arm64/arm32 # Gets a parameter - arm64/arm32
function Run_arm() { function Run_arm() {
tmp_dir=/data/local/tmp/net_train_test tmp_dir=/data/local/tmp/benchmark_train_test
if [ "$1" == arm64 ]; then if [ "$1" == arm64 ]; then
arm_path=${arm64_path} arm_path=${arm64_path}
process_unit="aarch64" process_unit="aarch64"
@ -122,34 +122,34 @@ function Run_arm() {
tar -zxf mindspore-lite-${version_arm}-train-android-${process_unit}.tar.gz || exit 1 tar -zxf mindspore-lite-${version_arm}-train-android-${process_unit}.tar.gz || exit 1
# If build with minddata, copy the minddata related libs # If build with minddata, copy the minddata related libs
cd ${net_train_test_path} || exit 1 cd ${benchmark_train_test_path} || exit 1
if [ -f ${arm_path}/mindspore-lite-${version_arm}-train-android-${process_unit}/lib/libminddata-lite.so ]; then if [ -f ${arm_path}/mindspore-lite-${version_arm}-train-android-${process_unit}/lib/libminddata-lite.so ]; then
cp -a ${arm_path}/mindspore-lite-${version_arm}-train-android-${process_unit}/third_party/libjpeg-turbo/lib/libjpeg.so ${net_train_test_path}/libjpeg.so || exit 1 cp -a ${arm_path}/mindspore-lite-${version_arm}-train-android-${process_unit}/third_party/libjpeg-turbo/lib/libjpeg.so ${benchmark_train_test_path}/libjpeg.so || exit 1
cp -a ${arm_path}/mindspore-lite-${version_arm}-train-android-${process_unit}/third_party/libjpeg-turbo/lib/libturbojpeg.so ${net_train_test_path}/libturbojpeg.so || exit 1 cp -a ${arm_path}/mindspore-lite-${version_arm}-train-android-${process_unit}/third_party/libjpeg-turbo/lib/libturbojpeg.so ${benchmark_train_test_path}/libturbojpeg.so || exit 1
cp -a ${arm_path}/mindspore-lite-${version_arm}-train-android-${process_unit}/third_party/opencv/lib/libopencv_core.so ${net_train_test_path}/libopencv_core.so || exit 1 cp -a ${arm_path}/mindspore-lite-${version_arm}-train-android-${process_unit}/third_party/opencv/lib/libopencv_core.so ${benchmark_train_test_path}/libopencv_core.so || exit 1
cp -a ${arm_path}/mindspore-lite-${version_arm}-train-android-${process_unit}/third_party/opencv/lib/libopencv_imgcodecs.so ${net_train_test_path}/libopencv_imgcodecs.so || exit 1 cp -a ${arm_path}/mindspore-lite-${version_arm}-train-android-${process_unit}/third_party/opencv/lib/libopencv_imgcodecs.so ${benchmark_train_test_path}/libopencv_imgcodecs.so || exit 1
cp -a ${arm_path}/mindspore-lite-${version_arm}-train-android-${process_unit}/third_party/opencv/lib/libopencv_imgproc.so ${net_train_test_path}/libopencv_imgproc.so || exit 1 cp -a ${arm_path}/mindspore-lite-${version_arm}-train-android-${process_unit}/third_party/opencv/lib/libopencv_imgproc.so ${benchmark_train_test_path}/libopencv_imgproc.so || exit 1
cp -a ${arm_path}/mindspore-lite-${version_arm}-train-android-${process_unit}/lib/libminddata-lite.so ${net_train_test_path}/libminddata-lite.so || exit 1 cp -a ${arm_path}/mindspore-lite-${version_arm}-train-android-${process_unit}/lib/libminddata-lite.so ${benchmark_train_test_path}/libminddata-lite.so || exit 1
fi fi
if [ "$1" == arm64 ]; then if [ "$1" == arm64 ]; then
cp -a ${arm_path}/mindspore-lite-${version_arm}-train-android-${process_unit}/third_party/hiai_ddk/lib/libhiai.so ${net_train_test_path}/libhiai.so || exit 1 cp -a ${arm_path}/mindspore-lite-${version_arm}-train-android-${process_unit}/third_party/hiai_ddk/lib/libhiai.so ${benchmark_train_test_path}/libhiai.so || exit 1
cp -a ${arm_path}/mindspore-lite-${version_arm}-train-android-${process_unit}/third_party/hiai_ddk/lib/libhiai_ir.so ${net_train_test_path}/libhiai_ir.so || exit 1 cp -a ${arm_path}/mindspore-lite-${version_arm}-train-android-${process_unit}/third_party/hiai_ddk/lib/libhiai_ir.so ${benchmark_train_test_path}/libhiai_ir.so || exit 1
cp -a ${arm_path}/mindspore-lite-${version_arm}-train-android-${process_unit}/third_party/hiai_ddk/lib/libhiai_ir_build.so ${net_train_test_path}/libhiai_ir_build.so || exit 1 cp -a ${arm_path}/mindspore-lite-${version_arm}-train-android-${process_unit}/third_party/hiai_ddk/lib/libhiai_ir_build.so ${benchmark_train_test_path}/libhiai_ir_build.so || exit 1
fi fi
cp -a ${arm_path}/mindspore-lite-${version_arm}-train-android-${process_unit}/lib/libmindspore-lite.so ${net_train_test_path}/libmindspore-lite.so || exit 1 cp -a ${arm_path}/mindspore-lite-${version_arm}-train-android-${process_unit}/lib/libmindspore-lite.so ${benchmark_train_test_path}/libmindspore-lite.so || exit 1
# if [ "$1" == arm64 ]; then # if [ "$1" == arm64 ]; then
# cp -a ${arm_path}/mindspore-lite-${version_arm}-runtime-${arm_type}-${process_unit}-train/lib/libmindspore-lite-fp16.so ${net_train_test_path}/libmindspore-lite-fp16.so || exit 1 # cp -a ${arm_path}/mindspore-lite-${version_arm}-runtime-${arm_type}-${process_unit}-train/lib/libmindspore-lite-fp16.so ${benchmark_train_test_path}/libmindspore-lite-fp16.so || exit 1
# cp -a ${arm_path}/mindspore-lite-${version_arm}-runtime-${arm_type}-${process_unit}-train/lib/libmindspore-lite-optimize.so ${net_train_test_path}/libmindspore-lite-optimize.so || exit 1 # cp -a ${arm_path}/mindspore-lite-${version_arm}-runtime-${arm_type}-${process_unit}-train/lib/libmindspore-lite-optimize.so ${benchmark_train_test_path}/libmindspore-lite-optimize.so || exit 1
# fi # fi
cp -a ${arm_path}/mindspore-lite-${version_arm}-train-android-${process_unit}/net_train/net_train ${net_train_test_path}/net_train || exit 1 cp -a ${arm_path}/mindspore-lite-${version_arm}-train-android-${process_unit}/benchmark_train/benchmark_train ${benchmark_train_test_path}/benchmark_train || exit 1
# adb push all needed files to the phone # adb push all needed files to the phone
adb -s ${device_id} push ${net_train_test_path} /data/local/tmp/ > ${adb_push_log_file} adb -s ${device_id} push ${benchmark_train_test_path} /data/local/tmp/ > ${adb_push_log_file}
# run adb ,run session ,check the result: # run adb ,run session ,check the result:
echo 'cd /data/local/tmp/net_train_test' > ${adb_cmd_file} echo 'cd /data/local/tmp/benchmark_train_test' > ${adb_cmd_file}
echo 'chmod 777 net_train' >> ${adb_cmd_file} echo 'chmod 777 benchmark_train' >> ${adb_cmd_file}
adb -s ${device_id} shell < ${adb_cmd_file} adb -s ${device_id} shell < ${adb_cmd_file}
@ -160,11 +160,11 @@ function Run_arm() {
continue continue
fi fi
# run net_train test without clib data # run benchmark_train test without clib data
echo ${model_name}'_train' >> "${run_arm_log_file}" echo ${model_name}'_train' >> "${run_arm_log_file}"
adb -s ${device_id} push ${train_io_path}/${model_name}_input*.bin ${train_io_path}/${model_name}_outputs.bin /data/local/tmp/net_train_test >> ${adb_push_log_file} adb -s ${device_id} push ${train_io_path}/${model_name}_input*.bin ${train_io_path}/${model_name}_outputs.bin /data/local/tmp/benchmark_train_test >> ${adb_push_log_file}
echo 'cd /data/local/tmp/net_train_test' > ${adb_cmd_run_file} echo 'cd /data/local/tmp/benchmark_train_test' > ${adb_cmd_run_file}
echo 'chmod 777 net_train' >> ${adb_cmd_run_file} echo 'chmod 777 benchmark_train' >> ${adb_cmd_run_file}
if [ "$1" == arm64 ]; then if [ "$1" == arm64 ]; then
echo 'cp /data/local/tmp/libc++_shared.so ./' >> ${adb_cmd_run_file} echo 'cp /data/local/tmp/libc++_shared.so ./' >> ${adb_cmd_run_file}
elif [ "$1" == arm32 ]; then elif [ "$1" == arm32 ]; then
@ -173,7 +173,7 @@ function Run_arm() {
echo "rm -f ${tmp_dir}/${model_name}_train_exported.ms" >> ${run_arm_log_file} echo "rm -f ${tmp_dir}/${model_name}_train_exported.ms" >> ${run_arm_log_file}
echo "rm -f ${tmp_dir}/${model_name}_train_exported.ms" >> ${adb_cmd_run_file} echo "rm -f ${tmp_dir}/${model_name}_train_exported.ms" >> ${adb_cmd_run_file}
adb_cmd=$(cat <<-ENDM adb_cmd=$(cat <<-ENDM
export LD_LIBRARY_PATH=./:/data/local/tmp/:/data/local/tmp/net_train_test;./net_train \ export LD_LIBRARY_PATH=./:/data/local/tmp/:/data/local/tmp/benchmark_train_test;./benchmark_train \
--epochs=${epoch_num} \ --epochs=${epoch_num} \
--modelFile=${model_name}_train.ms \ --modelFile=${model_name}_train.ms \
--inDataFile=${tmp_dir}/${model_name}_input1.bin,${tmp_dir}/${model_name}_input2.bin \ --inDataFile=${tmp_dir}/${model_name}_input1.bin,${tmp_dir}/${model_name}_input2.bin \
@ -186,9 +186,9 @@ ENDM
adb -s ${device_id} shell < ${adb_cmd_run_file} >> ${run_arm_log_file} adb -s ${device_id} shell < ${adb_cmd_run_file} >> ${run_arm_log_file}
# TODO: change to arm_type # TODO: change to arm_type
if [ $? = 0 ]; then if [ $? = 0 ]; then
run_result=$1': '${model_name}'_train pass'; echo ${run_result} >> ${run_net_train_result_file} run_result=$1': '${model_name}'_train pass'; echo ${run_result} >> ${run_benchmark_train_result_file}
else else
run_result=$1': '${model_name}'_train failed'; echo ${run_result} >> ${run_net_train_result_file}; return 1 run_result=$1': '${model_name}'_train failed'; echo ${run_result} >> ${run_benchmark_train_result_file}; return 1
fi fi
done < ${models_mindspore_train_config} done < ${models_mindspore_train_config}
} }
@ -218,7 +218,7 @@ function Print_Result() {
basepath=$(pwd) basepath=$(pwd)
echo ${basepath} echo ${basepath}
# Example:run_net_train.sh -r /home/emir/Work/TestingEnv/release -m /home/emir/Work/TestingEnv/train_models -i /home/emir/Work/TestingEnv/train_io -d "8KE5T19620002408" # Example:run_benchmark_train.sh -r /home/emir/Work/TestingEnv/release -m /home/emir/Work/TestingEnv/train_models -i /home/emir/Work/TestingEnv/train_io -d "8KE5T19620002408"
# For running on arm64, use -t to set platform tools path (for using adb commands) # For running on arm64, use -t to set platform tools path (for using adb commands)
epoch_num=1 epoch_num=1
train_io_path="" train_io_path=""
@ -336,9 +336,9 @@ else
fi fi
# Write net_train result to temp file # Write benchmark_train result to temp file
run_net_train_result_file=${logs_path}/run_net_train_result.txt run_benchmark_train_result_file=${logs_path}/run_benchmark_train_result.txt
echo ' ' > ${run_net_train_result_file} echo ' ' > ${run_benchmark_train_result_file}
# Create log files # Create log files
run_x86_log_file=${logs_path}/run_x86_log.txt run_x86_log_file=${logs_path}/run_x86_log.txt
@ -357,11 +357,11 @@ adb_cmd_arm32_file=${logs_path}/adb_arm32_cmd.txt
adb_cmd_arm32_run_file=${logs_path}/adb_arm32_cmd_run.txt adb_cmd_arm32_run_file=${logs_path}/adb_arm32_cmd_run.txt
# Copy the MindSpore models: # Copy the MindSpore models:
echo "Push files to net_train_test folder and run net_train" echo "Push files to benchmark_train_test folder and run benchmark_train"
net_train_test_path=${basepath}/net_train_test benchmark_train_test_path=${basepath}/benchmark_train_test
rm -rf ${net_train_test_path} rm -rf ${benchmark_train_test_path}
mkdir -p ${net_train_test_path} mkdir -p ${benchmark_train_test_path}
cp -a ${ms_models_path}/*.ms ${net_train_test_path} || exit 1 cp -a ${ms_models_path}/*.ms ${benchmark_train_test_path} || exit 1
# Run on x86 # Run on x86
echo "start Run x86 ..." echo "start Run x86 ..."
@ -371,7 +371,7 @@ sleep 1
# wait ${Run_x86_PID} # wait ${Run_x86_PID}
cat ${run_net_train_result_file} cat ${run_benchmark_train_result_file}
wait ${Run_x86_PID} wait ${Run_x86_PID}
Run_x86_status=$? Run_x86_status=$?
@ -395,11 +395,11 @@ function Print_Benchmark_Result() {
while read line; do while read line; do
arr=("${line}") arr=("${line}")
printf "%-20s %-100s %-7s\n" ${arr[0]} ${arr[1]} ${arr[2]} printf "%-20s %-100s %-7s\n" ${arr[0]} ${arr[1]} ${arr[2]}
done < ${run_net_train_result_file} done < ${run_benchmark_train_result_file}
MS_PRINT_TESTCASE_END_MSG MS_PRINT_TESTCASE_END_MSG
} }
# Check net_train result and return value # Check benchmark_train result and return value
if [[ ${Run_x86_status} != 0 ]];then if [[ ${Run_x86_status} != 0 ]];then
echo "Run_x86 failed" echo "Run_x86 failed"
cat ${run_x86_log_file} cat ${run_x86_log_file}

@ -5,34 +5,34 @@ set(COMMON_SRC
${CMAKE_CURRENT_SOURCE_DIR}/../../src/common/utils.cc ${CMAKE_CURRENT_SOURCE_DIR}/../../src/common/utils.cc
) )
add_executable(net_train add_executable(benchmark_train
${CMAKE_CURRENT_SOURCE_DIR}/main.cc ${CMAKE_CURRENT_SOURCE_DIR}/main.cc
${CMAKE_CURRENT_SOURCE_DIR}/net_train.cc ${CMAKE_CURRENT_SOURCE_DIR}/net_train.cc
${COMMON_SRC}) ${COMMON_SRC})
if (WIN32) if (WIN32)
add_dependencies(net_train fbs_src mindspore-lite_static) add_dependencies(benchmark_train fbs_src mindspore-lite_static)
else () else ()
add_dependencies(net_train fbs_src) add_dependencies(benchmark_train fbs_src)
endif () endif ()
if (PLATFORM_ARM32 OR PLATFORM_ARM64) if (PLATFORM_ARM32 OR PLATFORM_ARM64)
target_link_libraries(net_train mindspore-lite) target_link_libraries(benchmark_train mindspore-lite)
else() else()
if (WIN32) if (WIN32)
target_link_libraries(net_train mindspore-lite_static pthread cpu_kernel_mid nnacl_mid) target_link_libraries(benchmark_train mindspore-lite_static pthread cpu_kernel_mid nnacl_mid)
else () else ()
target_link_libraries(net_train mindspore-lite pthread) target_link_libraries(benchmark_train mindspore-lite pthread)
endif () endif ()
endif() endif()
if (PLATFORM_ARM32 OR PLATFORM_ARM64) if (PLATFORM_ARM32 OR PLATFORM_ARM64)
install(TARGETS net_train install(TARGETS benchmark_train
RUNTIME DESTINATION ${MAIN_DIR}-${RUNTIME_COMPONENT_NAME}/net_train COMPONENT ${RUNTIME_COMPONENT_NAME}) RUNTIME DESTINATION ${MAIN_DIR}-${RUNTIME_COMPONENT_NAME}/benchmark_train COMPONENT ${RUNTIME_COMPONENT_NAME})
else() else()
if (WIN32) if (WIN32)
install(TARGETS net_train install(TARGETS benchmark_train
RUNTIME DESTINATION ${MAIN_DIR}-${RUNTIME_COMPONENT_NAME}/net_train COMPONENT ${RUNTIME_COMPONENT_NAME}) RUNTIME DESTINATION ${MAIN_DIR}-${RUNTIME_COMPONENT_NAME}/benchmark_train COMPONENT ${RUNTIME_COMPONENT_NAME})
else () else ()
install(TARGETS net_train install(TARGETS benchmark_train
RUNTIME DESTINATION ${MAIN_DIR}-${RUNTIME_COMPONENT_NAME}/net_train COMPONENT ${RUNTIME_COMPONENT_NAME}) RUNTIME DESTINATION ${MAIN_DIR}-${RUNTIME_COMPONENT_NAME}/benchmark_train COMPONENT ${RUNTIME_COMPONENT_NAME})
endif () endif ()
endif() endif()

@ -15,7 +15,7 @@
*/ */
#include <malloc.h> #include <malloc.h>
#include "tools/net_train/net_train.h" #include "tools/benchmark_train/net_train.h"
#include "include/version.h" #include "include/version.h"
int main(int argc, const char **argv) { int main(int argc, const char **argv) {

@ -14,7 +14,7 @@
* limitations under the License. * limitations under the License.
*/ */
#include "tools/net_train/net_train.h" #include "tools/benchmark_train/net_train.h"
#define __STDC_FORMAT_MACROS #define __STDC_FORMAT_MACROS
#include <cinttypes> #include <cinttypes>
#undef __STDC_FORMAT_MACROS #undef __STDC_FORMAT_MACROS

@ -67,6 +67,9 @@ MetaGraphT *Converter::Convert(const converter::Flags *flag) {
int status = modelImporter->Import(flag); int status = modelImporter->Import(flag);
ReturnCode::GetSingleReturnCode()->UpdateReturnCode(status); ReturnCode::GetSingleReturnCode()->UpdateReturnCode(status);
graph = modelImporter->GetResult(); graph = modelImporter->GetResult();
if (graph == nullptr) {
return nullptr;
}
graph->set_attr("graph_name", MakeValue("main_graph")); graph->set_attr("graph_name", MakeValue("main_graph"));
} else { } else {
MS_ASSERT(nullptr != modelParser); MS_ASSERT(nullptr != modelParser);

@ -80,12 +80,12 @@ function android_release_package()
if [ ${name} == "train" ] if [ ${name} == "train" ]
then then
mkdir -p ${dst_android_pkg_name}/net_train/aarch64/ mkdir -p ${dst_android_pkg_name}/benchmark_train/aarch64/
mkdir -p ${dst_android_pkg_name}/net_train/aarch32/ mkdir -p ${dst_android_pkg_name}/benchmark_train/aarch32/
mkdir -p ${dst_android_pkg_name}/minddata/third_party/libjpeg-turbo/lib/aarch64/ mkdir -p ${dst_android_pkg_name}/minddata/third_party/libjpeg-turbo/lib/aarch64/
mkdir -p ${dst_android_pkg_name}/minddata/third_party/libjpeg-turbo/lib/aarch32/ mkdir -p ${dst_android_pkg_name}/minddata/third_party/libjpeg-turbo/lib/aarch32/
cp ${src_arm64_pkg_name}/net_train/* ${dst_android_pkg_name}/net_train/aarch64/ cp ${src_arm64_pkg_name}/benchmark_train/* ${dst_android_pkg_name}/benchmark_train/aarch64/
cp ${src_arm32_pkg_name}/net_train/* ${dst_android_pkg_name}/net_train/aarch32/ cp ${src_arm32_pkg_name}/benchmark_train/* ${dst_android_pkg_name}/benchmark_train/aarch32/
cp ${src_arm64_pkg_name}/minddata/third_party/libjpeg-turbo/lib/* ${dst_android_pkg_name}/minddata/third_party/libjpeg-turbo/lib/aarch64/ cp ${src_arm64_pkg_name}/minddata/third_party/libjpeg-turbo/lib/* ${dst_android_pkg_name}/minddata/third_party/libjpeg-turbo/lib/aarch64/
cp ${src_arm32_pkg_name}/minddata/third_party/libjpeg-turbo/lib/* ${dst_android_pkg_name}/minddata/third_party/libjpeg-turbo/lib/aarch32/ cp ${src_arm32_pkg_name}/minddata/third_party/libjpeg-turbo/lib/* ${dst_android_pkg_name}/minddata/third_party/libjpeg-turbo/lib/aarch32/
fi fi

Loading…
Cancel
Save