MPI归约操作实例代码

好的，以下是一个简单的MPI求和归约操作的示例代码：

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

int main(int argc, char *argv[])
{
    int rank, size, sum = 0, local_sum = 0;

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

    // 每个进程计算本地和
    for (int i = 1; i <= 100; i++) {
        local_sum += i + rank;
    }

    // 归约操作，将每个进程的 local_sum 求和
    MPI_Reduce(&local_sum, &sum, 1, MPI_INT, MPI_SUM, 0, MPI_COMM_WORLD);

    // 只有进程0打印最终结果
    if (rank == 0) {
        printf("Total sum is %d\n", sum);
    }

    MPI_Finalize();
    return 0;
}

在这个例子中，每个进程计算1到100的累加和，然后将每个进程的和进行归约操作，最终得到所有进程的和。在MPI_Reduce函数中，第一个参数是本地缓冲区的地址（即每个进程的和），第二个参数是目标缓冲区的地址（即所有进程的和），第三个参数是元素个数（这里是1），第四个参数是操作类型（这里是求和，即MPI_SUM），第五个参数是目标进程的排名（这里是0，表示归约操作的结果要发送到进程0），第六个参数是通信域（这里是MPI_COMM_WORLD）。在这个例子中，只有进程0会打印最终的结果。