(beam) branch master updated: [ReduceFnRunner] Fix Prefetches (#39082)

[sc...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

scwhittle pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/beam.git


The following commit(s) were added to refs/heads/master by this push:
     new cedfd0bb3e8 [ReduceFnRunner] Fix Prefetches (#39082)
cedfd0bb3e8 is described below

commit cedfd0bb3e86493442a4f3f6ea0f70a76c701d10
Author: Arun Pandian <pa...@google.com>
AuthorDate: Wed Jun 24 02:20:28 2026 -0700

    [ReduceFnRunner] Fix Prefetches (#39082)
    
    1. Add a prefetch for CombinedMetadata
    2. Prefetch ReduceFn using renamedContext instead of directContext. This improves prefetching on merging windows
---
 .../src/main/java/org/apache/beam/runners/core/ReduceFnRunner.java     | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/runners/core-java/src/main/java/org/apache/beam/runners/core/ReduceFnRunner.java b/runners/core-java/src/main/java/org/apache/beam/runners/core/ReduceFnRunner.java
index 78505f3c65f..7fe3b711aa0 100644
--- a/runners/core-java/src/main/java/org/apache/beam/runners/core/ReduceFnRunner.java
+++ b/runners/core-java/src/main/java/org/apache/beam/runners/core/ReduceFnRunner.java
@@ -1056,7 +1056,8 @@ public class ReduceFnRunner<K, InputT, OutputT, W extends BoundedWindow> {
     paneInfoTracker.prefetchPaneInfo(directContext);
     watermarkHold.prefetchExtract(renamedContext);
     nonEmptyPanes.isEmpty(renamedContext.state()).readLater();
-    reduceFn.prefetchOnTrigger(directContext.state());
+    directContext.state().access(METADATA_TAG).readLater();
+    reduceFn.prefetchOnTrigger(renamedContext.state());
   }
 
   /**