MPI_Barrier和递归

时间:2011-06-30 16:48:06

标签: c++ recursion mpi

我正在尝试使用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 

1 个答案:

答案 0 :(得分:3)

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

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