Skip to content

Commit e4447c3

Browse files
committed
Fix name of the (all) reduction operation
1 parent e9de22e commit e4447c3

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

yt/utilities/parallel_tools/parallel_analysis_interface.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -598,7 +598,7 @@ def parallel_objects(
598598
case "cat_on_root":
599599
new_storage = my_communicator.concat(to_share, root=0)
600600
case "sum" | "max" | "min":
601-
new_storage = my_communicator.reduce(to_share, op=reduction, root=0)
601+
new_storage = my_communicator.all_reduce(to_share, op=reduction)
602602
case _:
603603
raise NotImplementedError(
604604
f"Reduction operation {reduction} not implemented."
@@ -947,7 +947,7 @@ def mpi_allreduce(self, data, dtype=None, op="sum"):
947947
# relatively small ( < 1e7 elements )
948948
return self.comm.allreduce(data, op)
949949

950-
def reduce(self, data, op="sum", root=0):
950+
def all_reduce(self, data, op="sum"):
951951
reduction_op = {
952952
"sum": np.sum,
953953
"max": np.max,

0 commit comments

Comments
 (0)