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
|
## 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)
# --------------------------------------------------------------------
BUFSISE = 10000 + MPI.BSEND_OVERHEAD
buff = empty(BUFSISE, dtype='b')
MPI.Attach_buffer(buff)
buff2 = MPI.Detach_buffer()
MPI.Attach_buffer(buff2)
MPI.Detach_buffer()
# --------------------------------------------------------------------
assert len(buff2) == BUFSISE
# --------------------------------------------------------------------
|