More updates

Author: kirklandsign

extension/android/executorch_android/src/androidTest/java/org/pytorch/executorch/LlmModuleInstrumentationTest.java

@@ -94,8 +94,9 @@ public void onResult(String result) {
             }
 
             @Override
-            public void onStats(float tps) {
-                LlmModuleInstrumentationTest.this.onStats(tps);
+            public void onStats(String result) {
+                // TODO: Calculate tps
+                // LlmModuleInstrumentationTest.this.onStats(tps);
             }
         });
 

extension/benchmark/android/benchmark/app/src/main/java/org/pytorch/minibench/LlmBenchmarkActivity.java

@@ -79,8 +79,9 @@ public void onModelLoaded(int status) {
   public void onTokenGenerated(String token) {}
 
   @Override
-  public void onStats(String stats) {
-    mStatsInfo.tokens = stats;
+  public void onStats(String result) {
+    // TODO: Add tps
+    // mStatsInfo.tokens = stats;
   }
 
   @Override

extension/benchmark/android/benchmark/app/src/main/java/org/pytorch/minibench/ModelRunner.java

@@ -68,8 +68,8 @@ public void onResult(String result) {
   }
 
   @Override
-  public void onStats(float tps) {
-    mCallback.onStats("tokens/second: " + tps);
+  public void onStats(String result) {
+    mCallback.onStats(result);
   }
 }
 

extension/benchmark/android/benchmark/app/src/main/java/org/pytorch/minibench/ModelRunnerCallback.java

@@ -18,7 +18,7 @@ public interface ModelRunnerCallback {
 
   void onTokenGenerated(String token);
 
-  void onStats(String token);
+  void onStats(String result);
 
   void onGenerationStopped();
 }