fortran - 带 MPICH 的 nvprof

Question

我正在尝试分析 MPI/OpenACC Fortran 代码。我在这里找到了一个详细说明如何使用 MPI 运行 nvprof 的站点。给出的示例适用于 OpenMPI。但是，我仅限于 MPICH，我无法弄清楚等价物。有谁知道会是什么？

Answer 1

据我所知，nvprof 示例中唯一的 OpenMPI 特定部分是OMPI_COMM_WORLD_RANK用于获取每个等级的唯一文件名。根据这里的讨论，您可以使用其中一个PMI_RANK或PMI_ID代替。

在我的系统上，我有以下小程序

program env
  implicit none
  call system("echo $PMI_RANK")
end program env

我编译mpif90 env.f90 -o test并运行mpirun -np 2 ./test并获得

0
1

作为输出。所以我认为你可以替换-o output.%h.%p.%q{OMPI_COMM_WORLD_RANK}为-o output.%h.%p.%q{PMI_RANK}.

对于cray-mptmpi 库，我相信实际上是正确的变量ALPS_APP_PE。