我的理解是,一个主进程向所有其他进程发送消息。作为回报,所有其他进程都会向主进程发送消息。这足以成为工作障碍吗?如果没有,那还需要什么?
2 回答
让我们看一下OpenMPI 的 barrier 实现。虽然其他实现可能略有不同,但一般的通信模式应该是相同的。
首先要注意的是 MPI 的屏障没有设置成本:到达MPI_Barrier
调用的进程将阻塞,直到该组的所有其他成员也调用了MPI_Barrier
。请注意,MPI 不要求他们到达同一个呼叫,只要对MPI_Barrier
. 因此,由于组中的节点总数对于每个进程都是已知的,因此不需要为初始化调用分配额外的状态。
现在,让我们看一些代码:
/*
* Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
* University Research and Technology
* Corporation. All rights reserved.
* Copyright (c) 2004-2005 The University of Tennessee and The University
* of Tennessee Research Foundation. All rights
* reserved.
* Copyright (c) 2004-2005 High Performance Computing Center Stuttgart,
* University of Stuttgart. All rights reserved.
* Copyright (c) 2004-2005 The Regents of the University of California.
* All rights reserved.
* Copyright (c) 2012 Oak Ridge National Labs. All rights reserved.
* [...]
*/
[...]
/*
* barrier_intra_lin
*
* Function: - barrier using O(N) algorithm
* Accepts: - same as MPI_Barrier()
* Returns: - MPI_SUCCESS or error code
*/
int
mca_coll_basic_barrier_intra_lin(struct ompi_communicator_t *comm,
mca_coll_base_module_t *module)
{
int i;
int err;
int size = ompi_comm_size(comm);
int rank = ompi_comm_rank(comm);
首先,所有节点(除了排名为 0 的节点,即根节点)向根节点发送它们已到达屏障的通知:
/* All non-root send & receive zero-length message. */
if (rank > 0) {
err =
MCA_PML_CALL(send
(NULL, 0, MPI_BYTE, 0, MCA_COLL_BASE_TAG_BARRIER,
MCA_PML_BASE_SEND_STANDARD, comm));
if (MPI_SUCCESS != err) {
return err;
}
之后,他们阻止等待来自根的通知:
err =
MCA_PML_CALL(recv
(NULL, 0, MPI_BYTE, 0, MCA_COLL_BASE_TAG_BARRIER,
comm, MPI_STATUS_IGNORE));
if (MPI_SUCCESS != err) {
return err;
}
}
根节点实现通信的另一端。首先它阻塞直到它收到n-1
通知(来自组中每个节点的通知,除了他自己,因为他已经在屏障调用中):
else {
for (i = 1; i < size; ++i) {
err = MCA_PML_CALL(recv(NULL, 0, MPI_BYTE, MPI_ANY_SOURCE,
MCA_COLL_BASE_TAG_BARRIER,
comm, MPI_STATUS_IGNORE));
if (MPI_SUCCESS != err) {
return err;
}
}
一旦所有通知都到达,它会发出每个节点正在等待的消息,表示每个人都已到达屏障,之后它会自行离开屏障调用:
for (i = 1; i < size; ++i) {
err =
MCA_PML_CALL(send
(NULL, 0, MPI_BYTE, i,
MCA_COLL_BASE_TAG_BARRIER,
MCA_PML_BASE_SEND_STANDARD, comm));
if (MPI_SUCCESS != err) {
return err;
}
}
}
/* All done */
return MPI_SUCCESS;
}
因此,通信模式首先是n:1
从所有节点到根,然后是1:n
从根返回到所有节点。为了避免根节点的请求超载,OpenMPI 允许使用基于树的通信模式,但基本思想是相同的:所有节点在进入屏障时通知根节点,而根节点聚合结果并在到达时通知每个人准备继续。
不,这还不够。一旦主进程向所有其他进程发送消息,通知它们已到达屏障,并且所有其他进程都响应说它们也已到达屏障,则只有主进程知道所有进程都已到达屏障。在这种情况下,需要从主进程向其他进程发送另一条消息。
我没有声称在任何库中实际实现 MPI 屏障,特别是我并不是建议在实践中使用概述的消息序列,只是在理论上存在缺陷。