checkout nnie r1.3 branch name

This commit is contained in:
jianghui58 2021-08-25 15:58:04 +08:00
parent e428b5d29a
commit f4c786694f
1 changed files with 2 additions and 2 deletions

View File

@ -717,13 +717,13 @@ build_lite()
compile_nnie_script=${BASEPATH}/mindspore/lite/tools/providers/NNIE/Hi3516D/compile_nnie.sh
cd ${BASEPATH}/../
if [[ "${local_lite_platform}" == "x86_64" ]]; then
sh ${compile_nnie_script} -I x86_64 -b nnie_r1.3 -j $THREAD_NUM
sh ${compile_nnie_script} -I x86_64 -b nnie_3516_r1.3 -j $THREAD_NUM
if [[ $? -ne 0 ]]; then
echo "compile x86_64 for nnie failed."
exit 1
fi
elif [[ "${local_lite_platform}" == "arm32" ]]; then
sh ${compile_nnie_script} -I arm32 -b nnie_r1.3 -j $THREAD_NUM
sh ${compile_nnie_script} -I arm32 -b nnie_3516_r1.3 -j $THREAD_NUM
if [[ $? -ne 0 ]]; then
echo "compile arm32 for nnie failed."
exit 1