MPI只收集Root Process的收集

时间:2016-10-18 17:42:32

标签: c parallel-processing mpi openmp

首先,我一直在使用this code作为参考,其中显示MPI_Gather没有使用MPI_Scatter的情况,因为我正在尝试在此处实现这一点。我现在已经在这方面工作了很长时间,只是无法弄清楚这个问题。这种sobel边缘检测算法增强了图像内对象的轮廓。

我会在下面发布我的代码,因为没有太多,但我会先给出快速的代码说明。

我正在尝试将顺序程序转换为并行程序。所以所有非MPI代码都是正确的。

因此我的MPI代码只能出错。

int main(int argc, char **argv) {

     FILE *inFile, *oFile;
     int grayImage[N][N], edgeImage[N][N];
     char type[2];
     int w, h, max;
     int r, g, b, y, x, i, j, sum, sumx, sumy;
     int tid;

     int GX[3][3], GY[3][3];
     double elapsed_time;
     struct timeval tv1, tv2;
     int error = 0;
     char buffer[BUFSIZ];
     int rank, NP;

     // Code lies here for reading from the image file and storing into the grayImage array. 
     // This works so I saw no reason to include it

     /* 3x3 Sobel masks. */
     GX[0][0] = -1; GX[0][1] = 0; GX[0][2] = 1;
     GX[1][0] = -2; GX[1][1] = 0; GX[1][2] = 2;
     GX[2][0] = -1; GX[2][1] = 0; GX[2][2] = 1;

     GY[0][0] =  1; GY[0][1] =  2; GY[0][2] =  1;
     GY[1][0] =  0; GY[1][1] =  0; GY[1][2] =  0;
     GY[2][0] = -1; GY[2][1] = -2; GY[2][2] = -1;



     MPI_Init(NULL, NULL);

     MPI_Comm_size(MPI_COMM_WORLD, &NP);
     MPI_Comm_rank(MPI_COMM_WORLD, &rank);

     // This calculates the block size.MPI 
     // On 4 processors the block size for a 100x100 image would be 25x100 each

     int blksz = (int)ceil((double)N/NP);

     // This creates a local array for each processor, soon to be gathered

     int tempEdge[blksz][N];

     // this line shows it's working correctly

     printf("processor %d, width: %d, height: %d, blksz: %d, begin: %d, end: %d\n", rank, w, h, blksz, rank*blksz, (rank+1)*blksz);

     for(x=rank*blksz; x < (rank+1)*blksz && x<h; x++){

        // Any code in this loop can be ignored as it works correctly.

         for(y=0; y < w; ++y){

             sumx = 0;
             sumy = 0;
             // handle image boundaries 
             if(x==0 || x==(h-1) || y==0 || y==(w-1))
                 sum = 0;
             else{
                 //x gradient approx
                 for(i=-1; i<=1; i++)  {
                     for(j=-1; j<=1; j++){
                         sumx += (grayImage[x+i][y+j] * GX[i+1][j+1]);
                     }
                 }
                 //y gradient approx
                 for(i=-1; i<=1; i++)  {
                     for(j=-1; j<=1; j++){
                         sumy += (grayImage[x+i][y+j] * GY[i+1][j+1]);
                     }
                 }
                 //gradient magnitude approx
                 sum = (abs(sumx) + abs(sumy));
             }
             tempEdge[x][y] = clamp(sum);
         }
     }

     // Here is the line I am guessing is causing the problem

     MPI_Gather(&tempEdge, w*blksz, MPI_INT,
               &edgeImage, w*blksz, MPI_INT, 0,
               MPI_COMM_WORLD);


     // Finally, I output edgeImage to a file here.

     if(rank==0){

         // output edgeImage to File

     }

     MPI_Finalize();

     return 0;    
}

我正在使用的输入图像是:

enter image description here

但输出只是给出了这个:

enter image description here

正如您所看到的,它只是图像的前四分之一(N / 4)或blksz

这意味着MPI_Gather仅从排名为0的流程中收集?

我已经花了这么多时间在这上面,任何帮助都会非常感激!

1 个答案:

答案 0 :(得分:2)

不要因为其他代码中的错误而责怪MPI集体。实际上,您的代码在没有segfaulting的情况下生成损坏的图像实际上是一个奇迹。看看那部分:

int tempEdge[blksz][N];
             ~~~~~

for(x = rank*blksz; x < (rank+1)*blksz && x<h; x++){
        ~~~~~~~~~~
   for(y = 0; y < w; ++y){
      ...
      tempEdge[x][y] = clamp(sum);    (1)
               ~
   }
}

任何排名&gt; 0代码写入数组的末尾。将(1)处的陈述修改为:

tempEdge[x - rank*blksz][y] = clamp(sum);

另外,请移除&电话中的MPI_Gather

MPI_Gather(tempEdge, w*blksz, MPI_INT,
           edgeImage, w*blksz, MPI_INT, 0,
           MPI_COMM_WORLD);

它也适用于&,但这在技术上是不正确的。如果您坚持使用地址 - 运算符,请改用&tempEdge[0][0]&edgeImage[0][0]