c - MPI 子数组发送错误

Question

我首先初始化一个 4x4 矩阵，然后尝试使用 C 中的 MPI 将第一个 2x2 block 发送到从属进程。但是从属进程只接收 block 的第一行，第二行填充了来自电脑内存。我找不到丢失的东西。程序代码如下:

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

#define     SIZE        4

int main(int argc, char** argv)
{
int rank, nproc;
const int root = 0;
const int tag = 3;

int** table;
int* datas;

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

datas = malloc(SIZE * SIZE * sizeof(int));
table = malloc(SIZE * sizeof(int*));

for (int i = 0; i < SIZE; i++)
    table[i] = &(datas[i * SIZE]);

for (int i = 0; i < SIZE; i++)
    for (int k = 0; k < SIZE; k++)
        table[i][k] = 0;

table[0][1] = 1;
table[0][2] = 2;
table[1][0] = 3;
table[2][3] = 2;
table[3][1] = 3;
table[3][2] = 4;


if (rank == root){
    MPI_Datatype newtype;
    int sizes[2] = { 4, 4 };  // size of table 
    int subsizes[2] = { 2, 2 };  // size of sub-region 
    int starts[2] = { 0, 0 };

    MPI_Type_create_subarray(2, sizes, subsizes, starts, MPI_ORDER_C, MPI_INT, &newtype);
    MPI_Type_commit(&newtype);

    MPI_Send(&(table[0][0]), 1, newtype, 1, tag, MPI_COMM_WORLD);
}
else{
    int* local_datas = malloc(SIZE * SIZE * sizeof(int));
    int** local = malloc(SIZE * sizeof(int*));

    for (int i = 0; i < SIZE; i++)
        local[i] = &(local_datas[i * SIZE]);

    MPI_Recv(&(local[0][0]), 4, MPI_INT, root, tag, MPI_COMM_WORLD, MPI_STATUSES_IGNORE);

    for (int i = 0; i < 2; i++){
        for (int k = 0; k < 2; k++)
            printf("%3d ", local[i][k]);
        printf("\n");
    }
}


MPI_Finalize();

return 0;

Answer 1

您已指示接收操作将四个整数值连续放入内存中，因此 2x2 block 在接收时转换为 1x4 行(因为 local 是 4x4)。 local 的第二行包含随机值，因为内存从未被初始化。

您应该在发送方和接收方都使用 MPI_Type_create_subarray 以便将接收到的数据放入 2x2 block 中，或者将 local 重新定义为 2x2 矩阵而不是 4x4。