!20003 [lite]1.3 add header file for converter
Merge pull request !20003 from 徐安越/r1.3
This commit is contained in:
commit
54c93a227f
|
@ -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
|
||||
|
|
|
@ -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)
|
||||
|
|
|
@ -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_
|
||||
|
|
Loading…
Reference in New Issue