Question

我有一个矩阵，我分成几部分。每个部分都由处理器处理，最后我希望每个处理器都有完整的矩阵。对于例如A [100] [4]，拆分为2个处理器X适用于0到49行，处理器Y适用于50到99行。最后，我希望两个处理器都有A [100] [4]的新值。这是我的代码。

#include"mpi.h"
#include<iostream>
#include<cmath>

using namespace std;

#define MASTER  0
#define BEGIN   1


const int COLS=100;

int main(int argc, char *argv[])
{
     double t1=MPI_Wtime(),trans_grad[COLS][8];
    int taskid,
    numtasks,
    numworkers,columns,
    column_mean,
    offset,
    tag=BEGIN,
    mtag,
    source,
    ext,left,
    size,
    msgtype,dest;
    MPI_Status status;

MPI_Init(&argc,&argv);
MPI_Comm_rank(MPI_COMM_WORLD,&taskid);
MPI_Comm_size(MPI_COMM_WORLD,&numtasks);
numworkers=numtasks-1;

int columns_send[numworkers+1],sum_s;
column_mean=COLS/numworkers;
ext=COLS%numworkers;
offset=0;

if (taskid==MASTER)
{
    for (int i=1;i<=numworkers;i++)
    {

        dest=i;
        columns=(i<=ext) ? column_mean+1:column_mean;

        MPI_Send(&offset, 1, MPI_INT, dest, tag, MPI_COMM_WORLD);        
        MPI_Send(&columns, 1, MPI_INT, dest, tag, MPI_COMM_WORLD);

        offset=offset+columns;            

    }
    MPI_Finalize();

}

if (taskid!=MASTER)
{
    for (int i=0;i<COLS;i++)
    {
        for (int j=0;j<8;j++)
        {
            trans_grad[i][j]=0.0;
        }
    }

    source=MASTER;
    msgtype=BEGIN;

    MPI_Recv(&offset, 1, MPI_INT, source, msgtype, MPI_COMM_WORLD, &status);
    MPI_Recv(&columns, 1, MPI_INT, source, msgtype, MPI_COMM_WORLD, &status);

    columns_send[taskid]=columns;

    for (int i=offset;i<offset+columns;i++)
    {
        for (int j=0;j<8;j++)
        {
            trans_grad[i][j]=taskid+1.5;
        }
    }


    for (int j=1;j<=numworkers;j++)
    {
       if (j!=taskid)
        {
            mtag=taskid;
            source=j;

            MPI_Send(&columns_send[taskid], 1, MPI_INT, j, mtag, MPI_COMM_WORLD);                
            MPI_Recv(&columns_send[j], 1, MPI_INT, source, source, MPI_COMM_WORLD,&status);              
        }
    }

    sum_s=0;
    for (int yy=taskid-1;yy>=1;yy--)
    {
        sum_s=sum_s-columns_send[yy];
    }

    for (int j=1;j<=numworkers;j++)
    {
       if (j!=taskid)
        {
            mtag=taskid;
            source=j;
            MPI_Send(&trans_grad[offset][0], columns_send[taskid]*8, MPI_DOUBLE, j, mtag, MPI_COMM_WORLD);
            MPI_Recv(&trans_grad[offset+sum_s][0], columns_send[j]*8, MPI_DOUBLE, source, source, MPI_COMM_WORLD,&status);              
        }
        sum_s=sum_s+columns_send[j];                
    }
    MPI_Finalize();
}

} 该程序在COLS = 100和150时工作正常，但它只会卡住COLS = 500,1000。我不知道为什么会这样。

MPI程序适用于小输入大小但不适用于大输入

0 个答案: