如何实现C/C++程序同时兼容MPI版本与非MPI版本？

这个需求完全可以实现，只需要修复原代码的两处缺陷，调整条件编译逻辑即可。

原代码存在的问题 #

• 拼写错误：MPI初始化函数应为MPI_Init（首字母I大写），原代码写为MPI_init会直接编译失败
• 头文件依赖问题：无MPI环境时#include "mpi.h"会因找不到头文件直接报错，没有进入后续MPI_VERSION判断的机会

修改后的代码 #

#include <stdio.h>

// 兼容C++17及以上版本的头文件存在性判断，若使用更老的标准可替换为手动定义宏控制
#if defined(__has_include) && __has_include(<mpi.h>)
#include <mpi.h>
#define HAS_MPI 1
#else
#define HAS_MPI 0
#endif

int main(int argc, char *argv[]){
  int num_proc;
#if HAS_MPI && defined(MPI_VERSION)
  MPI_Init(&argc, &argv);
  MPI_Comm_size(MPI_COMM_WORLD, &num_proc);
  MPI_Finalize();
#else
  num_proc = 1;
#endif

  printf("%d\n", num_proc);
  return 0;
}

测试效果 #

• 无MPI环境直接编译运行：执行g++ main.cpp && ./a.out，代码走非MPI分支，输出1，符合要求
• MPI环境下编译运行：执行mpicxx main.cpp && mpiexec -n 2 ./a.out，代码走MPI分支，两个进程分别输出2，符合要求

老版本C++标准兼容方案 #

如果使用的C++版本不支持__has_include特性，可手动通过编译宏控制：

• 非MPI编译：执行g++ -DNO_MPI main.cpp && ./a.out，代码中新增NO_MPI宏的判断分支即可
• MPI编译：执行mpicxx main.cpp && mpiexec -n 2 ./a.out，无需额外参数

内容的提问来源于stack exchange，提问作者LI.LE