Question

我必须使用MPI从3D矩阵计算浓度值。

我将矩阵划分为多个切片，具体取决于将使用的proc数量。

我得到了更小的3D矩阵，它们可以独立工作，但我希望将它们计算出的浓度值写入proc 0中，例如将0到n-1行写入proc 0，将proc写入n到2n-1 1，..

我知道我可以在mpi_file_write_at函数中使用偏移量来执行此类操作，但是我不知道如何实现它。

我知道一个简单的解决方案是要求proc 0打开文件，编写并关闭它，然后proc 1打开。

因为代码本身还没有完成，所以不要介意

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

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

    double h, m, L, T_max, v_x, v_y, v_z, D, r_thresh;
    unsigned int S;

    FILE *file = fopen(name, "rb");


    if(!file)
    {
        fprintf(stderr, "Not able to open the file %s\n", name);
    }
    else
        printf("Able to open the file %s\n", name);

    fscanf(file, "%lf", &h);
    fscanf(file, "%lf", &m);
    fscanf(file, "%lf", &L);
    fscanf(file, "%lf", &T_max);
    fscanf(file, "%lf", &v_x);
    fscanf(file, "%lf", &v_y);
    fscanf(file, "%lf", &v_z);
    fscanf(file, "%lf", &D);
    fscanf(file, "%u", &S);
    fscanf(file, "%lf", &r_thresh);

    fclose(file);

    if(((L/h) % 2) != 0){
        printf("Uncorrect parameters L and h ! \n");
        return -1;
    }
    else{
        unsigned int edge = (L/h);
        unsigned int middle = edge/2;
        unsigned int N = edge+1;
    }
    if(((T_max/m) % 1) != 0){
        printf("Uncorrect parameters T_max and m ! \n");
        return -1;
    }
    else{
        unsigned int t_end = T_max/m;
    }

    unsigned int myrank, nbproc;
    MPI_Init(&argc, &argv) ;
    MPI_Comm_rank( MPI_COMM_WORLD , &myrank );
    MPI_Comm_size( MPI_COMM_WORLD , &nbproc );

    double w = N/nbproc;
    unsigned int width = round(w);
    unsigned int k_min, k_max;
    k_min = myrank*width;
    k_max = (myrank+1)*width-1;
    width = width+2; 
    if(nbproc==1){
        width = N;
    }
    else if(myrank==(nbproc-1)){
        k_max = N-1; 
        width = k_max-k_min+1; 
        width = width+1; 
    }
    else if(myrank==0){
        width = width-1; 
    }



    double ***C ;
    C = malloc(width * sizeof(*C)); 
    for(k=0 ; k < width ; k++){
        C[k] = malloc(N * sizeof(**C));
    }
    for(k=0 ; k < width ; k++){
        for(j=0 ; j < N ; j++){
            C[k][j] = malloc(N * sizeof(***C));
        }
    }


    unsigned int i,j,k,n;
    for(k=0 ; k<width ; k++){
        for(j=0 ; j<N ; j++){
            for(i=0 ; i<N ; i++){
                C[k][j][i]=0;
            }
        }
    }
    if(middle>=k_min && middle=<k_max){
        C[middle-k_min][middle][middle] = 1;
    }


    if(myrank==0 || nbproc==1){
        k_min = k_min +1 ;
    }
    if(myrank==(nbproc-1) || nbproc==1){
        k_max = k_max -1 ;
    }





    for(n=1 ; n<T_max ; n++){

        char *c = "c_";
        char buffer[10];
        itoa(n,buffer,2);
        strcat(c,buffer);

        // I want to write some value stored in a buffer now
        // computed by different process in the same file 
        // MPI_File_write_at ???


    }

    for(k=0 ; k < width ; k++){
        for(j=0 ; j < N ; j++){
            free(C[k][j]);
            }
        free(C[k]);
    }
    free(C);

    return 0;
}

Answer 1

MPI程序中的每个进程都有一个本地地址空间。文件I / O更改为：文件是全局可访问的线性字节流。您需要考虑每个进程在全局空间中写的位置。几种方法可以做到这一点：

有时，您只需将每个进程写入文件的相应块即可。零位将写入第一个兆字节，第1个将写入第二个兆字节，依此类推。对于您的3d阵列，可能太简单了。
要进行更复杂的分解，请设置“文件视图”：构造一个MPI数据类型来描述磁盘上数据的布局。在您的情况下，MPI_TYPE_CREATE_SUBARRAY可能就是您想要的：一个3d全局矩阵和每个进程对全局图像有一定的影响。

MPI_File_write C

1 个答案: