MPI_Send或MPI_Recv的限制?

时间:2016-05-13 10:10:31

标签: c mpi openmpi hpc

我们对MPI_SendMPI_Recv上的邮件大小有限制吗?或计算机限制?当我尝试发送大数据时,它无法完成。 这是我的代码:

#include <stdio.h>
#include <stdlib.h>
#include <mpi.h>
#include <math.h>
#include <string.h>

void AllGather_ring(void* data, int count, MPI_Datatype datatype,MPI_Comm communicator)
{
  int me;
  MPI_Comm_rank(communicator, &me);
  int world_size;
  MPI_Comm_size(communicator, &world_size);
  int next=me+1;
  if(next>=world_size)
      next=0;
  int prev=me-1;
  if(prev<0)
      prev=world_size-1;
  int i,curi=me;
  for(i=0;i<world_size-1;i++)
  {
     MPI_Send(data+curi*sizeof(int)*count, count, datatype, next, 0, communicator);
     curi=curi-1;
     if(curi<0)
         curi=world_size-1;
     MPI_Recv(data+curi*sizeof(int)*count, count, datatype, prev, 0, communicator, MPI_STATUS_IGNORE);
  }
}


void test(void* buff,int world_size,int count)
{
    MPI_Barrier(MPI_COMM_WORLD);
    AllGather_ring(buff,count,MPI_INT,MPI_COMM_WORLD);
    MPI_Barrier(MPI_COMM_WORLD);
    }
}
void main(int argc, char* argv[]) {
    int count = 20000;
    char processor_name[MPI_MAX_PROCESSOR_NAME];
    MPI_Init(&argc,&argv);
    int world_rank,world_size,namelen;
    MPI_Comm_size(MPI_COMM_WORLD, &world_size);
    MPI_Comm_rank(MPI_COMM_WORLD, &world_rank);
    int* buff=(int*) malloc(world_size*sizeof(int)*count);
      int i;
      for (i = 0; i < world_size; i++) {
          buff[i]=world_rank;
      }
    test(buff,world_size,count);
    MPI_Finalize();
}

当我尝试使用大约80000字节(40000整数)的缓冲区运行时它停止了 (按计数= 20000 + 4个流程)

1 个答案:

答案 0 :(得分:1)

您的代码不正确。您只有在相应的发送完成后才发布接收。 MPI_Send只能保证在发布相应的MPI_Recv后完成,因此您会遇到经典的死锁。

它恰好适用于小消息,因为它们的处理方式不同(使用意外的消息缓冲区作为性能优化)。在这种情况下,MPI_Send可以在发布MPI_Recv之前完成。

或者你可以:

  • 立即发送或接收(MPI_IsendMPI_Irecv)以解决僵局。
  • 使用MPI_Sendrecv
  • 使用MPI_Allgather

我推荐后者。