使用MPI散射连续的2D数组

时间:2016-04-30 16:03:31

标签: c multidimensional-array mpi

我似乎没有做到这一点,我想分散2D块并只打印分散的部分,这是为了测试目的
这就是我要做的事情......

#include <stdio.h>
#include <stdlib.h>
#include "mpi/mpi.h"

void print2D(int**, int, int);

int main(int argc, char* argv[])
{
    int rank, pSize, r1, c1, rowsPerProcess, i, j;
    int** matA = NULL, **partMat = NULL;

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

    if(rank == 0)
    {
    scanf("%d%d", &r1, &c1);
        rowsPerProcess = r1/pSize; // number of rows assigned to each process (Assume this produces no remainder)

    //Contiguous allocation
        int* mat1D = (int*)malloc(r1 * c1 * sizeof(int));
        matA = (int**)malloc(r1 * sizeof(int*));
        for(i=0; i<r1; i++)
          matA[i] = &mat1D[i * c1];

    for(i=0; i<r1; i++) //Fill the 2D array
      for(j=0; j<c1; j++)
        matA[i][j] = i + j;

    printf("Original Matrix:\n");
    print2D(matA, r1, c1);
    }

    MPI_Bcast(&c1, 1, MPI_INT, 0, MPI_COMM_WORLD); //Columns
    MPI_Bcast(&rowsPerProcess, 1, MPI_INT, 0, MPI_COMM_WORLD); //Rows in every process

    //Again, contiguous allocation for the partial 2D array
    int* partMat1D = (int*)malloc(rowsPerProcess * c1 * sizeof(int));
    partMat = (int**)malloc(rowsPerProcess * sizeof(int*));
    for(i=0; i<rowsPerProcess; i++)
          partMat[i] = &partMat1D[i * c1];

    MPI_Scatter(&(matA[0][0]), rowsPerProcess * c1, MPI_INT, &(partMat[0][0]),
    rowsPerProcess * c1, MPI_INT, 0, MPI_COMM_WORLD);

    printf("FROM PROCESS %d:\n", rank);
    print2D(partMat, rowsPerProcess, c1);

    MPI_Finalize();
    return 0;
}

void print2D(int** mat, int row, int col)
{
    if(mat != NULL)
    {
        int i, j;
        for(i=0; i<row; i++)
        {
            for(j=0; j<col; j++)
                printf("%d  ", mat[i][j]);
            printf("\n");
        }
    }
}

我在使用4个进程执行此操作时收到分段错误 例如

8 3
Original Matrix:
0  1  2  
1  2  3  
2  3  4  
3  4  5  
4  5  6  
5  6  7  
6  7  8  
7  8  9  
FROM PROCESS 0:
0  1  2  
1  2  3  

===================================================================================
=   BAD TERMINATION OF ONE OF YOUR APPLICATION PROCESSES
=   PID 2637 RUNNING AT minix-VirtualBox
=   EXIT CODE: 139
=   CLEANING UP REMAINING PROCESSES
=   YOU CAN IGNORE THE BELOW CLEANUP MESSAGES
===================================================================================
YOUR APPLICATION TERMINATED WITH THE EXIT STRING: Segmentation fault (signal 11)
This typically refers to a problem with your application.
Please see the FAQ page for debugging suggestions

只有进程0正确打印其部分。 我使用MPI_Scatter错了吗?我该如何解决这个问题? 我如何分散这个2D块?

1 个答案:

答案 0 :(得分:1)

在非主进程上,将matA初始化为NULL,然后继续评估表达式&matA[0][0]。这会导致matA取消引用,甚至在您的流程进入MPI_Scatter()之前就会崩溃。

由于您使用C编程,因此可以避免使用这些双指针:C具有自C99以来内置的非常好的多维数组支持,您可以使用它来使您的生活更轻松。只需声明matA作为指向行数组的指针

int (*matA)[c1];

然后为其分配r1行的数组

matA = malloc(r1*sizeof(*matA));

之后,您可以使用与上面代码中使用的代码相同的代码填充数组:

for(i=0; i<r1; i++) //Fill the 2D array
  for(j=0; j<c1; j++)
    matA[i][j] = i + j;

此外,由于您的2D阵列只需要一次malloc()来电,您也可以通过一次free()来电处理它:

free(matA);

注意:你不能在C ++中这样做,它不像C那样支持变长数组,也许永远不会。这是C比C ++强大得多的几点之一。