在套接字编程中,我们有select()函数,它允许我们同时检查多个套接字。我想知道MPI库中是否还有这样的功能?
在以下代码的第一个for循环中,我正在从一个节点向每个其他节点发送多个非阻塞发送和接收请求。在第二个for循环而不是按顺序等待每个节点,我想开始处理首先发送其数据的节点的数据。我想知道有没有办法做到这一点?
for(id=0; id<numtasks; id++){
if(id == taskid) continue;
if(sendCount[id] != 0) MPI_Isend(sendBuffer[id], N*sendCount[id], MPI_DOUBLE, id, tag, MPI_COMM_WORLD, &reqs[id]);
if(recvCount[id] != 0) MPI_Irecv(recvBuffer[id], N*recvCount[id], MPI_DOUBLE, id, tag, MPI_COMM_WORLD, &reqs[id]);
}
for(id=0; id<numtasks; id++){
if(id == taskid) continue;
if(recvCount[id] != 0){
MPI_Wait(&reqs[id], &status);
for(i=0; i<recvCount[id]; i++)
splitData(N, recvBuffer[id] + N*i, U[toRecv[id][i]]);
}
}
根据给出的答案,我试图修改我的代码,但我仍然在运行时遇到分段错误错误。请帮我弄清楚错误。
for(id=0; id<numtasks; id++){
if(id == taskid) continue;
if(sendCount[id] != 0) MPI_Isend(sendBuffer[id], N*sendCount[id], MPI_DOUBLE, id, tag, MPI_COMM_WORLD, &reqs[id]);
if(recvCount[id] != 0) MPI_Irecv(recvBuffer[id], N*recvCount[id], MPI_DOUBLE, id, tag, MPI_COMM_WORLD, &reqs[id]);
}
reqs[taskid] = reqs[numtasks-1];
for(i=0; i<numtasks-1; i++){
MPI_Waitany(numtasks-1, reqs, &id, &status);
if(id == taskid) id = numtasks-1;
for(i=0; i<recvCount[id]; i++)
splitData(N, recvBuffer[id] + N*i, U[toRecv[id][i]]);
}
答案 0 :(得分:3)
最接近的等价物是MPI_Waitsome
,您提供一个请求列表,并在完成至少一个请求后立即返回。但是,select
中没有超时。还有MPI_Waitany
,MPI_Waitall
以及MPI_Testany
,MPI_Testall
,MPI_Testsome
。
any
和some
变体的主要区别在于界面通知您一个或多个已完成的请求。
编辑:您需要为每个操作使用单独的请求,特别是发送和接收操作。