Add basic progress reporting
Bug: 132860960
Test: NeuralNetworksApiBenchmark.apk
Merged-In: I430ecb8aa3e57ef9f1ffda3a45fac29b312fc9da
Change-Id: I430ecb8aa3e57ef9f1ffda3a45fac29b312fc9da
(cherry picked from commit 085200181f33288aab942f2084336e30c671af73)
diff --git a/src/com/android/nn/benchmark/app/NNBenchmark.java b/src/com/android/nn/benchmark/app/NNBenchmark.java
index edb84b8..2e912f8 100644
--- a/src/com/android/nn/benchmark/app/NNBenchmark.java
+++ b/src/com/android/nn/benchmark/app/NNBenchmark.java
@@ -186,8 +186,20 @@
mTest.destroy();
}
+ TestModels.TestModelEntry testModel =
+ TestModels.modelsList().get(mTestList[ct]);
+ int testNumber = ct + 1;
+ runOnUiThread(() -> {
+ mTextView.setText(
+ String.format(
+ "Running test %d of %d: %s",
+ testNumber,
+ mTestList.length,
+ testModel.toString()));
+ });
+
// Select the next test
- mTest = changeTest(mTestList[ct]);
+ mTest = changeTest(testModel);
// If the user selected the "long pause" option, wait
if (mTogglePause) {
@@ -246,17 +258,13 @@
return tb;
}
- NNTestBase changeTest(int id) {
- return changeTest(TestModels.modelsList().get(id));
- }
-
@Override
protected void onCreate(Bundle savedInstanceState) {
super.onCreate(savedInstanceState);
- TextView textView = new TextView(this);
- textView.setTextSize(20);
- textView.setText("NN BenchMark Running.");
- setContentView(textView);
+ mTextView = new TextView(this);
+ mTextView.setTextSize(20);
+ mTextView.setText("Running NN benchmark...");
+ setContentView(mTextView);
getWindow().addFlags(WindowManager.LayoutParams.FLAG_KEEP_SCREEN_ON);
}