openmpi入门之第一个程序

编译运行hello world

首先给出代码：

#include <stdio.h>
#include "mpi.h"

int main(int argc, char* argv[])
{
    int rank, size, len;
    char version[MPI_MAX_LIBRARY_VERSION_STRING];

    MPI_Init(&argc, &argv);
    MPI_Comm_rank(MPI_COMM_WORLD, &rank);
    MPI_Comm_size(MPI_COMM_WORLD, &size);
    MPI_Get_library_version(version, &len);
    printf("Hello, world, I am %d of %d, (%s, %d)\n",
           rank, size, version, len);
    MPI_Finalize();

    return 0;
}

编译：

mpicc hello_c.c -o hello_c

当然也可以直接用gcc进行编译：

g++ -fopenmp -I$OPEN_MPI_ROOT/include -L$OPEN_MPI_ROOT/lib -lmpi hello_c.c -o hello_c

注意，这里OPEN_MPI_ROOT为安装目录。

运行：

mpirun -np 4 hello_c

如果添加ip则可以写为：

mpirun -np 4 -host 127.0.0.1:4 hello_c

这里-np 4表示运行时开启了4个进程，注意ip后面的:4是slot的个数，关于slot的概念后面会介绍，这里需要注意slot个数至少要大于等于对应机器上的进程数。

打印结果：

Hello, world, I am 2 of 4, (Open MPI v4.1.5, package: Open MPI s@s-ZHENGJIUZHE-REN7000 Distribution, ident: 4.1.5, repo rev: v4.1.5, Feb 23, 2023, 118)
Hello, world, I am 0 of 4, (Open MPI v4.1.5, package: Open MPI s@s-ZHENGJIUZHE-REN7000 Distribution, ident: 4.1.5, repo rev: v4.1.5, Feb 23, 2023, 118)
Hello, world, I am 1 of 4, (Open MPI v4.1.5, package: Open MPI s@s-ZHENGJIUZHE-REN7000 Distribution, ident: 4.1.5, repo rev: v4.1.5, Feb 23, 2023, 118)
Hello, world, I am 3 of 4, (Open MPI v4.1.5, package: Open MPI s@s-ZHENGJIUZHE-REN7000 Distribution, ident: 4.1.5, repo rev: v4.1.5, Feb 23, 2023, 118)

代码解析

第一个函数MPI_Init是MPI的初始化函数，其会完成所有初始化工作，以备后续MPI库的调用进行处理，该函数需要在调用其他MPI函数之前调用。

MPI_Comm_rank函数和MPI_Comm_size函数，在MPI中，每一个活动的进程都是某个通信阈中的成员，MPI_COMM_WORLD是默认的通信阈，如果不做其他声明，所有进程都属于这个通信阈，大多数情况下，这已足够用了。一个通信阈内的进程是有序的，每一个进程都有自己唯一的序号，对于有p个进程的通信阈，序号取值为0~p-1，通常我们通过这一序号来决定这个进程负责计算哪一部分任务，MPI_Comm_rank来确定进程的序号，MPI_Comm_size来确定当前通信阈的进程总数：

MPI_Comm_rank(MPI_COMM_WORLD, &id)
MPI_Comm_size(MPI_COMM_WORLD, &p)

MPI_Finalize函数用来让系统释放分配给MPI的资源。

运行MPI程序的命令是mpirun，-np标记要生成的进程个数。