C中的MPI和多维数组分配答案

【问题标题】：MPI and multidimensional array allocation in CC中的MPI和多维数组分配
【发布时间】：2012-10-04 16:43:49
【问题描述】：

我正在尝试使用 MPI_Scatter，发送矩阵行（动态分配），但它只发送一行，其他行是垃圾。当我使用静态内存分配时 - 一切都很好。

MPI_Init(&argc, &argv);
int **matrix, *matrix_stor, *row,rank, P;
MPI_Comm_size(MPI_COMM_WORLD, &P);
row = new int [P];
for(int i = 0; i < P; i++)
{
    row[i] = 0;
}
matrix = new int *[P];
for(int i = 0; i < P; i ++)
    matrix[i] = new int [P];

//int matrix[4][4], row[4], rank, size;
MPI_Comm_rank(MPI_COMM_WORLD, &rank);

if (rank == 0){
    for(int i = 0; i < P; i++){
        for(int j = 0; j < P; j++){
            matrix[i][j] = rand()%20;
            cout << matrix[i][j] << " ";
        }
        cout << endl;
    }
}
cout << endl;

MPI_Scatter(&matrix[0][0], P, MPI_INT,&row[0], P, MPI_INT,0,MPI_COMM_WORLD);
for(int i = 0; i < P; i++)
    cout << row[i] << " ";
cout << endl;
free(matrix);
free(row);
MPI_Finalize();
return 0;

结果是：
源矩阵：
1 7 14 0
9 4 18 18
2 4 5 5
1 7 1 11
收到的行数：
1 7 14 0
3626672 3626800 0 0
16 1 119 -33686019
-33686019 -572662307 524296 786765

【问题讨论】：

MPI data type for 2D array 的可能重复项

标签： c++ memory mpi

【解决方案1】：

这不是 MPI 问题，而是 C 和多维数组问题。

问题在于，在 C 语言中，不能保证您分配的矩阵实际上是 P*P 连续 ints。因此，当您尝试从 matrix[0][0] 开始分散 P*P 整数时，谁知道您发送的是什么。

这段代码：

matrix = new int *[P];
for(int i = 0; i < P; i ++)
    matrix[i] = new int [P];

首先分配Ppointers-to-int 的数组，然后将每个数组设置为已分配P int 的指针。但是完全不能保证matrix[1] 开始于matrix[0] 结束的地方。如果你想这样做 - 并且你这样做了，通常对于 MPI 和通常对于技术计算应用程序中的多维数组 - 你必须手动分配连续的内存块并让 matrix 指向它：

matrix = new int *[P];
int *data = new int [P*P];
for(int i = 0; i < P; i ++)
    matrix[i] = &(data[P*i]);

然后分散应该起作用。

还要注意，你应该使用delete，而不是free()，来释放new分配的内存。

【讨论】：