python - python 脚本是由 mpirun 还是 mpiexec 启动的？

Question

我编写了一个 python 脚本，可以以独立方式启动或在 mpi 支持下启动。

python myscript.py

对比

mpirun -np 2 python myscript.py

我如何知道脚本内部以何种方式启动脚本以执行某些条件操作？

Answer 1

你关心它是使用 MPI 运行还是在一个 MPI rank 上运行？对于已编译的 MPI 代码，仅运行程序仍会在 MPI 下启动它，但只有一个级别；因此您可能只需初始化 MPI 并检查 MPI_COMM_WORLD 的大小。如果您在没有 MPI 的情况下运行，则可能是您试图避免初始化 MPI(或者甚至需要有可用的 MPI 实现)。如果是这样，您可能需要检查特定的环境变量，这些变量似乎是特定于实现的。对于 Open MPI，列表位于 http://www.open-mpi.org/faq/?category=running#mpi-environmental-variables。 .对于 MPICH，各种来源提到 PMI_RANK 和 PMI_SIZE 通常被设置； Microsoft MPI 文档表明它设置了这些。它们可能特定于特定的 MPICH 版本或配置。在 http://www.roguewave.com/portals/0/products/threadspotter/docs/2012.1/linux/manual_html/apas03.html 中有一个要检查的变量列表这也可能有用。