!20003 [lite]1.3 add header file for converter

Merge pull request !20003 from 徐安越/r1.3
This commit is contained in:
i-robot 2021-07-12 07:13:12 +00:00 committed by Gitee
commit 54c93a227f
3 changed files with 13 additions and 7 deletions

View File

@ -137,11 +137,10 @@ if [[ $backend == "all" || $backend == "x86-all" || $backend == "x86" || $backen
fi
if [[ $backend == "all" || $backend == "arm32_3516D" ]]; then
exit 0
# sh $cur_path/scripts/nnie/run_converter_nnie.sh -r $release_path -m $models_path -d $device_id -e $backend
# hi3516_status=$?
# if [[ $hi3516_status -ne 0 ]]; then
# echo "Run nnie hi3516 failed"
# exit 1
# fi
sh $cur_path/scripts/nnie/run_converter_nnie.sh -r $release_path -m $models_path -d $device_id -e $backend
hi3516_status=$?
if [[ $hi3516_status -ne 0 ]]; then
echo "Run nnie hi3516 failed"
exit 1
fi
fi

View File

@ -35,7 +35,10 @@ if(MSLITE_COMPILE_NNIE AND TARGET_HIMIX200 AND PLATFORM_ARM)
add_executable(benchmark
${CMAKE_CURRENT_SOURCE_DIR}/main.cc
${CMAKE_CURRENT_SOURCE_DIR}/run_benchmark.cc
${CMAKE_CURRENT_SOURCE_DIR}/benchmark_base.cc
${CMAKE_CURRENT_SOURCE_DIR}/benchmark.cc
${CMAKE_CURRENT_SOURCE_DIR}/benchmark_unified_api.cc
${COMMON_SRC})
add_dependencies(benchmark fbs_src)

View File

@ -312,5 +312,9 @@ class MS_API BenchmarkBase {
std::mt19937 random_engine_;
};
#ifdef SUPPORT_NNIE
int SvpSysInit();
int SvpSysExit();
#endif
} // namespace mindspore::lite
#endif // MINNIE_BENCHMARK_BENCHMARK_BASE_H_