Benchmarks: fix OOM crash on a real iOS device. (#5209)

## Release Notes N/A
JetBrains · Jan 22, 2025 · b463a44 · b463a44
1 parent 1e1810c
commit b463a44
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/benchmarks/multiplatform/benchmarks/src/commonMain/kotlin/MeasureComposable.kt b/benchmarks/multiplatform/benchmarks/src/commonMain/kotlin/MeasureComposable.kt
@@ -57,21 +57,24 @@ suspend fun measureComposable(
         repeat(warmupCount) {
             scene.render(canvas, it * nanosPerFrame)
             surface.flushAndSubmit(false)
+            graphicsContext?.awaitGPUCompletion()
         }
 
-        graphicsContext?.awaitGPUCompletion()
-
         runGC()
 
         var renderTime = Duration.ZERO
+        var gpuTime = Duration.ZERO
         if (Args.isModeEnabled(Mode.CPU)) {
             renderTime = measureTime {
                 repeat(frameCount) {
                     scene.render(canvas, it * nanosPerFrame)
                     surface.flushAndSubmit(false)
+                    gpuTime += measureTime {
+                        graphicsContext?.awaitGPUCompletion()
+                    }
                 }
             }
-            graphicsContext?.awaitGPUCompletion()
+            renderTime -= gpuTime
         }
 
         val frames = MutableList(frameCount) {