X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/a06d371bdabe5d7c00f54b941d31068c01e6d756..3c0a14501628a82cc844761204585a496d67e06b:/teshsuite/smpi/compute3.c

diff --git a/teshsuite/smpi/compute3.c b/teshsuite/smpi/compute3.c
index f0445ba537..a501fbbd5b 100644
--- a/teshsuite/smpi/compute3.c
+++ b/teshsuite/smpi/compute3.c
@@ -4,46 +4,72 @@
 /* This program is free software; you can redistribute it and/or modify it
  * under the terms of the license (GNU LGPL) which comes with this package. */
 
-/* This example should be instructive to learn about SMPI_SAMPLE_LOCAL and 
+/* This example should be instructive to learn about SMPI_SAMPLE_LOCAL and
    SMPI_SAMPLE_GLOBAL macros for execution sampling */
 
 #include <stdio.h>
 #include <mpi.h>
 
+static double compute(double d0)
+{
+  double d = d0;
+  int j;
+  for (j = 0; j < 100 * 1000 * 1000; j++) { /* 100 kflop */
+    if (d < 100000) {
+      d = d * d;
+    } else {
+      d = 2;
+    }
+  }
+  return d;
+}
+
 int main(int argc, char *argv[])
 {
-  int i,j;
+  int verbose;
+  int i, n;
   double d;
   MPI_Init(&argc, &argv);
+  verbose = argc <= 1;
+  MPI_Comm_size(MPI_COMM_WORLD, &n);
   d = 2.0;
-  for (i=0;i<5;i++) {	
-     SMPI_SAMPLE_GLOBAL(3,-1)  { // I want no more than 3 benchs (thres<0)
-	fprintf(stderr,"[rank:%d] Run the first computation. It's globally benched, and I want no more than 3 benchmarks (thres<0)\n", smpi_process_index());
-    
-	for (j=0;j<100*1000*1000;j++) { // 100 kflop
-	   if (d < 100000) {
-	      d = d * d;
-	   } else {
-	      d = 2;
-	   }
-	}     
+  for (i = 0; i < 5; i++) {
+    /* I want no more than n + 1 benchs (thres < 0) */
+    SMPI_SAMPLE_GLOBAL(n + 1, -1) {
+      if (verbose)
+        fprintf(stderr, "(%12.6f) [rank:%d]", MPI_Wtime(), smpi_process_index());
+      else
+        fprintf(stderr, "(0)");
+      fprintf(stderr, " Run the first computation. It's globally benched, "
+              "and I want no more than %d benchmarks (thres<0)\n", n + 1);
+      d = compute(2.0);
      }
   }
 
-  for (i=0;i<5;i++) {	
-     SMPI_SAMPLE_LOCAL(0, 0.1) { // I want the standard error to go below 0.1 second. Two tests at least will be run (count is not >0)
-	fprintf(stderr,"[rank:%d] Run the first (locally benched) computation. It's locally benched, and I want the standard error to go below 0.1 second (count is not >0)\n", smpi_process_index());
-	for (j=0;j<100*1000*1000;j++) { // 100 kflop
-	   if (d < 100000) {
-	      d = d * d;
-	   } else {
-	      d = 2;
-	   }
-	}
+  n = 0;
+  for (i = 0; i < 5; i++) {
+    /* I want the standard error to go below 0.1 second.
+     * Two tests at least will be run (count is not > 0) */
+    SMPI_SAMPLE_LOCAL(0, 0.1) {
+      if (verbose || n++ < 2) {
+        if (verbose)
+          fprintf(stderr, "(%12.6f)", MPI_Wtime());
+        else
+          fprintf(stderr, "(1)");
+        fprintf(stderr,
+                " [rank:%d] Run the first (locally benched) computation. "
+                "It's locally benched, and I want the standard error to go "
+                "below 0.1 second (count is not >0)\n", smpi_process_index());
+      }
+      d = compute(d);
      }
   }
-   
-  fprintf(stderr,"[%d] The result of the computation is: %f\n", smpi_process_index(), d);
+
+  if (verbose)
+    fprintf(stderr, "(%12.6f) [rank:%d] The result of the computation is: %f\n",
+            MPI_Wtime(), smpi_process_index(), d);
+  else
+    fprintf(stderr, "(2) [rank:%d] Done.\n", smpi_process_index());
 
   MPI_Finalize();
   return 0;