我们对MPI_Send
或MPI_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个流程)
答案 0 :(得分:1)
您的代码不正确。您只有在相应的发送完成后才发布接收。 MPI_Send
只能保证在发布相应的MPI_Recv
后完成,因此您会遇到经典的死锁。
它恰好适用于小消息,因为它们的处理方式不同(使用意外的消息缓冲区作为性能优化)。在这种情况下,MPI_Send
可以在发布MPI_Recv
之前完成。
或者你可以:
MPI_Isend
,MPI_Irecv
)以解决僵局。MPI_Sendrecv
。MPI_Allgather
。我推荐后者。