0

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

我根据我将使用的 proc 数量的值将我的矩阵分成几块。

我得到了独立工作的较小的 3D 矩阵，但我希望它们计算的浓度值写入同一个文件中，例如 0 到 n-1 行由 proc 0 写入，n 到 2n-1 由 proc 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;
}
4

1 回答 1

0

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

  • 有时您可以让每个进程写入文件的相应块。rank 0 写入第一个 MB，rank 1 写入 2nd，依此类推。对于您的 3d 数组来说可能过于简单。

  • 对于更复杂的分解，您可以设置一个“文件视图”：您构建一个 MPI 数据类型来描述磁盘上数据的布局。在您的情况下， MPI_TYPE_CREATE_SUBARRAY 可能是您想要的：一个 3d 全局矩阵，每个进程都会为全局图片贡献一些子数组。

于 2018-11-30T03:20:05.763 回答