def merge_quadtree_buffers(self, qt, merge_style):
# This is a modified version of pairwise reduction from Lisandro Dalcin,
# in the reductions demo of mpi4py
size = self.comm.size
rank = self.comm.rank
mask = 1
buf = qt.tobuffer()
print("PROC", rank, buf[0].shape, buf[1].shape, buf[2].shape)
sys.exit()
args = qt.get_args() # Will always be the same
tgd = np.array([args[0], args[1]], dtype="int64")
sizebuf = np.zeros(1, "int64")
while mask < size:
if (mask & rank) != 0:
target = (rank & ~mask) % size
# print("SENDING FROM %02i to %02i" % (rank, target))
buf = qt.tobuffer()
self.send_quadtree(target, buf, tgd, args)
# qt = self.recv_quadtree(target, tgd, args)
else:
target = rank | mask
if target < size:
# print("RECEIVING FROM %02i on %02i" % (target, rank))
buf = self.recv_quadtree(target, tgd, args)
qto = QuadTree(tgd, args[2], qt.bounds)
qto.frombuffer(buf[0], buf[1], buf[2], merge_style)
merge_quadtrees(qt, qto, style=merge_style)
del qto
# self.send_quadtree(target, qt, tgd, args)
mask <<= 1
if rank == 0:
buf = qt.tobuffer()
sizebuf[0] = buf[0].size
self.comm.Bcast([sizebuf, MPI.LONG], root=0)
if rank != 0:
buf = [
np.empty((sizebuf[0], ), "int32"),
np.empty((sizebuf[0], args[2]), "float64"),
np.empty((sizebuf[0], ), "float64"),
]
self.comm.Bcast([buf[0], MPI.INT], root=0)
self.comm.Bcast([buf[1], MPI.DOUBLE], root=0)
self.comm.Bcast([buf[2], MPI.DOUBLE], root=0)
self.refined = buf[0]
if rank != 0:
qt = QuadTree(tgd, args[2], qt.bounds)
qt.frombuffer(buf[0], buf[1], buf[2], merge_style)
return qt
def merge_quadtree_buffers(self, qt, merge_style):
# This is a modified version of pairwise reduction from Lisandro Dalcin,
# in the reductions demo of mpi4py
size = self.comm.size
rank = self.comm.rank
mask = 1
buf = qt.tobuffer()
print("PROC", rank, buf[0].shape, buf[1].shape, buf[2].shape)
sys.exit()
args = qt.get_args() # Will always be the same
tgd = np.array([args[0], args[1]], dtype='int64')
sizebuf = np.zeros(1, 'int64')
while mask < size:
if (mask & rank) != 0:
target = (rank & ~mask) % size
#print "SENDING FROM %02i to %02i" % (rank, target)
buf = qt.tobuffer()
self.send_quadtree(target, buf, tgd, args)
#qt = self.recv_quadtree(target, tgd, args)
else:
target = (rank | mask)
if target < size:
#print "RECEIVING FROM %02i on %02i" % (target, rank)
buf = self.recv_quadtree(target, tgd, args)
qto = QuadTree(tgd, args[2], qt.bounds)
qto.frombuffer(buf[0], buf[1], buf[2], merge_style)
merge_quadtrees(qt, qto, style = merge_style)
del qto
#self.send_quadtree(target, qt, tgd, args)
mask <<= 1
if rank == 0:
buf = qt.tobuffer()
sizebuf[0] = buf[0].size
self.comm.Bcast([sizebuf, MPI.LONG], root=0)
if rank != 0:
buf = [np.empty((sizebuf[0],), 'int32'),
np.empty((sizebuf[0], args[2]),'float64'),
np.empty((sizebuf[0],),'float64')]
self.comm.Bcast([buf[0], MPI.INT], root=0)
self.comm.Bcast([buf[1], MPI.DOUBLE], root=0)
self.comm.Bcast([buf[2], MPI.DOUBLE], root=0)
self.refined = buf[0]
if rank != 0:
qt = QuadTree(tgd, args[2], qt.bounds)
qt.frombuffer(buf[0], buf[1], buf[2], merge_style)
return qt
