Improve test_all_tests scripts to run in every mode (FP32, FP16, INT8)

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
               Davide Sapienza <sapienza.dav@gmail.com>
This commit is contained in:
Micaela Verucchi
2020-04-08 15:29:49 +02:00
parent 56c03e3a12
commit 01e667d4f9
+28 -1
View File
@@ -1,10 +1,10 @@
#!/bin/bash #!/bin/bash
cd build cd build
rm *rt
RED='\033[1;31m' RED='\033[1;31m'
GREEN='\033[1;32m' GREEN='\033[1;32m'
ORANGE='\033[1;33m'
NC='\033[0m' # No Color NC='\033[0m' # No Color
function print_output { function print_output {
@@ -20,6 +20,31 @@ function print_output {
out_file=results.log out_file=results.log
rm $out_file rm $out_file
modes=( 1 ) # only FP32
# modes=( 1 2 ) # FP32 and FP16
# modes=( 1 2 3 ) # FP32, FP16 and INT8
for i in "${modes[@]}"
do
rm *rt
if [ $i -eq 1 ]
then
export TKDNN_MODE=FP32
echo -e "${ORANGE}Test FP32${NC}"
fi
if [ $i -eq 2 ]
then
export TKDNN_MODE=FP16
echo -e "${ORANGE}Test FP16${NC}"
fi
if [ $i -eq 3 ]
then
export TKDNN_MODE=INT8
export TKDNN_CALIB_LABEL_PATH=../demo/COCO_val2017/all_labels.txt
export TKDNN_CALIB_IMG_PATH=../demo/COCO_val2017/all_images.txt
echo -e "${ORANGE}Test INT8${NC}"
fi
./test_resnet101_cnet &>> $out_file ./test_resnet101_cnet &>> $out_file
res_resnet101_cnet=$? res_resnet101_cnet=$?
print_output $res_resnet101_cnet test_resnet101_cnet print_output $res_resnet101_cnet test_resnet101_cnet
@@ -84,4 +109,6 @@ print_output $res_dla34_cnet test_dla34_cnet
res_yolo3_coco4=$? res_yolo3_coco4=$?
print_output $res_yolo3_coco4 test_yolo3_coco4 print_output $res_yolo3_coco4 test_yolo3_coco4
done
echo "If errors occured, check logfile $out_file" echo "If errors occured, check logfile $out_file"