arrays - OpenMPI:广播数组的某些元素

Question

我需要我的进程拥有源数组的副本，然后让进程有选择地操作数组的某些元素，然后当一切完成后，我需要通过更改不同的元素来更新所有进程的源数组来自不同进程的数组。在操作开始之前，所有进程必须拥有相同的数组。下面的代码可以说明这个问题:

#include <mpi.h>
#include <stdio.h>
int main()
{
    int id, p, array[10];
    MPI_Init(NULL,NULL);
    MPI_Comm_size(MPI_COMM_WORLD, &p);
    MPI_Comm_rank(MPI_COMM_WORLD, &id);
    int k = id;
    while (k<10)
    {
        array[k]=k*(1+id);
        printf("%d %d %d \n",id,k,array[k]);
        MPI_Bcast(&array[k],1,MPI_INT,id,MPI_COMM_WORLD);
        k += p ;
    }
    MPI_Finalize();
    if ( id == 0 )
    {
        for (k=0;k<10;k++)
        {
            printf("%d\n", array[k]);
        }
    }
    return 0;
}

使用 np 2 运行，我希望输出类似于 0, 2, 2, 6, 4, 10, 6, 14, 8, 18。但是，虽然我得到了正确的 id = 0 数字，但id = 1 数字没有按预期输出，我认为这是两个进程的内存地址不相同的问题。我该如何解决这个问题？

Answer 1

您的问题至少有一部分是由此语句引起的

    MPI_Bcast(&array[k],1,MPI_INT,id,MPI_COMM_WORLD);

因为 id 在每个进程上都有不同的值。

广播的整个概念是一个进程向所有其他进程发送值。在对 MPI_Bcast 的调用中，第四个参数(此处的 id)标识执行发送的进程，并且该值在参与广播的每个进程上应该是相同的值。您可能应该将其更改为 0。或者可能是k。

将数组的初始状态广播到所有进程后，您就可以考虑在每个进程上操作它，然后将数据收集回根进程。我认为您太聪明了，试图通过多次调用 MPI_Bcast 在一个循环中完成所有这些工作。