如果我想定义自己的类型,并使用它作为MPI_Send的数据类型,只从矩阵中提取偶数行,那么该矩阵(发送缓冲区)是否必须静态分配?
当我动态地分配它时,我似乎遇到了问题。这是因为需要连续的地址才能发送数据吗?
发布于 2014-05-09 07:39:27
不,与MPI_Send一起发送的内存不必静态分配。
要发送数组子集,您可能需要使用MPI_Type_indexed。下面是来自MPI_Type_indexed的一个稍微修改过的示例版本,在这里我替换了静态分配的缓冲区
int buffer[27];到动态分配的缓冲区。
int* buffer = (int*)malloc(27 * sizeof(int));我希望它能帮上忙:
#include <mpi.h>
#include <stdio.h>
int main(int argc, char *argv[])
{
    int rank, size, i;
    MPI_Datatype type, type2;
    int blocklen[3] = { 2, 3, 1 };
    int displacement[3] = { 0, 3, 8 };
    int* buffer = (int*)malloc(27 * sizeof(int)); //int buffer[27];
    MPI_Status status;
    MPI_Init(&argc, &argv);
    MPI_Comm_size(MPI_COMM_WORLD, &size);
    if (size < 2)
    {
        printf("Please run with 2 processes.\n");
        MPI_Finalize();
        return 1;
    }
    MPI_Comm_rank(MPI_COMM_WORLD, &rank);
    MPI_Type_contiguous(3, MPI_INT, &type2);
    MPI_Type_commit(&type2);
    MPI_Type_indexed(3, blocklen, displacement, type2, &type);
    MPI_Type_commit(&type);
    if (rank == 0)
    {
        for (i=0; i<27; i++)
            buffer[i] = i;
        MPI_Send(buffer, 1, type, 1, 123, MPI_COMM_WORLD);
    }
    if (rank == 1)
    {
        for (i=0; i<27; i++)
            buffer[i] = -1;
        MPI_Recv(buffer, 1, type, 0, 123, MPI_COMM_WORLD, &status);
        for (i=0; i<27; i++)
            printf("buffer[%d] = %d\n", i, buffer[i]);
        fflush(stdout);
    }
    MPI_Finalize();
    free(buffer);
    return 0;
}https://stackoverflow.com/questions/23552968
复制相似问题