pingpong1

from mpi4py import MPI

import sys

PING_PONG_LIMIT = 10 #limit=10被触发的时候,进程就停止了发送和接收。

world_rank = MPI.COMM_WORLD.Get_rank() #Get_rank（）函数为进程编号

world_size = MPI.COMM_WORLD.Get_size() #Get_size()函数，0号进程指导的进程数

print(world_rank)

print(world_size)

if world_size != 2: #进程数不等于2时

print("World size must be two for %s\n")

sys.exit(1)#sys.exit(int n),n为0正常退出，非0为非正常退出当前程序

ping_pong_count = 0 #每次发送消息之后count会递增1

partner_rank = (world_rank + 1) % world_size #接收方当进程编号

while ping_pong_count < PING_PONG_LIMIT: #当发送次数小于10时

if world_rank == ping_pong_count % 2:

ping_pong_count += 1

MPI.COMM_WORLD.send(ping_pong_count,dest=partner_rank,tag=0) #参数分别为 传送数据个数，接收方进程，标签：接受方需要标签匹配才接受

print(str(world_rank) + " send and incremented pin_pong_count " + str(ping_pong_count) + " to " + str(partner_rank))

else: ping_pong_count = MPI.COMM_WORLD.recv(source=partner_rank,tag=0) #发送方进程

print(str(world_rank) + " received pin_pong_count " + str(ping_pong_count) + " from " + str(partner_rank))

MPI.Finalize()#结束MPI执行环境