C语言MPI程序进程数>2时冻结问题求助（单发送多监听场景）

问题根源 #

你当前的代码逻辑只完成了一次固定方向的发送：rank 0硬编码发给rank 1，当进程数N>2时，rank 2及以上的进程会卡在MPI_Recv调用上——没有任何进程会给它们发送数据，这些进程会一直阻塞等待，最终导致整个程序冻结。

而当N=2时，rank 0发送后，rank 1接收，所有进程都完成了当前代码块的操作，所以程序能正常结束。

修复思路 #

要实现「重复10次随机发送令牌」的需求，核心逻辑应该是：

• 初始由rank 0发起第一次发送
• 每个收到令牌的进程，随机选择一个非自身的目标进程发送令牌
• 循环执行10次这样的发送操作

修正后的代码示例 #

#include <stdio.h>
#include <stdlib.h>
#include <mpi.h>
#include <time.h>

int main(int argc, char** argv) {
    MPI_Init(&argc, &argv);
    
    int my_rank, world_size;
    MPI_Comm_rank(MPI_COMM_WORLD, &my_rank);
    MPI_Comm_size(MPI_COMM_WORLD, &world_size);

    // 处理单进程特殊情况
    if (world_size == 1) {
        printf("单进程无需传递令牌\n");
        MPI_Finalize();
        return 0;
    }

    int token;
    // 每个进程用不同种子生成随机数，避免重复序列
    srand(time(NULL) + my_rank);

    if (my_rank == 0) {
        // 首次发送：随机选非0的目标进程
        int dest = rand() % (world_size - 1) + 1;
        token = 1; // 用token标记轮次
        MPI_Send(&token, 1, MPI_INT, dest, 0, MPI_COMM_WORLD);
        printf("进程 %d 发送令牌（轮次%d）到进程 %d\n", my_rank, token, dest);
    }

    // 前9轮：收令牌后再转发
    for (int round = 1; round < 10; round++) {
        MPI_Recv(&token, 1, MPI_INT, MPI_ANY_SOURCE, 0, MPI_COMM_WORLD, MPI_STATUS_IGNORE);
        printf("进程 %d 收到令牌（轮次%d）\n", my_rank, token);

        // 随机选非自身的目标
        int dest;
        do {
            dest = rand() % world_size;
        } while (dest == my_rank);

        token = round + 1;
        MPI_Send(&token, 1, MPI_INT, dest, 0, MPI_COMM_WORLD);
        printf("进程 %d 发送令牌（轮次%d）到进程 %d\n", my_rank, token, dest);
    }

    // 第10轮：最后一个接收方无需转发
    if (my_rank != 0) {
        MPI_Recv(&token, 1, MPI_INT, MPI_ANY_SOURCE, 0, MPI_COMM_WORLD, MPI_STATUS_IGNORE);
        printf("进程 %d 收到令牌（轮次%d），结束传递\n", my_rank, token);
    }

    MPI_Finalize();
    return 0;
}

关键细节说明 #

• 随机目标生成：用do-while循环确保不会把令牌发给自己
• 轮次控制：通过令牌值标记当前轮次，确保总共执行10次发送操作
• 随机种子：每个进程用time(NULL)+my_rank作为种子，避免不同进程生成相同的随机序列
• 阻塞通信正确性：每一次MPI_Send都对应一个MPI_Recv，不会出现无匹配的阻塞

内容的提问来源于stack exchange，提问作者Guille Luengo