MPI_Barrier 和递归

Question

我正在尝试使用 MPI_Barrier (OpenMPI) 来强制所有进程处于递归调用的相同深度。

这是代码

#include <iostream>
#include <fstream>
#include <stdio.h>
#include <mpi.h>

using namespace std;

void recursive_function(int,int,int);

int main(int argc, char *argv[]) {
    int rank, size;

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

    recursive_function(0,3,rank);

    MPI_Finalize();
}

void recursive_function(int depth, int limit, int rank) {
    printf("depth: %d, processor %d\n", depth, rank);

    MPI_Barrier(MPI_COMM_WORLD);
    if(depth == limit) return;
    else recursive_function(depth+1,limit,rank);
}

我得到的是（使用 mpirun -np 2 屏障运行）

depth: 0, processor 0
depth: 1, processor 0
depth: 2, processor 0
depth: 3, processor 0
depth: 0, processor 1
depth: 1, processor 1
depth: 2, processor 1
depth: 3, processor 1  

虽然我希望像

depth: 0, processor 0
depth: 0, processor 1
depth: 1, processor 0
depth: 1, processor 1
depth: 2, processor 1
depth: 2, processor 0
depth: 3, processor 1
depth: 3, processor 0 

Answer 1

不保证 MPI 进程中的 stdout 以任何方式排序。

也就是说，您需要让进程进行通信以证明它们都处于相同的递归深度。例如。在屏障之后，每个进程 != 0 都会向排名 0 发送一条消息，该消息会打印一些内容。