C++并行计算 MPI Tutorial-9

---

在以前的所有教程中，我们都使用通讯器 MPI_COMM_WORLD。对于简单的应用程序，这已经足够了，因为我们的进程数量相对较少，并且通常要么一次要与它们之一对话，要么一次要与所有进程对话。当应用程序开始变大时，这变得不那么实用了，我们可能只想一次与几个进程进行对话。在本课程中，我们将展示如何创建新的通讯器 ，以便一次与原始进程组的子集进行通讯。

通讯器概述

对于简单的应用程序，使用 MPI_COMM_WORLD 进行所有操作并不罕见，但是对于更复杂的用例，拥有更多的通讯器可能会有所帮助。例如，如果您想对网格中一部分进程进行计算。例如，每一行中的所有进程都可能希望对一个值求和。这将我们带到了用于创建新的通讯器的第一个也是最常见的功能：

MPI_Comm_split(
	MPI_Comm comm,
	int color,
	int key,
	MPI_Comm* newcomm)

顾名思义，MPI_Comm_split 通过基于输入值 color 和 key 将通讯器 “拆分”为一组子通讯器来创建新的通讯器 。在这里需要注意的是，原始的通讯器并没有消失，但是在每个过程中都会创建一个新的通讯器。第一个参数 comm 是将用作新通讯器基础的通讯器。这可以是 MPI_COMM_WORLD，但也可以是任何其他通讯器。第二个参数 color 确定每个进程将属于哪个新的通讯器。所有传递相同 color 的进程都分配给同一通讯器。如果颜色是 MPI_UNDEFINED，则该进程将不会包含在任何新的通讯器中。第三个参数 key 确定每个新通讯器中的顺序（秩）。传递键的最小值的进程将为秩0 ，第二个最小值将为秩1 ，依此类推。如果出现相等的情况，则在原始通讯器中秩较低的进程将是第一位。最后一个参数 newcomm 是 MPI 如何将新的通讯器返回给用户。

使用多个通讯器的示例

现在，让我们看一个简单的示例，在该示例中，我们尝试将单个全局通讯器拆分为一组较小的通讯器。在此示例中，我们将假设我们已将逻辑模型原始的通讯器布局为由 16 个进程组成的 4x4 网格，并且希望按行划分网格。为此，每一行将获得自己的颜色。在下图中，您可以看到左侧具有相同颜色的每组过程如何最终在右侧拥有自己的通讯器。

代码如下：

// 获取原始通讯器的秩和大小
int world_rank, world_size;
MPI_Comm_rank(MPI_COMM_WORLD, &world_rank);
MPI_Comm_size(MPI_COMM_WORLD, &world_size);

int color = world_rank / 4; // 根据行确定角色

// 根据颜色拆分通讯器，使用原始秩排名
MPI_Comm row_comm;
MPI_Comm_split(MPI_COMM_WORLD, color, world_rank, &row_comm);

int row_rank, row_size;
MPI_Comm_rank(row_comm, &row_rank);
MPI_Comm_size(row_comm, &row_size);

printf("WORLD RANK/SIZE: %d/%d \t ROW RANK/SIZE: %d/%d\n",
	world_rank, world_size, row_rank, row_size);

MPI_Comm_free(&row_comm);<