For optimized builds, use baseline profiles as startup profiles if no

startup profiles are explicitly provided. Guarded by ACL.

PiperOrigin-RevId: 834965863
Change-Id: I4fa7447e31e32cda5adc10b4dc47d415e521e0b3

Author: A Googler

rules/acls.bzl

@@ -78,6 +78,7 @@ load("//rules/acls:resource_translation_merging_rollout.bzl", "RESOURCE_TRANSLAT
 load("//rules/acls:shared_library_resource_linking.bzl", "SHARED_LIBRARY_RESOURCE_LINKING_ALLOWLIST")
 load("//rules/acls:stamp_signing.bzl", "STAMP_SIGNING_FALLBACK", "STAMP_SIGNING_ROLLOUT")
 load("//rules/acls:test_to_instrument_test_rollout.bzl", "TEST_TO_INSTRUMENT_TEST_FALLBACK", "TEST_TO_INSTRUMENT_TEST_ROLLOUT")
+load("//rules/acls:use_baseline_as_startup_profile.bzl", "USE_BASELINE_AS_STARTUP_PROFILE_FALLBACK", "USE_BASELINE_AS_STARTUP_PROFILE_ROLLOUT")
 
 visibility(PROJECT_VISIBILITY)
 
@@ -227,6 +228,9 @@ def _in_enable_exported_lint_checks(fqn):
 def _get_aapt2_feature_flags(_):
     return AAPT2_FEATURE_FLAGS
 
+def _use_baseline_as_startup_profile(fqn):
+    return matches(fqn, USE_BASELINE_AS_STARTUP_PROFILE_ROLLOUT_DICT) and not matches(fqn, USE_BASELINE_AS_STARTUP_PROFILE_FALLBACK_DICT)
+
 def make_dict(lst):
     """Do not use this method outside of acls directory."""
     return {t: True for t in lst}
@@ -297,6 +301,8 @@ STAMP_SIGNING_FALLBACK_DICT = make_dict(STAMP_SIGNING_FALLBACK)
 RESOURCE_TRANSLATION_MERGING_ROLLOUT_DICT = make_dict(RESOURCE_TRANSLATION_MERGING_ROLLOUT)
 RESOURCE_TRANSLATION_MERGING_FALLBACK_DICT = make_dict(RESOURCE_TRANSLATION_MERGING_FALLBACK)
 ENABLE_EXPORTED_LINT_CHECKS_DICT = make_dict(ENABLE_EXPORTED_LINT_CHECKS)
+USE_BASELINE_AS_STARTUP_PROFILE_ROLLOUT_DICT = make_dict(USE_BASELINE_AS_STARTUP_PROFILE_ROLLOUT)
+USE_BASELINE_AS_STARTUP_PROFILE_FALLBACK_DICT = make_dict(USE_BASELINE_AS_STARTUP_PROFILE_FALLBACK)
 
 def matches(fqn, dct):
     # Labels with workspace names ("@workspace//pkg:target") are not supported.
@@ -396,6 +402,7 @@ acls = struct(
     in_resource_translation_merging_rollout = _in_resource_translation_merging_rollout,
     in_enable_exported_lint_checks = _in_enable_exported_lint_checks,
     get_aapt2_feature_flags = _get_aapt2_feature_flags,
+    use_baseline_as_startup_profile = _use_baseline_as_startup_profile,
 )
 
 # Visible for testing

rules/acls/use_baseline_as_startup_profile.bzl

@@ -0,0 +1,24 @@
+# Copyright 2025 The Bazel Authors. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""Targets that use baseline profiles as startup profiles if no startup profile is provided."""
+
+load("//rules:visibility.bzl", "PROJECT_VISIBILITY")
+
+visibility(PROJECT_VISIBILITY)
+
+USE_BASELINE_AS_STARTUP_PROFILE_ROLLOUT = [
+    "//test/rules/android_binary/baselineprofiles/use_baseline_as_startup_profiles:__subpackages__",
+]
+
+USE_BASELINE_AS_STARTUP_PROFILE_FALLBACK = []

rules/android_binary/attrs.bzl

@@ -222,6 +222,10 @@ ATTRS = _attrs.replace(
                 allow_single_file = True,
                 cfg = android_platforms_transition,
             ),
+            _flags = attr.label(
+                default = "//rules/flags",
+                cfg = android_platforms_transition,
+            ),
             _bytecode_optimizer = attr.label(
                 default = configuration_field(
                     fragment = "java",

rules/android_binary/impl.bzl

@@ -45,6 +45,7 @@ load(
     "utils",
 )
 load("//rules:visibility.bzl", "PROJECT_VISIBILITY")
+load("//rules/flags:flags.bzl", _flags = "flags")
 load("@rules_java//java/common:java_common.bzl", "java_common")
 load("@rules_java//java/common:java_info.bzl", "JavaInfo")
 load("@rules_java//java/common:java_plugin_info.bzl", "JavaPluginInfo")
@@ -815,6 +816,9 @@ def _process_optimize(ctx, validation_ctx, deploy_ctx, packaged_resources_ctx, b
     if acls.in_baseline_profiles_optimizer_integration(str(ctx.label)) and bp_ctx.baseline_profile_output:
         startup_profile = bp_ctx.baseline_profile_output.startup_profile
         baseline_profile = bp_ctx.baseline_profile_output.baseline_profile
+    if not startup_profile and (_flags.get(ctx).use_baseline_as_startup_profile or
+                                acls.use_baseline_as_startup_profile(str(ctx.label))):
+        startup_profile = baseline_profile
 
     enable_rewrite_resources_through_optimizer = enable_resource_shrinking and ctx.attr._rewrite_resources_through_optimizer
 

rules/flags/flag_defs.bzl

@@ -93,3 +93,9 @@ def define_flags():
         default = True,
         description = "Use Studio Deployer to install apks",
     )
+
+    flags.DEFINE_bool(
+        name = "use_baseline_as_startup_profile",
+        default = False,
+        description = "For testing/validation only. Use baseline profiles as startup profiles in optimized builds.",
+    )