python_parallel_computing/pointToPointCommunication.py at master · qiweihan/python_parallel_computing · GitHub

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
from mpi4py import MPI

comm = MPI.COMM_WORLD
rank = comm.rank
print("my rank is : ", rank)

if rank == 0:
    data = 10000000
    destination_process = 4
    comm.send(data, dest=destination_process)
    print("sending data %s to process %d" % (data, destination_process))

if rank == 1:
    destination_process = 8
    data = "hello"
    comm.send(data, dest=destination_process)
    print("sending data %s to process %d" % (data, destination_process))

if rank == 4:
    data = comm.recv(source=0)
    print("data received is = %s" % data)

if rank == 8:
    data1 = comm.recv(source=1)
    print("data received is = %s" % data1)