We want the output for rank 0 as well...

[simgrid.git] / src / smpi / smpi_bench.c

diff --git a/src/smpi/smpi_bench.c b/src/smpi/smpi_bench.c
index cc81f08..b9f0ab8 100644 (file)
--- a/src/smpi/smpi_bench.c
+++ b/src/smpi/smpi_bench.c
@@ -1,82 +1,78 @@
 #include "private.h"
-#include <string.h>
 
 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(smpi_bench, smpi,
                                 "Logging specific to SMPI (benchmarking)");
 
-void smpi_execute(double duration)
-{
-  smx_host_t host = SIMIX_host_self();
-  smx_mutex_t mutex = smpi_process_mutex();
-  smx_cond_t cond = smpi_process_cond();
+static void smpi_execute(double duration) {
+  smx_host_t host;
   smx_action_t action;
+  smx_mutex_t mutex;
+  smx_cond_t cond;
   e_surf_action_state_t state;
 
-  if (duration < 0.001)
-    return;
-  DEBUG1("Sleep for %f to handle real computation time",duration);
-  SIMIX_mutex_lock(mutex);
-
-  action =
-    SIMIX_action_execute(host, "execute",
-                         duration * smpi_global->reference_speed);
-
-  SIMIX_register_action_to_condition(action, cond);
-  for (state = SIMIX_action_get_state(action);
-       state == SURF_ACTION_READY ||
-       state == SURF_ACTION_RUNNING; state = SIMIX_action_get_state(action)
-    ) {
-    SIMIX_cond_wait(cond, mutex);
+  if(duration > 0.001) {
+    host = SIMIX_host_self();
+    mutex = SIMIX_mutex_init();
+    cond = SIMIX_cond_init();
+    DEBUG1("Sleep for %f to handle real computation time", duration);
+    duration *= xbt_cfg_get_double(_surf_cfg_set, "reference_speed");
+    action = SIMIX_action_execute(host, "computation", duration);
+    SIMIX_mutex_lock(mutex);
+    SIMIX_register_action_to_condition(action, cond);
+    for(state = SIMIX_action_get_state(action);
+        state == SURF_ACTION_READY ||
+        state == SURF_ACTION_RUNNING; state = SIMIX_action_get_state(action)) {
+      SIMIX_cond_wait(cond, mutex);
+    }
+    SIMIX_unregister_action_to_condition(action, cond);
+    SIMIX_mutex_unlock(mutex);
+    SIMIX_action_destroy(action);
+    SIMIX_cond_destroy(cond);
+    SIMIX_mutex_destroy(mutex);
   }
-  SIMIX_unregister_action_to_condition(action, cond);
-  SIMIX_action_destroy(action);
-
-  SIMIX_mutex_unlock(mutex);
-
-  return;
 }
 
-void smpi_start_timer()
-{
-  xbt_os_timer_start(smpi_global->timer);
-}
+void smpi_bench_begin(int rank, const char* mpi_call) {
+  double now = SIMIX_get_clock();
 
-double smpi_stop_timer()
-{
-  double duration;
-  xbt_os_timer_stop(smpi_global->timer);
-  duration = xbt_os_timer_elapsed(smpi_global->timer);
-  return duration;
+  if(mpi_call && rank >= 0 && xbt_cfg_get_int(_surf_cfg_set, "SMPE")) {
+    INFO3("SMPE: ts=%f rank=%d type=end et=%s", now, rank, mpi_call);
+  }
+  xbt_os_timer_start(smpi_process_timer());
 }
 
-void smpi_bench_begin()
-{
-  smpi_start_timer();
-}
+void smpi_bench_end(int rank, const char* mpi_call) {
+  xbt_os_timer_t timer = smpi_process_timer();
+  double now = SIMIX_get_clock();
 
-void smpi_bench_end()
-{
-  smpi_execute(smpi_stop_timer());
+  xbt_os_timer_stop(timer);
+  smpi_execute(xbt_os_timer_elapsed(timer));
+  if(mpi_call && rank >= 0 && xbt_cfg_get_int(_surf_cfg_set, "SMPE")) {
+    INFO3("SMPE: ts=%f rank=%d type=begin et=%s", now, rank, mpi_call);
+  }
 }
 
+/*
+TODO
 void smpi_do_once_1(const char *file, int line)
 {
   smpi_do_once_duration_node_t curr, prev;
+
   smpi_bench_end();
   SIMIX_mutex_lock(smpi_global->do_once_mutex);
   prev = NULL;
-  for (curr = smpi_global->do_once_duration_nodes;
-       NULL != curr && (strcmp(curr->file, file) || curr->line != line);
-       curr = curr->next) {
+  for(curr = smpi_global->do_once_duration_nodes;
+      NULL != curr && (strcmp(curr->file, file) || curr->line != line);
+      curr = curr->next) {
     prev = curr;
   }
-  if (NULL == curr) {
+  if(NULL == curr) {
     curr = xbt_new(s_smpi_do_once_duration_node_t, 1);
     curr->file = xbt_strdup(file);
     curr->line = line;
     curr->duration = -1;
     curr->next = NULL;
-    if (NULL == prev) {
+    if(NULL == prev) {
       smpi_global->do_once_duration_nodes = curr;
     } else {
       prev->next = curr;
@@ -88,7 +84,8 @@ void smpi_do_once_1(const char *file, int line)
 int smpi_do_once_2()
 {
   double duration = *(smpi_global->do_once_duration);
-  if (0 > duration) {
+
+  if(0 > duration) {
     smpi_start_timer();
     return 1;
   }
@@ -102,3 +99,4 @@ void smpi_do_once_3()
 {
   *(smpi_global->do_once_duration) = smpi_stop_timer();
 }
+*/