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.
61 lines
1.3 KiB
Bash
Executable File
61 lines
1.3 KiB
Bash
Executable File
#!/bin/bash
|
|
set -ex
|
|
|
|
echo "tpu samples regression for cv183x platform"
|
|
|
|
if [ -z "$MODEL_PATH" ] || [ ! -e $MODEL_PATH ]; then
|
|
echo "MODEL_PATH $MODEL_PATH does not exist"
|
|
echo "Please set MODEL_PATH to cvimodel_regression dir"
|
|
exit 1
|
|
fi
|
|
|
|
if [ -z "$TPU_ROOT" ] || [ ! -e $TPU_ROOT ]; then
|
|
echo "TPU_ROOT $TPU_ROOT does not exist"
|
|
echo "Please set TPU_ROOT to cvitek_tpu_sdk dir"
|
|
exit 1
|
|
fi
|
|
|
|
#export TPU_ENABLE_PMU=1
|
|
|
|
samples_list=(
|
|
"run_classifier.sh"
|
|
"run_classifier_fused_preprocess.sh"
|
|
#"run_classifier_multi_batch.sh"
|
|
)
|
|
|
|
samples_extra_list=(
|
|
#"run_alphapose.sh"
|
|
"run_alphapose_fused_preprocess.sh"
|
|
#"run_classifier_yuv420.sh"
|
|
"run_detector_yolov3_fused_preprocess.sh"
|
|
#"run_detector_yolov3.sh"
|
|
"run_detector_yolov5_fused_preprocess.sh"
|
|
#"run_detector_yolov5.sh"
|
|
#"run_detector_yolox_s.sh"
|
|
"run_insightface_fused_preprocess.sh"
|
|
)
|
|
|
|
function test_sample() {
|
|
for sample in $*
|
|
do
|
|
echo $sample
|
|
sh -ex ./$sample
|
|
test $? -ne 0 && echo "$sample failed !!" && return 1
|
|
done
|
|
return 0
|
|
}
|
|
|
|
pushd $TPU_ROOT/samples
|
|
# test samples
|
|
test_sample ${samples_list[@]}
|
|
test $? -ne 0 && echo "test samples failed !!" && popd && exit 1
|
|
popd
|
|
|
|
pushd $TPU_ROOT/samples/samples_extra
|
|
# test samples_extra
|
|
test_sample ${samples_extra_list[@]}
|
|
test $? -ne 0 && echo "test samples extra failed !!" && popd && exit 1
|
|
popd
|
|
|
|
exit 0
|