如何使用MPI实现接力棒传递同步以保证进程按序输出

问题解答 #

1. 是否需要循环执行接力棒逻辑？ #

不需要。你构想的「接收接力棒→打印→发送接力棒」的基础逻辑是正确的，该场景下每个进程仅需要完成1次接收、打印、发送的流程即可实现按秩升序打印的需求，不需要额外套循环。你提供的现有代码里的for循环是无效的同步逻辑，完全起不到控制顺序的作用，仅做了无用的空循环判断。

2. 实现接力棒同步的核心逻辑 #

因为0号进程是起始节点，所以需要单独处理：

• 0号进程不需要等待接收，初始化接力棒后直接执行打印，再把接力棒发送给1号进程
• 非0号且非最后1号的进程：先从rank-1进程接收接力棒，打印完成后将接力棒发送给rank+1进程
• 最后1号（rank = size-1）进程：仅需要从rank-1接收接力棒，完成打印即可，不需要向外发送

3. 修改后可运行的代码 #

#include <stdio.h>
#include "mpi.h"

int main( int argc, char *argv[] )
{
    int rank;
    int size;
    int baton; // 接力棒变量

    MPI_Init(&argc,&argv);
    MPI_Comm_rank(MPI_COMM_WORLD, &rank);
    MPI_Comm_size(MPI_COMM_WORLD, &size);

    if (rank == 0) {
        // 0号进程初始接力棒为1
        baton = 1;
        printf("Hello world from process %d of %d\n", rank, size);
        // 发送给下一个进程
        MPI_Send(&baton, 1, MPI_INT, rank+1, 0, MPI_COMM_WORLD);
    } else {
        // 其余进程先阻塞接收前一个进程的接力棒
        MPI_Recv(&baton, 1, MPI_INT, rank-1, 0, MPI_COMM_WORLD, MPI_STATUS_IGNORE);
        // 收到后执行打印
        printf("Hello world from process %d of %d\n", rank, size);
        // 非最后一个进程，转发接力棒给下一个
        if (rank != size - 1) {
            MPI_Send(&baton, 1, MPI_INT, rank+1, 0, MPI_COMM_WORLD);
        }
    }

    MPI_Finalize();
    return 0;
}

4. 实现原理说明 #

这里利用了MPI阻塞收发函数的同步特性：MPI_Recv会一直阻塞调用进程，直到收到对应来源的消息，自然保证了只有前一个进程完成打印、发出接力棒之后，后一个进程才会结束阻塞、执行打印操作，完全实现按秩升序输出的需求。接力棒仅用作同步触发信号，不需要额外修改取值。

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