使用MPISend和MPIReceive时,主节点上的内存使用量增加

时间:2013-03-05 06:33:27

标签: mpi

我正在使用MPI编写程序,其中master将任务分配给从属节点。每个从节点在本地执行任务,并将结果(大小为100000的int数组)发送到主节点。

虽然我得到了正确的结果,但内存使用并不是线性的。我发现主节点占用N * m内存,其中N是节点数,m是从节点通常使用的内存。

任何人都知道为什么会这样,并且有任何解决方案可以减少主节点上的内存使用。

以下是从节点将一些数据/结果发送到主节点的示例代码,我想知道为什么在主节点使用的内存为N * m。我使用Linux命令top检查了内存使用情况。

#include<iostream>  
#include <mpi.h>
using namespace std;
int main(int argv, char *argp[256])
{
    int rank,size,master_rank=0,i=0;
    int jc=0,jpt,jobsperthread=0,exjpt=0;;
    int ii=0,index=0,remaining=0,tobesent=0,tobereceived=0;
    int totsendreceivesize=100000,k=0;
    int innodes=11;
    MPI_Status status;
    int *arr_anti_net=(int*)malloc(sizeof(int)*(totsendreceivesize+100));
    MPI_Init (&argv, &argp);
    MPI_Comm_rank (MPI_COMM_WORLD, &rank);
    MPI_Comm_size (MPI_COMM_WORLD, &size);
    for(i=0;i<totsendreceivesize;i++)
       arr_anti_net[i]=i
    if(rank!=master_rank)
    {
       remaining=totsendreceivesize;
       tobesent=256;
       k=0;
       while(remaining!=0)
       {
        if(remaining<256)
            tobesent=remaining;
        MPI_Send(&arr_anti_net[k],tobesent,MPI_INT,0,11,MPI_COMM_WORLD);
        k+=tobesent;
        remaining-=tobesent;
       }
    }
    else
    {
       ii=0;
       index=0;
       for(ii=1;ii<size;ii++)
       {
           jc=0;
           jpt=0;
           jobsperthread=innodes/size;
           jpt=innodes/size;
           exjpt=0;
           if(innodes%size!=0)
           {
            if(ii<innodes%size)
            {
                jobsperthread+=1;
                exjpt=ii;
            }
            else
                exjpt=innodes%size;
           }
           remaining=256;//totsendreceivesize;
           tobereceived=256;
           k=0;
           while(remaining!=0)
           {
            if(remaining<256)
                tobereceived=remaining;
            MPI_Recv(& arr_anti_net[k],tobereceived,MPI_INT,ii,11,MPI_COMM_WORLD,&status);
            k+=tobereceived;
            remaining-=tobereceived;
           }
       }
      }
   MPI_Barrier(MPI_COMM_WORLD);

   MPI_Finalize();
}

非常感谢

0 个答案:

没有答案