e25f20f7a3
commit 3f4938648950a7f3bf9a19c320ca9fae7c52de20 Author: sophgo-forum-service <forum_service@sophgo.com> Date: Mon May 13 13:44:23 2024 +0800 [feat] cviruntime opensource for cv18xx soc. - a4b6a3, add cumsum and gatherelements_pt.
70 lines
1.7 KiB
Bash
Executable File
70 lines
1.7 KiB
Bash
Executable File
set -e
|
|
|
|
echo "cvimodels regression for cv181x platform"
|
|
|
|
if [[ -z "$MODEL_PATH" ]]; then
|
|
MODEL_PATH=$TPU_ROOT/../cvimodel_regression
|
|
fi
|
|
if [ ! -e $MODEL_PATH ]; then
|
|
echo "MODEL_PATH $MODEL_PATH does not exist"
|
|
echo "Please set MODEL_PATH to cvimodel_regression dir"
|
|
exit 1
|
|
fi
|
|
export MODEL_PATH=$MODEL_PATH
|
|
|
|
if [ -f "/sys/kernel/debug/ion/cvi_carveout_heap_dump/total_mem" ]; then
|
|
total_ion_size=$(cat /sys/kernel/debug/ion/cvi_carveout_heap_dump/total_mem)
|
|
else
|
|
# if ion size is unknown then execute basic tests.
|
|
total_ion_size=20000001
|
|
fi
|
|
|
|
# ION requirement >= 20 MB
|
|
if [ "$total_ion_size" -gt "20000000" ]; then
|
|
model_list="resnet18 mobilenet_v1 mobilenet_v2 squeezenet_v1.1 shufflenet_v2 googlenet densenet_121 efficientnet-lite_b0 nasnet_mobile retinaface_mnet25 retinaface_mnet25_600 mobilenet_ssd yolo_v3_tiny"
|
|
fi
|
|
|
|
if [ "$total_ion_size" -gt "35000000" ]; then
|
|
model_list="$model_list densenet_201 senet_res50 resnext50 yolo_v1_448 inception_v3"
|
|
fi
|
|
|
|
if [ "$total_ion_size" -gt "45000000" ]; then
|
|
model_list="$model_list arcface_res50"
|
|
fi
|
|
|
|
# turn on PMU
|
|
export TPU_ENABLE_PMU=1
|
|
|
|
if [ ! -e sdk_regression_out ]; then
|
|
mkdir sdk_regression_out
|
|
fi
|
|
|
|
ERR=0
|
|
cd sdk_regression_out
|
|
|
|
if [ -z $1 ]; then
|
|
for model in ${model_list}
|
|
do
|
|
echo "test $model"
|
|
model_runner \
|
|
--input $MODEL_PATH/${model}_bs1_in_fp32.npz \
|
|
--model $MODEL_PATH/${model}_bs1.cvimodel \
|
|
--reference $MODEL_PATH/${model}_bs1_out_all.npz 2>&1
|
|
if [ "$?" -ne "0" ]; then
|
|
echo "$model test FAILED" >> verdict.log
|
|
ERR=1
|
|
else
|
|
echo "$model test PASSED" >> verdict.log
|
|
fi
|
|
done
|
|
fi
|
|
|
|
# VERDICT
|
|
if [ $ERR -eq 0 ]; then
|
|
echo $0 ALL TEST PASSED
|
|
else
|
|
echo $0 FAILED
|
|
fi
|
|
|
|
exit $ERR
|