1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36
|
# mpiexec -n 1 python ex-2.35.py
# Calls to attach and detach buffers
# --------------------------------------------------------------------
from mpi4py import MPI
try:
from numpy import empty
except ImportError:
from array import array
def empty(size, dtype):
return array(dtype, [0] * size)
# --------------------------------------------------------------------
BUFSIZE = 10000 + MPI.BSEND_OVERHEAD
buff = empty(BUFSIZE, dtype="b")
MPI.Attach_buffer(buff)
buff2 = MPI.Detach_buffer()
MPI.Attach_buffer(buff2)
MPI.Detach_buffer()
# --------------------------------------------------------------------
assert len(buff2) == BUFSIZE
# --------------------------------------------------------------------
|