commit cf5b51c94871c08719927f865ffc3f89f5d0042e
author Christoffer Adamsen <christofferqa@google.com> Tue Mar 26 07:31:44 2024 +0000
committer Christoffer Adamsen <christofferqa@google.com> Tue Mar 26 07:33:46 2024 +0000
tree 160a69a76cf7e1fcd0cc7737341a10e356935a82
parent 4ea6cb47173d65d1eee954dfd58f9048165cd4bd

Process Composable waves in parallel

Bug: b/b/302483644
Change-Id: Idf0d5d801d0731e6c0d58eaec825480f7d90a917

src/main/java/com/android/tools/r8/optimize/compose/ComposeMethodProcessor.java

diff --git a/src/main/java/com/android/tools/r8/optimize/compose/ComposeMethodProcessor.java b/src/main/java/com/android/tools/r8/optimize/compose/ComposeMethodProcessor.java
index 1aca973..ec12fa0 100644
--- a/src/main/java/com/android/tools/r8/optimize/compose/ComposeMethodProcessor.java
+++ b/src/main/java/com/android/tools/r8/optimize/compose/ComposeMethodProcessor.java
@@ -37,6 +37,7 @@
 import com.android.tools.r8.shaking.AppInfoWithLiveness;
 import com.android.tools.r8.utils.IterableUtils;
 import com.android.tools.r8.utils.LazyBox;
+import com.android.tools.r8.utils.ThreadUtils;
 import com.android.tools.r8.utils.Timing;
 import com.android.tools.r8.utils.collections.ProgramMethodSet;
 import com.google.common.collect.Iterables;
@@ -63,12 +64,12 @@
     this.converter = converter;
   }
 
-  // TODO(b/302483644): Process wave concurrently.
   public Set<ComposableCallGraphNode> processWave(
       Set<ComposableCallGraphNode> wave, ExecutorService executorService)
       throws ExecutionException {
     ProcessorContext processorContext = appView.createProcessorContext();
-    wave.forEach(
+    ThreadUtils.processItems(
+        wave,
         node -> {
           assert !processed.contains(node);
           converter.processDesugaredMethod(
@@ -77,7 +78,9 @@
               this,
               processorContext.createMethodProcessingContext(node.getMethod()),
               MethodConversionOptions.forLirPhase(appView));
-        });
+        },
+        appView.options().getThreadingModule(),
+        executorService);
     processed.addAll(wave);
     return optimizeComposableFunctionsCalledFromWave(wave, executorService);
   }