Parallel-Computing/p2p_numpy.py at master · raykodali/Parallel-Computing · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
from mpi4py import MPI

import numpy as np

comm = MPI.COMM_WORLD

assert comm.size == 2

rank = comm.rank

if rank == 0:
    data = np.arange(10, dtype=np.float64)
    comm.Send(data, dest=1, tag=13)
elif rank == 1:
    data = np.empty(10, dtype=np.float64)
    comm.Recv(data, source=0, tag=13)
    print("[%02d] Received: %s" % (rank, data))