线性搜索 MPI（在其他进程中停止）

Question

我正在尝试编写一个简单的多进程程序来查找数组中的值。

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

int* create_array(int num_items) {
    int* tmp = new int[num_items];
    for(int i = 0; i < num_items; i++)
        tmp[i] = i;

    return tmp;
}

int main() {

    int num_items = 1000;
    int item = 999;

    MPI_Init(NULL, NULL);
    int world_rank, world_size, num_items_per_proc;
    MPI_Comm_rank(MPI_COMM_WORLD, &world_rank);
    MPI_Comm_size(MPI_COMM_WORLD, &world_size);

    MPI_Request* inReq;

    int* array;
    if(world_rank == 0) {
        array = create_array(num_items);
        num_items_per_proc = (num_items / world_size) + 1;
    }

    int* sub_array = new int[num_items_per_proc];
    MPI_Scatter(array, num_items_per_proc, MPI_INT, sub_array,
                num_items_per_proc, MPI_INT, 0, MPI_COMM_WORLD);

    bool found = false;
    MPI_Irecv(&found, 1, MPI::BOOL, MPI_ANY_SOURCE, MPI_ANY_TAG,
              MPI_COMM_WORLD, inReq);

    for(int i = 0; i < num_items_per_proc && !found; i++) {
        if (sub_array[i] == item) {
            found = true;
            printf("Elemento %d trovato in posizione: %d\n", item, i);
            for(int j = 0; j < world_size; j++)
                if(j != world_rank)
                    MPI_Send(&found, 1, MPI::BOOL, j, j, MPI_COMM_WORLD);
        }
    }

    if(world_rank == 0) delete[] array;
    delete[] sub_array;

    MPI_Barrier(MPI_COMM_WORLD);
    MPI_Finalize();

    return 0;
}

当他们中的一个人在数组的一部分中找到值时，我试图停止所有操作，但我收到了 Irecv 的分段错误。我该如何解决这个问题？

Answer 1

您的代码不起作用的原因是您必须向MPI_Irecv 提供一个实际的MPI_Request - 而不仅仅是一个未初始化的 指针！

MPI_Request inReq;
MPI_Irecv(&found, 1, MPI_CXX_BOOL, MPI_ANY_SOURCE, MPI_ANY_TAG,
          MPI_COMM_WORLD, &inReq);

您处理found 的方式是错误的。您不能修改给定异步请求的变量，也不能假设它是在后台更新的。非阻塞消息不是单方面的远程内存操作。因此，您必须调用测试，如果状态指示收到消息，您可以中止循环。确保每个请求都已完成，并且在找到结果的排名中也是如此。

此外，num_items_per_proc 必须在所有等级上都有效（用于分配内存，以及用于在MPI_Scatter 中指定recvcount。

MPI_Finalize 之前的屏障是多余的，最后，MPI 的 C++ 绑定被移除，使用MPI_CXX_BOOL 而不是MPI::BOOL。

您可以在answers of this question 中找到解决您问题的更复杂的方法。