X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/b110f915460bff6576e1fc6d3a4f391992852714..3c072dea92bbf0e4df46b1d8d56cd62e68883b33:/src/smpi/smpi_sender.c

diff --git a/src/smpi/smpi_sender.c b/src/smpi/smpi_sender.c
index 0d2c433ab6..eb5e727d58 100644
--- a/src/smpi/smpi_sender.c
+++ b/src/smpi/smpi_sender.c
@@ -3,8 +3,9 @@
 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(smpi_sender, smpi,
                                 "Logging specific to SMPI (sender)");
 
-int smpi_sender(int argc, char **argv)
+int smpi_sender(int argc, char *argv[])
 {
+  smpi_process_data_t mydata = SIMIX_process_get_data(SIMIX_process_self());
   smx_process_t self;
   smx_host_t shost;
 
@@ -12,8 +13,6 @@ int smpi_sender(int argc, char **argv)
 
   xbt_fifo_t request_queue;
 
-  int running_hosts_count;
-
   smpi_mpi_request_t request;
 
   smx_host_t dhost;
@@ -26,25 +25,19 @@ int smpi_sender(int argc, char **argv)
 
   int dindex;
 
-  smx_process_t receiver_process;
-
   self = SIMIX_process_self();
   shost = SIMIX_host_self();
 
-  index = smpi_host_index();
-
-  request_queue = smpi_global->pending_send_request_queues[index];
+  index = mydata->index;
 
-  smpi_global->sender_processes[index] = self;
+  DEBUG0("Up and running");
 
-  do {
+  request_queue = mydata->pending_send_request_queue;
 
+  while (1) {
     request = xbt_fifo_shift(request_queue);
 
-    if (NULL == request) {
-      SIMIX_process_suspend(self);
-    } else {
-
+    if (NULL != request) {
       message = xbt_mallocator_get(smpi_global->message_mallocator);
 
       SIMIX_mutex_lock(request->mutex);
@@ -58,8 +51,12 @@ int smpi_sender(int argc, char **argv)
              request->datatype->size * request->count);
 
       dindex = request->comm->rank_to_index_map[request->dst];
-      dhost = smpi_global->hosts[dindex];
+      smpi_process_data_t remote_process =
+        SIMIX_process_get_data(smpi_global->main_processes[dindex]);
+      dhost = SIMIX_process_get_host(smpi_global->main_processes[dindex]);
 
+      DEBUG4("handle send request %p to %s (req_dst=%d,req_tag=%d)",
+                      request,SIMIX_host_get_name(dhost),request->dst,message->tag);
       message->forward = (request->forward - 1) / 2;
       request->forward = request->forward / 2;
 
@@ -68,6 +65,8 @@ int smpi_sender(int argc, char **argv)
           (request->dst + message->forward + 1) % request->comm->size;
         xbt_fifo_push(request_queue, request);
       } else {
+        DEBUG4("DONE Handling send request %p to %s (req_dst=%d,req_tag=%d)",
+                        request, SIMIX_host_get_name(dhost),request->dst,message->tag);
         request->completed = 1;
       }
 
@@ -86,7 +85,7 @@ int smpi_sender(int argc, char **argv)
         SIMIX_cond_wait(request->cond, request->mutex);
       }
 
-      xbt_fifo_push(smpi_global->received_message_queues[dindex], message);
+      xbt_fifo_push(remote_process->received_message_queue, message);
 
       SIMIX_unregister_action_to_condition(action, request->cond);
       SIMIX_action_destroy(action);
@@ -94,18 +93,18 @@ int smpi_sender(int argc, char **argv)
       SIMIX_mutex_unlock(request->mutex);
 
       // wake up receiver if necessary
-      receiver_process = smpi_global->receiver_processes[dindex];
-      if (SIMIX_process_is_suspended(receiver_process)) {
-        SIMIX_process_resume(receiver_process);
-      }
+      SIMIX_process_resume(remote_process->receiver);
 
+    } else if (mydata->finalize > 0) {  /* main wants me to die and nothing to do */
+      DEBUG0("===Main wants me to die and I'm done. Bye, guys.===");
+      mydata->finalize--;
+      SIMIX_cond_signal(mydata->cond);
+      return 0;
+    } else {
+      DEBUG0("Nothing to do. Let's get a nap");
+      SIMIX_process_suspend(self);
+      DEBUG0("===Uh? Someone called me?===");
     }
-
-    SIMIX_mutex_lock(smpi_global->running_hosts_count_mutex);
-    running_hosts_count = smpi_global->running_hosts_count;
-    SIMIX_mutex_unlock(smpi_global->running_hosts_count_mutex);
-
-  } while (0 < running_hosts_count);
-
+  }
   return 0;
 }