c - 数组接收的 MPI 总和仅适用于一个级别

Question

我正在尝试使用 MPI 查找长度为 100 个元素的数组的总和，在仅使用 MPI_Send 和 MPI_receive 的限制下，我编写的代码找到了每个处理器的总和，但是在重新发送到主处理器(rank=0)我的代码只从一个处理器接收

我的代码

#include "stdafx.h"
#include <stdio.h>
#include <string.h>
#include "mpi.h"
#include "math.h"

int val = 1;
int main(int argc, char* argv[]) {

    int my_rank;
    int p;
    int ierr;
    int i;
    int a[100];
    int q=0;
    for (i = 0; i <100; i++)
    {
        a[i] = i+1;
    }
    int send,recv;
    MPI_Status status;
    MPI_Init(&argc, &argv);
    MPI_Comm_rank(MPI_COMM_WORLD, &my_rank);
    MPI_Comm_size(MPI_COMM_WORLD, &p);
    int part = 100 /(p-1);
    if (my_rank == 0)
    {
        for (i = 1; i < p; i++)
        {
            send = part * (i-1);
            MPI_Send(&send, 1, MPI_INT, i, 0, MPI_COMM_WORLD);
        }

    }
    else
    {
        MPI_Recv(&recv, 1, MPI_INT, 0, 0, MPI_COMM_WORLD, &status);
        for (i = recv; i < recv + part; i++)
        {
            val = val+a[i];

        }
        printf("%d\n", val);
        MPI_Send(&val, 1, MPI_INT, 0, 0, MPI_COMM_WORLD);
    }
    if (my_rank == 0)
    {
        MPI_Recv(&val, 1, MPI_INT, MPI_ANY_SOURCE, 0, MPI_COMM_WORLD, &status);
        printf("%d", val);
        q = q + val;

    }
    MPI_Finalize();
    if (my_rank == 0)
    {
        printf("The output is %d\n", q);
    }
    return 0;
}

我的输出

我哪里错了

Answer 1

因为您只收到一个过程的结果。要接收所有结果，迭代进程等级:

if (my_rank == 0)
{
    for (rank = 1; rank < proc_cnt; rank++)
    {
        MPI_Recv(&val, 1, MPI_INT, rank, 0, MPI_COMM_WORLD, &status);
        printf("value of rank %d is %d", rank, val);
        q = q + val;
    }

}

通常，这是一种不好的做法，可能会导致死锁。如果允许，请使用 mpi_gather()。