Block fusion for barrier tasks

dask · Nov 20, 2024 · 1b14f4a · 1b14f4a
1 parent 750cb91
commit 1b14f4a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/distributed/shuffle/_core.py b/distributed/shuffle/_core.py
@@ -584,6 +584,7 @@ def p2p_barrier(id: ShuffleId, *run_ids: int) -> int:
 
 class P2PBarrierTask(Task):
     spec: ShuffleSpec
+    block_fusion: bool
 
     __slots__ = tuple(__annotations__)
 
@@ -598,6 +599,7 @@ def __init__(
     ):
         self.spec = spec
         super().__init__(key, func, *args, **kwargs)
+        self.block_fusion = True
 
     def __repr__(self) -> str:
         return f"P2PBarrierTask({self.key!r})"