計算機システム MPIサンプル [6]

mpi-6.c

/*
 * 計算機システム MPIサンプル [6]
 * MPIのプロセスが実際にどのホストで実行されているかを調べる
 */

#include <stdio.h>
#include <stdlib.h>
#include <assert.h>
#include <mpi.h>

#include <sys/utsname.h>

void print_host() {
  int rank;
  struct utsname uts;

  MPI_Comm_rank(MPI_COMM_WORLD, &rank);
  uname(&uts);

  /*
   * 注意：環境により、行毎に表示されたり、
   * 行同士が混ざって表示されることがある.
   * プロセス同士での表示の順番は保証されない
   */
  printf("rank %d: %s\n", rank, uts.nodename);
}

int main(int argc, char **argv) {
  MPI_Init(&argc, &argv);

  print_host();

  MPI_Finalize();
}