Macbook M1 MPI环境安装与测试案例

一、安装MPI

brew install open-mpi

经过一段时间等待后，输入如下命令。若输出相关版本信息，则安装成功。

二、MPI代码构成

MPI代码基本结构如下所示：

#include<mpi.h>
#include<stdio.h>
int main(int argc, char* argv[])
{
	int rank, size;
    MPI_Init(&argc, &argv);
    MPI_Comm_rank(MPI_COMM_WORLD, &rank);
    MPI_Comm_size(MPI_COMM_WORLD, &size);
    “”“
    并行化代码
    ”“”
    MPI_Finalize();
    return 0;
}

接下来我们逐一进行解析。

2.1 argc与argv

int main(int argc, char* argv)

在MPI程序中，需要向主函数传入int型变量argc与char*变量argv。这两个参数用于获取在使用命令行运行MPI程序时获取到进程数量参数n。

2.2 rank与size

可以看到在上述的程序结构中，我们定义了两个整型变量rank和size。

int rank, size;

其中rank用作记录当前进程的进程号；size则为整个程序的进程数。

MPI_Comm_rank(MPI_COMM_WORLD, &rank);
MPI_Comm_size(MPI_COMM_WORLD, &size);

这里我们使用MPI_Comm_rank和MPI_Comm_size分别获取当前进程号和总进程数量。
注意到我们传入的第一个标识(?)MPI_COMM_WORLD在mpi.h中代表当前程序的进程池，包含了当前程序的所有进程。

2.3 MPI_Init与MPI_Finalize

MPI_Init(&argc, &argv);

在程序开头部分，使用init函数来建立并行环境。这里传入命令行的数量参数以便获取准确唯一的并行环境。

MPI_Finalize();

在并行化程序部分结束之后和主函数return前使用MPI_Finalize函数来终止并行化环境。

三、HelloMPI

以下为程序示例：

// hello_mpi.cpp

#include<mpi.h>
#include<stdio.h>
int main(int argc, char* argv[])
{
	int rank, size;
    MPI_Init(&argc, &argv);
    MPI_Comm_rank(MPI_COMM_WORLD, &rank);
    MPI_Comm_size(MPI_COMM_WORLD, &size);
    printf("Process%d: hello!\n", rank);
    MPI_Finalize();
    return 0;
}