[minibench] Add PSS data to result

Differential Revision: D70721477

Pull Request resolved: #8984

Author: kirklandsign

extension/benchmark/android/benchmark/app/build.gradle.kts

@@ -14,8 +14,8 @@ android {
 
   defaultConfig {
     applicationId = "org.pytorch.minibench"
-    minSdk = 28
-    targetSdk = 33
+    minSdk = 34
+    targetSdk = 34
     versionCode = 1
     versionName = "1.0"
 

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

@@ -12,6 +12,7 @@
 import android.content.Intent;
 import android.os.AsyncTask;
 import android.os.Bundle;
+import android.os.Debug;
 import android.system.ErrnoException;
 import android.system.Os;
 import com.google.gson.Gson;
@@ -47,6 +48,8 @@ protected void onCreate(Bundle savedInstanceState) {
     int numIter = intent.getIntExtra("num_iter", 50);
     int numWarmupIter = intent.getIntExtra("num_warm_up_iter", 10);
 
+    long pssIdle = Debug.getPss();
+
     // TODO: Format the string with a parsable format
     Stats stats = new Stats();
 
@@ -103,6 +106,10 @@ protected void onPostExecute(Void aVoid) {
                 0.0f));
         // Load status
         results.add(new BenchmarkMetric(benchmarkModel, "load_status", stats.errorCode, 0));
+        // RAM PSS usage
+        results.add(
+            new BenchmarkMetric(
+                benchmarkModel, "ram_pss_usage(mb)", (Debug.getPss() - pssIdle) / 1024, 0));
 
         try (FileWriter writer = new FileWriter(getFilesDir() + "/benchmark_results.json")) {
           Gson gson = new Gson();