mpi4py问题：通过send/recv向从节点发送消息遇阻

我看了你的MPI代码，现在遇到的问题核心是从节点没有接收消息的逻辑，同时也缺少处理终止信号的判断，导致主节点和从节点出现通信阻塞/空转的情况。咱们一步步来修复：

现存问题点 #

• 从节点（rank == 1）的while True循环里只有打印语句，完全没有调用comm.recv()来接收主节点发送的消息，这会导致主节点的阻塞式send一直等待从节点接收，而从节点永远在空转打印，形成死锁。
• 从节点没有判断终止信号（tag=1）的逻辑，就算收到消息也无法退出循环。

修正后的代码 #

from mpi4py import MPI
comm = MPI.COMM_WORLD
rank = comm.Get_rank()
status = MPI.Status()

if rank == 0:
    data = ['string1', 'string2', 'string3']
    for k in data:
        comm.send(k, dest=1, tag=11)
    # 发送终止信号，这里可以用空字符串+tag=1，也可以用其他标识
    comm.send('', dest=1, tag=1)
elif rank == 1:
    while True:
        # 接收任意tag的消息，同时通过status获取消息的tag
        msg = comm.recv(source=0, status=status)
        current_tag = status.Get_tag()
        
        if current_tag == 1:
            # 收到终止信号，退出循环
            print("Received termination signal, exiting...")
            break
        elif current_tag == 11:
            # 处理正常消息
            print(f"Received message: {msg}")

关键改进说明 #

1. 添加消息接收逻辑：在从节点的循环里加入comm.recv()，指定从主节点（source=0）接收消息，并通过status对象获取消息的tag，这样能区分正常数据和终止信号。
2. 终止信号判断：当收到tag=1的消息时，直接break退出循环，结束从节点的运行。
3. 避免死锁：主节点的send和从节点的recv一一对应，所有发送的消息都会被接收，不会出现主节点一直等待的情况。

Slurm集群运行注意事项 #

在Slurm上运行时，要确保申请的进程数和代码里的rank数量匹配，比如用以下命令提交（假设脚本名为mpi_test.py）：

srun -n 2 python mpi_test.py

这里-n 2表示申请2个进程，对应主节点（rank0）和从节点（rank1）。

内容的提问来源于stack exchange，提问作者Farzad Vertigo