如何将mpi各进程结果写入文件

在MPI程序中，可以使用MPI_File_write_shared函数将各个进程的结果写入同一个文件中。这个函数可以让多个进程同时写入同一个文件，避免了多个进程互相干扰的问题。

下面是一个示例代码，假设每个进程都有一个结果数组data，需要将这些结果写入到同一个文件中。

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

int main(int argc, char** argv) {
    MPI_Init(&argc, &argv);

    int rank, size;
    MPI_Comm_rank(MPI_COMM_WORLD, &rank);
    MPI_Comm_size(MPI_COMM_WORLD, &size);

    // 假设每个进程都有一个结果数组data
    int data[10];
    for (int i = 0; i < 10; i++) {
        data[i] = rank * 10 + i;
    }

    MPI_File fh;
    MPI_File_open(MPI_COMM_WORLD, "output.txt", MPI_MODE_CREATE|MPI_MODE_WRONLY, MPI_INFO_NULL, &fh);

    // 计算每个进程的偏移量和写入的长度
    MPI_Offset offset = rank * 10 * sizeof(int);
    MPI_Offset length = 10 * sizeof(int);

    // 将数据写入文件
    MPI_File_write_shared(fh, data, 10, MPI_INT, MPI_STATUS_IGNORE);

    MPI_File_close(&fh);

    MPI_Finalize();
    return 0;
}

这个程序中，每个进程都会将自己的结果数组写入文件，使用MPI_File_write_shared函数实现。MPI_File_open函数用于打开文件，MPI_Offset用于计算每个进程的偏移量和写入的长度。最后，使用MPI_File_close函数关闭文件。