在MPI_Allreduce中使用相同输入输出缓冲区是否安全？

通常情况下是安全的，以下是具体说明：

• MPI标准层面：MPI官方标准明确允许这种“就地操作”（in-place operation）。规范要求实现必须先完成所有进程输入数据的收集，再执行归约计算并将结果写回缓冲区，不会因为提前覆盖输入数据导致计算错误，主流的OpenMPI、MPICH等实现都完全支持这一特性。

• OpenMPI的实际表现：你在Linux/Ubuntu上测试的结果符合预期，OpenMPI对就地操作做了完善的内部处理，会暂存输入数据避免冲突，所以代码能正常运行。

• 需要注意的细节：

  • 你的示例代码存在类型不匹配问题：变量max_quantities是int类型，但MPI数据类型指定了MPI_DOUBLE，这会触发未定义行为，建议修正为匹配的类型：

    int max_quantities = ...;
    MPI_Allreduce(&max_quantities, &max_quantities, 1, MPI_INT, MPI_MAX, comm);
    

    或者将变量改为double类型并对应使用MPI_DOUBLE。
  • 兼容性问题：极少数老旧或小众MPI实现可能不支持就地操作，如果代码需要跨平台运行，建议提前确认目标环境的MPI实现是否支持；另外，自定义归约操作可能对就地操作有额外限制，需参考对应实现的文档。

内容的提问来源于stack exchange，提问作者kstn