Internal change

PiperOrigin-RevId: 925663237
Change-Id: Iafb2b289520e72f551241cbf25e7eb5572499324

rules/android_application/android_application_rule.bzl

diff --git a/rules/android_application/android_application_rule.bzl b/rules/android_application/android_application_rule.bzl
index 809525e..a450426 100644
--- a/rules/android_application/android_application_rule.bzl
+++ b/rules/android_application/android_application_rule.bzl
@@ -19,6 +19,7 @@
     "AndroidBundleInfo",
     "AndroidFeatureModuleInfo",
     "AndroidIdeInfo",
+    "AndroidOptimizationInfo",
     "AndroidPreDexJarInfo",
     "AndroidSandboxedSdkBundleInfo",
     "ApkInfo",
@@ -374,6 +375,11 @@
         metadata["com.android.tools.build.profiles/baseline.prof"] = base_art_profile_info.baseline_profile
         metadata["com.android.tools.build.profiles/baseline.profm"] = base_art_profile_info.baseline_profile_metadata
 
+    if AndroidOptimizationInfo in ctx.attr.base_module:
+        opt_info = ctx.attr.base_module[AndroidOptimizationInfo]
+        if opt_info.d8_optimization_info:
+            metadata["com.android.tools/d8.json"] = opt_info.d8_optimization_info
+
     # Create .aab
     _bundletool.build(
         ctx,