X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/773e09f91e02cdd7e30e221f2bd6f39e97896b16..fccf53a380a9a6de6c8bd9eafb182a91d79e1d3c:/src/smpi/smpi_bench.c

diff --git a/src/smpi/smpi_bench.c b/src/smpi/smpi_bench.c
index b64c2f7a3a..e531e85913 100644
--- a/src/smpi/smpi_bench.c
+++ b/src/smpi/smpi_bench.c
@@ -1,72 +1,276 @@
+/* Copyright (c) 2007, 2009, 2010. The SimGrid Team.
+ * All rights reserved.                                                     */
+
+/* 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. */
+
 #include "private.h"
+#include "xbt/dict.h"
+#include "xbt/sysdep.h"
+#include "xbt/ex.h"
+#include "surf/surf.h"
+
+XBT_LOG_NEW_DEFAULT_SUBCATEGORY(smpi_bench, smpi,
+                                "Logging specific to SMPI (benchmarking)");
+
+xbt_dict_t allocs = NULL;       /* Allocated on first use */
+xbt_dict_t samples = NULL;      /* Allocated on first use */
+xbt_dict_t calls = NULL;        /* Allocated on first use */
+
+typedef struct {
+  int count;
+  char data[];
+} shared_data_t;
+
+typedef struct {
+  int count;
+  double sum;
+  double sum_pow2;
+  double mean;
+  double relstderr;
+  int iters;
+  double threshold;
+  int started;
+} local_data_t;
+
+void smpi_bench_destroy(void)
+{
+  if (allocs) {
+    xbt_dict_free(&allocs);
+  }
+  if (samples) {
+    xbt_dict_free(&samples);
+  }
+  if(calls) {
+    xbt_dict_free(&calls);
+  }
+}
+
+static void smpi_execute_flops(double flops)
+{
+  smx_action_t action;
+  smx_host_t host;
+  host = SIMIX_host_self();
+
+  XBT_DEBUG("Handle real computation time: %f flops", flops);
+  action = SIMIX_req_host_execute("computation", host, flops, 1);
+#ifdef HAVE_TRACING
+  SIMIX_req_set_category (action, TRACE_internal_smpi_get_category());
+#endif
+  SIMIX_req_host_execution_wait(action);
+}
+
+static void smpi_execute(double duration)
+{
+  if (duration >= xbt_cfg_get_double(_surf_cfg_set, "smpi/cpu_threshold")) {
+    XBT_DEBUG("Sleep for %f to handle real computation time", duration);
+    smpi_execute_flops(duration *
+                       xbt_cfg_get_double(_surf_cfg_set,
+                                          "smpi/running_power"));
+  }
+}
+
+void smpi_bench_begin(void)
+{
+  xbt_os_timer_start(smpi_process_timer());
+}
 
-void smpi_execute(double duration) {
-        smx_host_t host = SIMIX_host_self();
-        smx_action_t action;
+void smpi_bench_end(void)
+{
+  xbt_os_timer_t timer = smpi_process_timer();
 
-	SIMIX_mutex_lock(smpi_global->execute_mutex);
+  xbt_os_timer_stop(timer);
+  smpi_execute(xbt_os_timer_elapsed(timer));
+}
 
-	action = SIMIX_action_execute(host, "computation", duration * SMPI_DEFAULT_SPEED);
+unsigned int smpi_sleep(unsigned int secs)
+{
+  smpi_execute((double) secs);
+  return secs;
+}
 
-        SIMIX_register_action_to_condition(action, smpi_global->execute_cond);
-        SIMIX_cond_wait(smpi_global->execute_cond, smpi_global->execute_mutex);
-        SIMIX_unregister_action_to_condition(action, smpi_global->execute_cond);
-        SIMIX_action_destroy(action);
+int smpi_gettimeofday(struct timeval *tv, struct timezone *tz)
+{
+  double now = SIMIX_get_clock();
 
-        SIMIX_mutex_unlock(smpi_global->execute_mutex);
+  if (tv) {
+    tv->tv_sec = (time_t) now;
+    tv->tv_usec = (suseconds_t) (now * 1e6);
+  }
+  return 0;
+}
 
-	return;
+static char *sample_location(int global, const char *file, int line)
+{
+  if (global) {
+    return bprintf("%s:%d", file, line);
+  } else {
+    return bprintf("%s:%d:%d", file, line, smpi_process_index());
+  }
 }
 
-void smpi_bench_begin()
+int smpi_sample_1(int global, const char *file, int line, int iters, double threshold)
 {
-	SIMIX_mutex_lock(smpi_global->timer_mutex);
-	xbt_os_timer_start(smpi_global->timer);
-	return;
+  char *loc = sample_location(global, file, line);
+  local_data_t *data;
+
+  smpi_bench_end();     /* Take time from previous MPI call into account */
+  if (!samples) {
+    samples = xbt_dict_new();
+  }
+  data = xbt_dict_get_or_null(samples, loc);
+  if (!data) {
+    data = (local_data_t *) xbt_new(local_data_t, 1);
+    data->count = 0;
+    data->sum = 0.0;
+    data->sum_pow2 = 0.0;
+    data->iters = iters;
+    data->threshold = threshold;
+    data->started = 0;
+    xbt_dict_set(samples, loc, data, &free);
+    return 0;
+  }
+  free(loc);
+  return 1;
 }
 
-double smpi_bench_end()
+int smpi_sample_2(int global, const char *file, int line)
 {
-	double duration;
+  char *loc = sample_location(global, file, line);
+  local_data_t *data;
 
-	xbt_os_timer_stop(smpi_global->timer);
+  xbt_assert(samples, "You did something very inconsistent, didn't you?");
+  data = xbt_dict_get_or_null(samples, loc);
+  if (!data) {
+    xbt_assert(data, "Please, do thing in order");
+  }
+  if (!data->started) {
+    if ((data->iters > 0 && data->count >= data->iters)
+        || (data->count > 1 && data->threshold > 0.0 && data->relstderr <= data->threshold)) {
+      XBT_DEBUG("Perform some wait of %f", data->mean);
+      smpi_execute(data->mean);
+    } else {
+      data->started = 1;
+      data->count++;
+    }
+  } else {
+    data->started = 0;
+  }
+  free(loc);
+  smpi_bench_begin();
+  smpi_process_simulated_start();
+  return data->started;
+}
 
-	duration = xbt_os_timer_elapsed(smpi_global->timer);
+void smpi_sample_3(int global, const char *file, int line)
+{
+  char *loc = sample_location(global, file, line);
+  local_data_t *data;
+  double sample, n;
 
-	SIMIX_mutex_unlock(smpi_global->timer_mutex);
+  xbt_assert(samples, "You did something very inconsistent, didn't you?");
+  data = xbt_dict_get_or_null(samples, loc);
+  smpi_bench_end();
+  if(data && data->started && data->count < data->iters) {
+    sample = smpi_process_simulated_elapsed();
+    data->sum += sample;
+    data->sum_pow2 += sample * sample;
+    n = (double)data->count;
+    data->mean = data->sum / n;
+    data->relstderr = sqrt((data->sum_pow2 / n - data->mean * data->mean) / n) / data->mean;
+    XBT_DEBUG("Average mean after %d steps is %f, relative standard error is %f (sample was %f)", data->count,
+           data->mean, data->relstderr, sample);
+  }
+  free(loc);
+}
 
-	smpi_execute(duration);
+void smpi_sample_flops(double flops)
+{
+  smpi_execute_flops(flops);
+}
+
+void *smpi_shared_malloc(size_t size, const char *file, int line)
+{
+  char *loc = bprintf("%s:%d:%zu", file, line, size);
+  shared_data_t *data;
 
-	return duration;
+  if (!allocs) {
+    allocs = xbt_dict_new();
+  }
+  data = xbt_dict_get_or_null(allocs, loc);
+  if (!data) {
+    data = (shared_data_t *) xbt_malloc0(sizeof(int) + size);
+    data->count = 1;
+    xbt_dict_set(allocs, loc, data, &free);
+  } else {
+    data->count++;
+  }
+  free(loc);
+  return data->data;
 }
 
+void smpi_shared_free(void *ptr)
+{
+  shared_data_t *data = (shared_data_t *) ((int *) ptr - 1);
+  char *loc;
 
-void smpi_bench_skip() {
-	double duration = smpi_global->times[0];
-	smpi_execute(duration);
-	return;
+  if (!allocs) {
+    XBT_WARN("Cannot free: nothing was allocated");
+    return;
+  }
+  loc = xbt_dict_get_key(allocs, data);
+  if (!loc) {
+    XBT_WARN("Cannot free: %p was not shared-allocated by SMPI", ptr);
+    return;
+  }
+  data->count--;
+  if (data->count <= 0) {
+    xbt_dict_remove(allocs, loc);
+  }
 }
 
-void smpi_do_once_1() {
-	smpi_bench_end();
-	SIMIX_mutex_lock(smpi_global->times_mutex);
-	if (0 < smpi_global->times[0]) {
-		smpi_bench_skip();
-	}
-	return;
+int smpi_shared_known_call(const char* func, const char* input) {
+   char* loc = bprintf("%s:%s", func, input);
+   xbt_ex_t ex;
+   int known;
+
+   if(!calls) {
+      calls = xbt_dict_new();
+   }
+   TRY {
+      xbt_dict_get(calls, loc); /* Succeed or throw */
+      known = 1;
+   } CATCH(ex) {
+      if(ex.category == not_found_error) {
+         known = 0;
+         xbt_ex_free(ex);
+      } else {
+         RETHROW;
+      }
+   }
+   free(loc);
+   return known;
 }
 
-int smpi_do_once_2() {
-	int retval = 1;
-	if (0 < smpi_global->times[0]) {
-		SIMIX_mutex_unlock(smpi_global->times_mutex);
-		retval = 0;
-	}
-	smpi_bench_begin();
-	return retval;
+void* smpi_shared_get_call(const char* func, const char* input) {
+   char* loc = bprintf("%s:%s", func, input);
+   void* data;
+
+   if(!calls) {
+      calls = xbt_dict_new();
+   }
+   data = xbt_dict_get(calls, loc);
+   free(loc);
+   return data;
 }
 
-void smpi_do_once_3() {
-	smpi_global->times[0] = smpi_bench_end();
-	return;
+void* smpi_shared_set_call(const char* func, const char* input, void* data) {
+   char* loc = bprintf("%s:%s", func, input);
+
+   if(!calls) {
+      calls = xbt_dict_new();
+   }
+   xbt_dict_set(calls, loc, data, NULL);
+   free(loc);
+   return data;
 }