在MPI_IScatterv()问题中设置位移数组

时间:2016-04-24 22:25:17

标签: c++ mpi

嘿我在MPI_IScatterv()函数中设置位移数组时遇到问题。

我需要将部分图片分散到不同的进程中。我已经制作了自己的MPI_Datatype:

private: MPI_Datatype createMPI_PixelType(){
MPI_Datatype new_type;

int count = 3;
int blocklens[] = { 1,1,1 };

MPI_Aint indices[3];
indices[0] = (MPI_Aint)offsetof(struct Pixel, Red);
indices[1] = (MPI_Aint)offsetof(struct Pixel, Green);
indices[2] = (MPI_Aint)offsetof(struct Pixel, Blue);

MPI_Datatype old_types[] = { MPI_INT, MPI_INT, MPI_INT };

MPI_Type_create_struct(count, blocklens, indices, old_types, &new_type);
MPI_Type_commit(&new_type);

return new_type;}

结构Pixel

struct Pixel {
int Red;
int Green;
int Blue; }

作为函数参数,我需要传递位移数组。 我的不足之处是每个进程都开始从指定为disp [proc_id]的不同位置读取sendBuffer。例如,进程0 disp = 0, 进程1 disp = sizeof 0元素, 进程2 disp = sizeof 1 + 0元素等 这是代码

...
    //specify arrays for scathering
    sendCountArray = calcSendCounts(workTable);
    displacementArray = calcDisplacementArray(workTable);

    Pixel *fullPicture = pictureToPixelsArray();
    cout << endl;
    for (int i = 0; i < pixelsOnPicture; i++) {
        cout <<
            " r:" << fullPicture[i].Red <<
            " g:" << fullPicture[i].Green <<
            " b:" << fullPicture[i].Blue << endl;
    }

    cout << endl << "sendArray:" << endl;
    for (int i = 0; i < worldSize; i++) {
        cout << "i:" << i << " " << sendCountArray[i] << endl;
    }
    cout << endl <<  "displacementArray:" << endl;
    for (int i = 0; i < worldSize; i++) {
        cout << "i:" << i << " " << displacementArray[i] << endl;
    }
}
cout << endl; 
//
MPI_Scatter(sendCountArray, 1, MPI_INT, &pixelsSizeBuffer, 1, MPI_INT, 0, MPI_COMM_WORLD);

//sending part of picture
MPI_Request request;
MPI_Datatype mMPI_PIXEL = createMPI_PixelType();
partPicture = new Pixel[pixelsSizeBuffer];
MPI_Iscatterv(pictureToPixelsArray(), sendCountArray, displacementArray, mMPI_PIXEL, partPicture, pixelsSizeBuffer, mMPI_PIXEL, 0, MPI_COMM_WORLD, &request);
for (int i = 0; i < pixelsSizeBuffer; i++) {
    cout << "proc:" << procID <<
        " r:" << partPicture[i].Red <<
        " g:" << partPicture[i].Green <<
        " b:" << partPicture[i].Blue << endl;
}

这里是测试输出(6个过程)

pixelsOnPicture:9
workTableSize: 6

 r:255 g:242 b:0
 r:63 g:72 b:204
 r:237 g:28 b:36
 r:0 g:162 b:232
 r:163 g:73 b:164
 r:63 g:11 b:15
 r:34 g:177 b:76
 r:255 g:242 b:0
 r:63 g:10 b:16

sendArray:
i:0 2
i:1 2
i:2 2
i:3 1
i:4 1
i:5 1

displacementArray:
i:0 0
i:1 2
i:2 4
i:3 6
i:4 7
i:5 8

proc:4 r:-842150451 g:-842150451 b:-842150451
proc:0 r:255 g:242 b:0
proc:2 r:-842150451 g:-842150451 b:-842150451
proc:5 r:-842150451 g:-842150451 b:-842150451
proc:0 r:63 g:72 b:204
proc:2 r:-842150451 g:-842150451 b:-842150451
proc:1 r:-842150451 g:-842150451 b:-842150451
proc:3 r:-842150451 g:-842150451 b:-842150451
proc:1 r:-842150451 g:-842150451 b:-842150451

前2个元素正确发送..

1 个答案:

答案 0 :(得分:2)

您尚未等待请求,因此无法保证在检查输出时Iscatterv已完成。调用普通的旧Scatterv将确保它在调用后完成,这样您就可以检查是否已正确设置所有参数和数据类型。或者在调用Iscatterv之后添加MPI_Wait(&amp; request),但请注意,您在此处所做的只是重新实现阻塞调用。