From 47530814691140c0f529d4e134df9bcb64bbf466 Mon Sep 17 00:00:00 2001
From: Pun Butrach <pun.butrach@gmail.com>
Date: Sun, 8 Feb 2026 03:50:54 +0700
Subject: [PATCH] build: Make baselineprofile compile all variants into one

Signed-off-by: Pun Butrach <pun.butrach@gmail.com>
---
 .../src/main/java/app/lawnchair/baseline/Constants.kt     | 8 +++++++-
 build.gradle                                              | 6 ++++++
 2 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/baseline-profile/src/main/java/app/lawnchair/baseline/Constants.kt b/baseline-profile/src/main/java/app/lawnchair/baseline/Constants.kt
index 90ba5cef2b..d5c6cef48c 100644
--- a/baseline-profile/src/main/java/app/lawnchair/baseline/Constants.kt
+++ b/baseline-profile/src/main/java/app/lawnchair/baseline/Constants.kt
@@ -1,5 +1,11 @@
 package app.lawnchair.baseline
 
+import androidx.test.platform.app.InstrumentationRegistry
+
 object Constants {
-    val PACKAGE_NAME = "app.lawnchair"
+    val PACKAGE_NAME: String
+        get() = InstrumentationRegistry.getArguments()
+            .getString("androidx.benchmark.targetPackageName")
+            ?: InstrumentationRegistry.getInstrumentation().targetContext.packageName
+                .removeSuffix(".baseline")
 }
diff --git a/build.gradle b/build.gradle
index f242466e63..05e0f8296d 100644
--- a/build.gradle
+++ b/build.gradle
@@ -354,6 +354,12 @@ android {
     }
 }
 
+baselineProfile {
+    saveInSrc = true
+    mergeIntoMain = true
+    automaticGenerationDuringBuild = false
+}
+
 composeCompiler {
     stabilityConfigurationFiles.addAll(
         layout.projectDirectory.file("compose_compiler_config.conf"),