teshsuite/smpi/macro-sample/macro-sample.c

   1 /* Copyright (c) 2009-2018. The SimGrid Team.
   2  * All rights reserved.                                                     */
   3
   4 /* This program is free software; you can redistribute it and/or modify it
   5  * under the terms of the license (GNU LGPL) which comes with this package. */
   6
   7 /* This example should be instructive to learn about SMPI_SAMPLE_LOCAL and
   8    SMPI_SAMPLE_GLOBAL macros for execution sampling */
   9
  10 #include <stdio.h>
  11 #include <mpi.h>
  12
  13 static double compute(double d0)
  14 {
  15   double d = d0;
  16   for (int j = 0; j < 100 * 1000 * 1000; j++) { /* 100 kflop */
  17     if (d < 100000) {
  18       d = d * d;
  19     } else {
  20       d = 2;
  21     }
  22   }
  23   return d;
  24 }
  25
  26 int main(int argc, char *argv[])
  27 {
  28   int n;
  29   int rank;
  30   MPI_Init(&argc, &argv);
  31   int verbose = argc <= 1;
  32   MPI_Comm_size(MPI_COMM_WORLD, &n);
  33   MPI_Comm_rank(MPI_COMM_WORLD, &rank);
  34   double d = 2.0;
  35   for (int i = 0; i < 5; i++) {
  36     /* I want no more than n + 1 benchs (thres < 0) */
  37     SMPI_SAMPLE_GLOBAL(n + 1, -1) {
  38       if (verbose)
  39         fprintf(stderr, "(%12.6f) [rank:%d]", MPI_Wtime(), rank);
  40       else
  41         fprintf(stderr, "(0)");
  42       fprintf(stderr, " Run the first computation. It's globally benched, "
  43               "and I want no more than %d benchmarks (thres<0)\n", n + 1);
  44       d = compute(2.0);
  45      }
  46   }
  47
  48   n = 0;
  49   for (int i = 0; i < 5; i++) {
  50     /* I want the standard error to go below 0.1 second.
  51      * Two tests at least will be run (count is not > 0) */
  52     SMPI_SAMPLE_LOCAL(0, 0.1) {
  53       if (verbose || n < 2) {
  54         n++;
  55         if (verbose)
  56           fprintf(stderr, "(%12.6f)", MPI_Wtime());
  57         else
  58           fprintf(stderr, "(1)");
  59         fprintf(stderr,
  60                 " [rank:%d] Run the first (locally benched) computation. It's locally benched, and I want the "
  61                 "standard error to go below 0.1 second (count is not >0)\n", rank);
  62       }
  63       d = compute(d);
  64      }
  65   }
  66
  67   if (verbose)
  68     fprintf(stderr, "(%12.6f) [rank:%d] The result of the computation is: %f\n", MPI_Wtime(), rank, d);
  69   else
  70     fprintf(stderr, "(2) [rank:%d] Done.\n", rank);
  71
  72   MPI_Finalize();
  73   return 0;
  74 }