Add a model loading/unloading stress test
To ensure stability of NNAPI, for every model, this test continuously loads
the model for 30 minutes without doing any inference.
Bug: 123042197
Test: build_and_run_benchmark.sh model-loading-stress
Change-Id: I6b8522b627092e7cafb8370037aa5d66d71f76aa
diff --git a/build_and_run_benchmark.sh b/build_and_run_benchmark.sh
index 83ca2ab..ab40c49 100755
--- a/build_and_run_benchmark.sh
+++ b/build_and_run_benchmark.sh
@@ -4,18 +4,21 @@
#
# Output is logged to a temporary folder and summarized in txt and JSON formats.
-case "$1" in
- ""|scoring)
- MODE=scoring
+MODE="${1:=scoring}"
+
+case "$MODE" in
+ scoring)
CLASS=com.android.nn.benchmark.app.NNScoringTest
;;
- stress)
- MODE=stress
- CLASS=com.android.nn.benchmark.app.NNStressTest
+ inference-stress)
+ CLASS=com.android.nn.benchmark.app.NNInferenceStressTest
+ ;;
+ model-loading-stress)
+ CLASS=com.android.nn.benchmark.app.NNModelLoadingStressTest
;;
*)
echo "Unknown execution mode: $1"
- echo "Known modes: scoring (default), stress"
+ echo "Known modes: scoring (default), inference-stress, model-loading-stress"
exit 1
;;
esac