Flush scheduling window before tree reduction is performed (nv-legate…

…#203)
ipdemes · Apr 19, 2022 · ae5140d · ae5140d
1 parent d9b38a7
commit ae5140d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/legate/core/context.py b/legate/core/context.py
@@ -236,6 +236,10 @@ def tree_reduce(
         result = self.create_store(store.type)
         unique_op_id = self.get_unique_op_id()
 
+        # Make sure we flush the scheduling window, as we will bypass
+        # the partitioner below
+        self.runtime.flush_scheduling_window()
+
         # A single Reduce operation is mapepd to a whole reduction tree
         task = Reduce(self, task_id, radix, mapper_id, unique_op_id)
         task.add_input(store)