Skip to content

Commit c5ee33d

Browse files
committed
force ntasks tot to be an int, ceil to be conservative
1 parent e642c7b commit c5ee33d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pyuvsim/uvsim.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -725,7 +725,7 @@ def run_uvdata_uvsim(
725725
Ntasks_tot = Ntasks_local * Nsky_parts
726726
# Sum all the tasks across each node
727727
Nsky_parts = comm.allreduce(Nsky_parts, op=mpi.MPI.MAX)
728-
Ntasks_tot = comm.allreduce(Ntasks_tot, op=mpi.MPI.SUM)
728+
Ntasks_tot = int(np.ceil(comm.allreduce(Ntasks_tot, op=mpi.MPI.SUM)))
729729
if rank == 0 and not quiet:
730730
if Nsky_parts > 1:
731731
print(

0 commit comments

Comments
 (0)