Beispiel #1
0
def pprint(message=None, comm=None, root=0):
    if comm is None:
        comm = COMM_WORLD
    inbuf = comm.SERIALIZER.dump(message)
    result = _mpi.gather_string(inbuf, None, root, comm)
    if comm.Get_rank() == root:
        for i in result:
            msg = comm.SERIALIZER.load(i)
            print msg
Beispiel #2
0
def pprint(message = None, comm = None, root = 0):
    if comm is None:
        comm = COMM_WORLD
    inbuf = comm.SERIALIZER.dump(message)
    result = _mpi.gather_string(inbuf, None, root, comm)
    if comm.Get_rank() == root:
        for i in result:
            msg = comm.SERIALIZER.load(i)
            print msg
Beispiel #3
0
 def Gather(self, sendbuf, recvbuf=None, root=0):
     sendbuf, sfastmode = _mpi.make_buf(sendbuf)
     recvbuf, rfastmode = _mpi.make_buf(recvbuf)
     if sfastmode or rfastmode:
         _mpi.gather(sendbuf, recvbuf, root, self)
         return
     else:
         serializer = self.SERIALIZER
         if _mpi.comm_test_inter(self):
             if root != _mpi.ROOT and root != _mpi.PROC_NULL:
                 sendbuf = serializer.dump(sendbuf)
             else:
                 sendbuf = None
         else:
             sendbuf = serializer.dump(sendbuf)
         recvbuf = _mpi.gather_string(sendbuf, recvbuf, root, self)
         if recvbuf is not None:
             recvbuf = map(serializer.load, recvbuf)
         return recvbuf
Beispiel #4
0
 def Gather(self, sendbuf, recvbuf = None, root = 0):
     sendbuf, sfastmode = _mpi.make_buf(sendbuf)
     recvbuf, rfastmode = _mpi.make_buf(recvbuf)
     if sfastmode or rfastmode:
         _mpi.gather(sendbuf, recvbuf, root, self)
         return
     else:
         serializer = self.SERIALIZER
         if _mpi.comm_test_inter(self):
             if root != _mpi.ROOT and root != _mpi.PROC_NULL:
                 sendbuf = serializer.dump(sendbuf)
             else:
                 sendbuf = None
         else:
             sendbuf = serializer.dump(sendbuf)
         recvbuf = _mpi.gather_string(sendbuf, recvbuf, root, self)
         if recvbuf is not None:
             recvbuf = map(serializer.load, recvbuf)
         return recvbuf