尝试发送 2D 向量的列时MPI_Scatterv分段错误

问：

我有一个大小为 NxM 的 2d 向量，其中 N=13 和 M=17 和 8 个过程。我想将这个 2d 向量的列分发到进程中，因此 p0 将有 3 列，而所有其他进程将有 2 列（所有列将有 13 个整数）。 我无法在我的代码中看到任何错误，但它在执行MPI_Scatterv函数时给了我分段错误。

这是我的代码：

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

#define N 13
#define M 17

int main(int argc, char *argv[]){
    int matrix[N*M];
    int menum, nproc, i, j;
    
    MPI_Init(&argc, &argv);
    
    MPI_Comm_rank(MPI_COMM_WORLD, &menum);
    MPI_Comm_size(MPI_COMM_WORLD, &nproc);
    
    //Matrix initialization
    if (menum == 0) {
        for (i = 0; i < N; i++) {
            for (j = 0; j < M; j++) {
                matrix[(i*M)+j] = i * M + j;
            }
        }
    }
    
    //Every process has its vector of size nloc*mloc
    int nloc = N;
    int mloc = menum < M%nproc ? (M/nproc)+1 : M/nproc;
    int *recv_vec = (int *)malloc(nloc*mloc*sizeof(int));
    
    //Create a custom datatype for a vector of nloc integers
    MPI_Datatype col_type;
    MPI_Type_vector(nloc, 1, M, MPI_INT, &col_type);
    MPI_Type_commit(&col_type);
    
    //Distributing data to other processes
    int sendcounts[]={3,2,2,2,2,2,2,2};
    int displs[]={0,3,5,7,9,11,13,15};
    MPI_Scatterv(matrix, sendcounts, displs, col_type, recv_vec, nloc*mloc, MPI_INT, 0, MPI_COMM_WORLD);
    
    //Freeing up memory and terminate MPI environment
    free(recv_vec);
    MPI_Type_free(&col_type);
    MPI_Finalize();

    return 0;
}

编辑：

如评论中所述，我不得不使用 ，因此可以计算以下调用的正确偏移量。MPI_Type_create_resized()MPI_Scatterv