From b70322b5a6dc01759d89e9fe46346f4539a6eda2 Mon Sep 17 00:00:00 2001
From: Greyson Parrelli <greyson@signal.org>
Date: Thu, 7 May 2026 09:07:57 -0400
Subject: [PATCH] Fix baseline profile build.

---
 app/build.gradle.kts | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/app/build.gradle.kts b/app/build.gradle.kts
index aba3c36522..b9779cae8d 100644
--- a/app/build.gradle.kts
+++ b/app/build.gradle.kts
@@ -91,12 +91,15 @@ android.sourceSets.all {
   kotlinExt.sourceSets.findByName(name) ?: kotlinExt.sourceSets.create(name)
 }
 // AGP 9.0's built-in Kotlin doesn't pick up extra java.srcDir entries from Android
-// source sets, so add the testShared dir directly to test/androidTest Kotlin
-// compile tasks.
+// source sets, so add shared dirs directly to the relevant Kotlin compile tasks.
 tasks.withType(org.jetbrains.kotlin.gradle.tasks.KotlinCompile::class.java).configureEach {
-  if (name.contains("UnitTest") || name.contains("AndroidTest")) {
+  val isTestTask = name.contains("UnitTest") || name.contains("AndroidTest")
+  if (isTestTask) {
     source("$projectDir/src/testShared")
   }
+  if (!isTestTask && (name.contains("Mocked") || name.contains("Benchmark"))) {
+    source("$projectDir/src/benchmarkShared/java")
+  }
 }
 
 wire {