使用MPI发送二维数组

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

#define ROWS 3
#define COLS 3

int main(int argc, char** argv) {
    int rank, size;
    int array[ROWS][COLS] = {{1, 2, 3}, {4, 5, 6}, {7, 8, 9}};
    int sendbuf[ROWS * COLS];
    int recvbuf[ROWS * COLS];

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

    // 将二维数组展开为一维数组
    for (int i = 0; i < ROWS; i++) {
        for (int j = 0; j < COLS; j++) {
            sendbuf[i * COLS + j] = array[i][j];
        }
    }

    // 发送和接收数据
    MPI_Sendrecv(sendbuf, ROWS * COLS, MPI_INT, (rank + 1) % size, 0,
                 recvbuf, ROWS * COLS, MPI_INT, (rank - 1 + size) % size, 0,
                 MPI_COMM_WORLD, MPI_STATUS_IGNORE);

    // 打印接收到的数据
    printf("Process %d received:\n", rank);
    for (int i = 0; i < ROWS; i++) {
        for (int j = 0; j < COLS; j++) {
            printf("%d ", recvbuf[i * COLS + j]);
        }
        printf("\n");
    }

    MPI_Finalize();
    return 0;
}