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

   1 /* Copyright (c) 2009-2014. 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   MPI_Init(&argc, &argv);
  30   int verbose = argc <= 1;
  31   MPI_Comm_size(MPI_COMM_WORLD, &n);
  32   double d = 2.0;
  33   for (int i = 0; i < 5; i++) {
  34     /* I want no more than n + 1 benchs (thres < 0) */
  35     SMPI_SAMPLE_GLOBAL(n + 1, -1) {
  36       if (verbose)
  37         fprintf(stderr, "(%12.6f) [rank:%d]", MPI_Wtime(), smpi_process_index());
  38       else
  39         fprintf(stderr, "(0)");
  40       fprintf(stderr, " Run the first computation. It's globally benched, "
  41               "and I want no more than %d benchmarks (thres<0)\n", n + 1);
  42       d = compute(2.0);
  43      }
  44   }
  45
  46   n = 0;
  47   for (int i = 0; i < 5; i++) {
  48     /* I want the standard error to go below 0.1 second.
  49      * Two tests at least will be run (count is not > 0) */
  50     SMPI_SAMPLE_LOCAL(0, 0.1) {
  51       if (verbose || n < 2) {
  52         n++;
  53         if (verbose)
  54           fprintf(stderr, "(%12.6f)", MPI_Wtime());
  55         else
  56           fprintf(stderr, "(1)");
  57         fprintf(stderr,
  58                 " [rank:%d] Run the first (locally benched) computation. It's locally benched, and I want the "
  59                 "standard error to go below 0.1 second (count is not >0)\n", smpi_process_index());
  60       }
  61       d = compute(d);
  62      }
  63   }
  64
  65   if (verbose)
  66     fprintf(stderr, "(%12.6f) [rank:%d] The result of the computation is: %f\n", MPI_Wtime(), smpi_process_index(), d);
  67   else
  68     fprintf(stderr, "(2) [rank:%d] Done.\n", smpi_process_index());
  69
  70   MPI_Finalize();
  71   return 0;
  72 }