diff --git a/extension/benchmark/android/benchmark/app/build.gradle.kts b/extension/benchmark/android/benchmark/app/build.gradle.kts index dcf99ca9cd0..6f35d9a34f6 100644 --- a/extension/benchmark/android/benchmark/app/build.gradle.kts +++ b/extension/benchmark/android/benchmark/app/build.gradle.kts @@ -10,12 +10,12 @@ plugins { id("com.android.application") } android { namespace = "org.pytorch.minibench" - compileSdk = 34 + compileSdk = 35 defaultConfig { applicationId = "org.pytorch.minibench" - minSdk = 28 - targetSdk = 33 + minSdk = 35 + targetSdk = 35 versionCode = 1 versionName = "1.0" diff --git a/extension/benchmark/android/benchmark/app/src/main/java/org/pytorch/minibench/BenchmarkActivity.java b/extension/benchmark/android/benchmark/app/src/main/java/org/pytorch/minibench/BenchmarkActivity.java index 4b2ba56099e..647e49b6cf8 100644 --- a/extension/benchmark/android/benchmark/app/src/main/java/org/pytorch/minibench/BenchmarkActivity.java +++ b/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; @@ -103,6 +104,8 @@ protected void onPostExecute(Void aVoid) { 0.0f)); // Load status results.add(new BenchmarkMetric(benchmarkModel, "load_status", stats.errorCode, 0)); + // RAM RSS usage + results.add(new BenchmarkMetric(benchmarkModel, "peak_inference_mem_usage(mb)", Debug.getRss(), 0)); try (FileWriter writer = new FileWriter(getFilesDir() + "/benchmark_results.json")) { Gson gson = new Gson();