write output to file

Author: kirklandsign

examples/demo-apps/android/MiniBench/app/src/main/java/org/pytorch/minibench/BenchmarkActivity.java

@@ -11,18 +11,36 @@
 import android.app.Activity;
 import android.content.Intent;
 import android.os.Bundle;
-import android.util.Log;
 
 import org.pytorch.executorch.Module;
 
+import java.io.FileWriter;
+import java.io.IOException;
+
 public class BenchmarkActivity extends Activity {
     @Override
     protected void onCreate(Bundle savedInstanceState) {
         super.onCreate(savedInstanceState);
-        Log.e("ABENCHBENCHABENCHBENCHABENCHBENCH", "ABENCHBENCHABENCHBENCHABENCHBENCH");
         Intent intent = getIntent();
         String modelPath = intent.getStringExtra("model_path");
+        int numIter = intent.getIntExtra("num_iter", 10);
+
+        // TODO: Format the string with a parsable format
+        StringBuilder resultText = new StringBuilder();
+
         Module module = Module.load(modelPath);
-        module.forward();
+        for (int i = 0; i < numIter; i++) {
+            long start = System.currentTimeMillis();
+            module.forward();
+            long forwardMs = System.currentTimeMillis() - start;
+            resultText.append(forwardMs).append(";");
+        }
+
+        try (FileWriter writer = new FileWriter(getFilesDir() + "/benchmark_results.txt")) {
+            writer.write(resultText.toString());
+        } catch (IOException e) {
+            e.printStackTrace();
+        }
+
     }
 }