blob: c697f1bc56e86bc183179757eb7ea2947b9859fe (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
#!/bin/bash
if [[ $# -le 3 ]]; then
echo "USAGE: $0 [nnkit-run path] [tflite backend path] [working directory] [prefix 0] [prefix 1] ..."
exit 255
fi
NNKIT_RUN_PATH="$1"; shift
TFLITE_BACKEND_PATH="$1"; shift
WORKDIR="$1"; shift
echo "-- Found nnkit-run: ${NNKIT_RUN_PATH}"
echo "-- Found tflite backend: ${TFLITE_BACKEND_PATH}"
echo "-- Found workdir: ${WORKDIR}"
TESTED=()
PASSED=()
FAILED=()
pushd "${WORKDIR}"
while [[ $# -ne 0 ]]; do
PREFIX="$1"; shift
TESTED+=("${PREFIX}")
PASSED_TAG="${PREFIX}.passed"
rm -f "${PASSED_TAG}"
cat > "${PREFIX}.log" <(
exec 2>&1
echo "'${NNKIT_RUN_PATH}' --backend '${TFLITE_BACKEND_PATH}' --backend-arg '${PREFIX}.tflite'"
"${NNKIT_RUN_PATH}" --backend "${TFLITE_BACKEND_PATH}" --backend-arg "${PREFIX}.tflite"
if [[ $? -eq 0 ]]; then
touch "${PASSED_TAG}"
fi
)
if [[ -f "${PASSED_TAG}" ]]; then
PASSED+=("$PREFIX")
else
FAILED+=("$PREFIX")
fi
done
popd
echo "SUMMARY: ${#PASSED[@]} PASS AND ${#FAILED[@]} FAIL AMONG ${#TESTED[@]} TESTS"
if [[ ${#TESTED[@]} -ne ${#PASSED[@]} ]]; then
echo "FAILED"
for TEST in "${FAILED[@]}"
do
echo "- ${TEST}"
done
exit 255
fi
exit 0
|