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.8 KiB
Bash
70 lines
1.8 KiB
Bash
set -e
|
|
|
|
echo "cvimodels regression for cv181x platform"
|
|
|
|
if [[ -z "$MODEL_PATH" ]]; then
|
|
MODEL_PATH=$TPU_ROOT/../cvimodel_regression_int8_cv181x
|
|
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_v1 mobilenet_v2_cf squeezenet_v1.1_cf shufflenet_v2 googlenet_cf densenet121-12 nasnet_mobile blazeface retinaface_mnet_with_det mobilenetv2_ssd_cf yolov3_tiny yolov8n"
|
|
fi
|
|
|
|
if [ "$total_ion_size" -gt "35000000" ]; then
|
|
model_list="$model_list densenet201 se-resnet50 resnext50_cf efficientdet-d0 pp_yolox yolov5s inception_v3 resnet50_v1"
|
|
fi
|
|
|
|
if [ "$total_ion_size" -gt "45000000" ]; then
|
|
model_list="$model_list arcface_res50 alphapose_res50 retinaface pp_yoloe_m"
|
|
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}_in_f32.npz \
|
|
--model $MODEL_PATH/${model}_bs1.cvimodel \
|
|
--reference $MODEL_PATH/${model}_bs1_out_all.npz 2>&1 | tee $model.log
|
|
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
|