X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/04b83ff67daf1924db99ae07054ac5d058ed4a19..1372e35c79dba117ae1b592394821a3f716a3f13:/src/smpi/smpi_sender.c

diff --git a/src/smpi/smpi_sender.c b/src/smpi/smpi_sender.c
index 57150580b8..9098aa2c94 100644
--- a/src/smpi/smpi_sender.c
+++ b/src/smpi/smpi_sender.c
@@ -1,108 +1,99 @@
 #include "private.h"
 
-int smpi_sender(int argc, char **argv)
-{
-	smx_process_t self;
-	smx_host_t shost;
-	int index;
+XBT_LOG_NEW_DEFAULT_SUBCATEGORY(smpi_sender, smpi,
+                                "Logging specific to SMPI (sender)");
 
-	xbt_fifo_t request_queue;
-	smx_mutex_t request_queue_mutex;
+int smpi_sender(int argc,char*argv[]) {
+	smpi_host_data_t mydata = SIMIX_process_get_data(SIMIX_process_self());
+  smx_process_t self;
+  smx_host_t shost;
 
-	int running_hosts_count;
+  int index;
 
-	smpi_mpi_request_t request;
+  xbt_fifo_t request_queue;
 
-	smx_host_t dhost;
+  smpi_mpi_request_t request;
 
-	char communication[] = "communication";
-	smx_action_t action;
+  smx_host_t dhost;
 
-	smpi_received_message_t message;
+  smx_action_t action;
 
-	int dindex;
+  e_surf_action_state_t state;
 
-	smx_process_t receiver_process;
+  smpi_received_message_t message;
 
-	self  = SIMIX_process_self();
-	shost = SIMIX_host_self();
+  int dindex;
 
-	// make sure root is done before own initialization
-	SIMIX_mutex_lock(smpi_global->start_stop_mutex);
-	if (!smpi_global->root_ready) {
-		SIMIX_cond_wait(smpi_global->start_stop_cond, smpi_global->start_stop_mutex);
-	}
-	SIMIX_mutex_unlock(smpi_global->start_stop_mutex);
+  self = SIMIX_process_self();
+  shost = SIMIX_host_self();
 
-	index = smpi_host_index();
+  index = mydata->index;
 
-	request_queue       = smpi_global->pending_send_request_queues[index];
-	request_queue_mutex = smpi_global->pending_send_request_queues_mutexes[index];
+  request_queue = smpi_global->pending_send_request_queues[index];
 
-	smpi_global->sender_processes[index] = self;
+  while (1) {
+    request = xbt_fifo_shift(request_queue);
 
-	// wait for all nodes to signal initializatin complete
-	SIMIX_mutex_lock(smpi_global->start_stop_mutex);
-	smpi_global->ready_process_count++;
-	if (smpi_global->ready_process_count < 3 * smpi_global->host_count) {
-		SIMIX_cond_wait(smpi_global->start_stop_cond, smpi_global->start_stop_mutex);
-	} else {
-		SIMIX_cond_broadcast(smpi_global->start_stop_cond);
-	}
-	SIMIX_mutex_unlock(smpi_global->start_stop_mutex);
+    if (NULL != request) {
+      message = xbt_mallocator_get(smpi_global->message_mallocator);
 
-	do {
+      SIMIX_mutex_lock(request->mutex);
 
-		SIMIX_mutex_lock(request_queue_mutex);
-		request = xbt_fifo_shift(request_queue);
-		SIMIX_mutex_unlock(request_queue_mutex);
+      message->comm = request->comm;
+      message->src = request->comm->index_to_rank_map[index];
+      message->tag = request->tag;
+      message->data = request->data;
+      message->buf = xbt_malloc(request->datatype->size * request->count);
+      memcpy(message->buf, request->buf,
+             request->datatype->size * request->count);
 
-		if (NULL == request) {
-			SIMIX_process_suspend(self);
-		} else {
+      dindex = request->comm->rank_to_index_map[request->dst];
+      dhost = smpi_global->hosts[dindex];
 
-			message       = xbt_mallocator_get(smpi_global->message_mallocator);
+      message->forward = (request->forward - 1) / 2;
+      request->forward = request->forward / 2;
 
-			SIMIX_mutex_lock(request->mutex);
+      if (0 < request->forward) {
+        request->dst =
+          (request->dst + message->forward + 1) % request->comm->size;
+        xbt_fifo_push(request_queue, request);
+      } else {
+        request->completed = 1;
+      }
 
-			message->comm = request->comm;
-			message->src  = smpi_mpi_comm_rank(request->comm);
-			message->tag  = request->tag;
-			message->buf  = xbt_malloc(request->datatype->size * request->count);
-			memcpy(message->buf, request->buf, request->datatype->size * request->count);
+      action =
+        SIMIX_action_communicate(shost, dhost, "communication",
+                                 request->datatype->size * request->count,
+                                 -1.0);
 
-			dindex = request->comm->rank_to_index_map[request->dst];
-			dhost  = smpi_global->hosts[dindex];
+      SIMIX_register_action_to_condition(action, request->cond);
 
-			SIMIX_mutex_lock(smpi_global->received_message_queues_mutexes[dindex]);
-			xbt_fifo_push(smpi_global->received_message_queues[dindex], message);
-			SIMIX_mutex_unlock(smpi_global->received_message_queues_mutexes[dindex]);
+      for (state = SIMIX_action_get_state(action);
+           state == SURF_ACTION_READY ||
+           state == SURF_ACTION_RUNNING;
+           state = SIMIX_action_get_state(action)
+        ) {
+        SIMIX_cond_wait(request->cond, request->mutex);
+      }
 
-			request->completed = 1;
+      xbt_fifo_push(smpi_global->received_message_queues[dindex], message);
 
-			action = SIMIX_action_communicate(shost, dhost, communication, request->datatype->size * request->count, -1.0);
+      SIMIX_unregister_action_to_condition(action, request->cond);
+      SIMIX_action_destroy(action);
 
-			SIMIX_register_action_to_condition(action, request->cond);
-			SIMIX_cond_wait(request->cond, request->mutex);
-			SIMIX_unregister_action_to_condition(action, request->cond);
+      SIMIX_mutex_unlock(request->mutex);
 
-			SIMIX_mutex_unlock(request->mutex);
+      // wake up receiver if necessary
+      smpi_host_data_t remote_host = SIMIX_host_get_data(SIMIX_process_get_host(smpi_global->main_processes[dindex]));
+      SIMIX_process_resume(remote_host->receiver);
 
-			//SIMIX_action_destroy(action);
-
-			// 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_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;
+    } else if (mydata->finalize>0) { /* main wants me to die and nothing to do */
+    	mydata->finalize--;
+    	SIMIX_cond_signal(mydata->cond);
+    	return 0;
+    } else {
+    	SIMIX_process_suspend(self);
+    }
+  }
+  return 0;
 }