Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
7309af15b9d600550989f1aeffcb82bee2a334f7
[simgrid.git] / teshsuite / smpi / macro-sample / macro-sample.c
1 /* Copyright (c) 2009-2019. 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     /* I want no more than n + 1 benchs (thres < 0) */
36   SMPI_SAMPLE_GLOBAL(int i = 0, i < 5, i++, n + 1, -1,
37       if (verbose)
38         fprintf(stderr, "(%12.6f) [rank:%d]", MPI_Wtime(), rank);
39       else
40         fprintf(stderr, "(0)");
41       fprintf(stderr, " Run the first computation. It's globally benched, "
42               "and I want no more than %d benchmarks (thres<0)\n", n + 1);
43       d = compute(2.0);
44   )
45
46   n = 0;
47   SMPI_SAMPLE_LOCAL (int i = 0, i < 5, i++,0, 0.1,
48       if (verbose || n < 2) {
49         n++;
50         if (verbose)
51           fprintf(stderr, "(%12.6f)", MPI_Wtime());
52         else
53           fprintf(stderr, "(1)");
54         fprintf(stderr,
55                 " [rank:%d] Run the second (locally benched) computation. It's locally benched, and I want the "
56                 "standard error to go below 0.1 second (count is not >0)\n", rank);
57       }
58       d = compute(d);
59   )
60
61   if (verbose)
62     fprintf(stderr, "(%12.6f) [rank:%d] The result of the computation is: %f\n", MPI_Wtime(), rank, d);
63   else
64     fprintf(stderr, "(2) [rank:%d] Done.\n", rank);
65
66   MPI_Finalize();
67   return 0;
68 }