- SIMIX_mutex_lock(smpi_global->start_stop_mutex);
- smpi_global->ready_process_count--;
- SIMIX_mutex_unlock(smpi_global->start_stop_mutex);
-
- if (0 >= i) {
-
- // wake up senders/receivers
- for (i = 0; i < smpi_mpi_global->mpi_comm_world->size; i++) {
- if (SIMIX_process_is_suspended(smpi_global->sender_processes[i])) {
- SIMIX_process_resume(smpi_global->sender_processes[i]);
- }
- if (SIMIX_process_is_suspended(smpi_global->receiver_processes[i])) {
- SIMIX_process_resume(smpi_global->receiver_processes[i]);
- }
- }
-
- // wait for senders/receivers to exit...
- SIMIX_mutex_lock(smpi_global->start_stop_mutex);
- if (smpi_global->ready_process_count > 0) {
- SIMIX_cond_wait(smpi_global->start_stop_cond, smpi_global->start_stop_mutex);
- }
- SIMIX_mutex_unlock(smpi_global->start_stop_mutex);
-
- SIMIX_mutex_destroy(smpi_mpi_global->mpi_comm_world->barrier_mutex);
- SIMIX_cond_destroy(smpi_mpi_global->mpi_comm_world->barrier_cond);
- xbt_free(smpi_mpi_global->mpi_comm_world->processes);
- xbt_free(smpi_mpi_global->mpi_comm_world);
-
- xbt_free(smpi_mpi_global->mpi_byte);
- xbt_free(smpi_mpi_global->mpi_int);
- xbt_free(smpi_mpi_global->mpi_double);
-
- xbt_free(smpi_mpi_global->mpi_land);
- xbt_free(smpi_mpi_global->mpi_sum);
-
- xbt_free(smpi_mpi_global);
- }
-
-}
-
-// FIXME: could cause trouble with multithreaded procs on same host...
-void smpi_bench_begin()
-{
- int rank = smpi_mpi_comm_rank_self(smpi_mpi_global->mpi_comm_world);
- SIMIX_mutex_lock(smpi_global->timers_mutexes[rank]);
- xbt_os_timer_start(smpi_global->timers[rank]);
- return;
-}
-
-void smpi_bench_end()
-{
- int rank = smpi_mpi_comm_rank_self(smpi_mpi_global->mpi_comm_world);
- double duration;
- smx_host_t host;
- smx_action_t compute_action;
- smx_mutex_t mutex;
- smx_cond_t cond;
-
- xbt_os_timer_stop(smpi_global->timers[rank]);
-
- duration = xbt_os_timer_elapsed(smpi_global->timers[rank]);
- SIMIX_mutex_unlock(smpi_global->timers_mutexes[rank]);
-
- host = smpi_mpi_global->mpi_comm_world->hosts[rank];
- compute_action = SIMIX_action_execute(host, NULL, duration * SMPI_DEFAULT_SPEED);
- mutex = SIMIX_mutex_init();
- cond = SIMIX_cond_init();
-
- SIMIX_mutex_lock(mutex);
- SIMIX_register_action_to_condition(compute_action, cond);
- SIMIX_cond_wait(cond, mutex);
- //SIMIX_unregister_action_to_condition(compute_action, cond);
- SIMIX_mutex_unlock(mutex);
-
- SIMIX_mutex_destroy(mutex);
- SIMIX_cond_destroy(cond);
-
- // FIXME: check for success/failure?
-
- return;
-}
-
-void smpi_barrier(smpi_mpi_communicator_t *comm)
-{
-
- SIMIX_mutex_lock(comm->barrier_mutex);
- if(++comm->barrier_count < comm->size) {
- SIMIX_cond_wait(comm->barrier_cond, comm->barrier_mutex);
- } else {
- comm->barrier_count = 0;
- SIMIX_cond_broadcast(comm->barrier_cond);
- }
- SIMIX_mutex_unlock(comm->barrier_mutex);
-
- return;
-}
-
-// FIXME: smarter algorithm...
-int smpi_comm_rank(smpi_mpi_communicator_t *comm, smx_host_t host)
-{
- int i;
- for(i = 0; i < comm->size && host != comm->hosts[i]; i++);
- if (i >= comm->size) i = -1;
- return i;
-}
-
-int smpi_create_request(void *buf, int count, smpi_mpi_datatype_t *datatype,
- int src, int dst, int tag, smpi_mpi_communicator_t *comm, smpi_mpi_request_t **request)
-{
- int retval = MPI_SUCCESS;
-
- *request = NULL;
-
- if (0 > count) {
- retval = MPI_ERR_COUNT;
- } else if (NULL == buf) {
- retval = MPI_ERR_INTERN;
- } else if (NULL == datatype) {
- retval = MPI_ERR_TYPE;
- } else if (NULL == comm) {
- retval = MPI_ERR_COMM;
- } else if (MPI_ANY_SOURCE != src && (0 > src || comm->size <= src)) {
- retval = MPI_ERR_RANK;
- } else if (0 > dst || comm->size <= dst) {
- retval = MPI_ERR_RANK;
- } else if (0 > tag) {
- retval = MPI_ERR_TAG;
- } else {
- *request = xbt_mallocator_get(smpi_global->request_mallocator);
- (*request)->comm = comm;
- (*request)->src = src;
- (*request)->dst = dst;
- (*request)->tag = tag;
- (*request)->buf = buf;
- (*request)->count = count;
- (*request)->datatype = datatype;
- }
- return retval;
-}
-
-int smpi_isend(smpi_mpi_request_t *request)
-{
- int retval = MPI_SUCCESS;
- int rank = smpi_mpi_comm_rank_self(smpi_mpi_global->mpi_comm_world);
-
- if (NULL != request) {
- SIMIX_mutex_lock(smpi_global->pending_send_request_queues_mutexes[rank]);
- xbt_fifo_push(smpi_global->pending_send_request_queues[rank], request);
- SIMIX_mutex_unlock(smpi_global->pending_send_request_queues_mutexes[rank]);
- }
-
- if (SIMIX_process_is_suspended(smpi_global->sender_processes[rank])) {
- SIMIX_process_resume(smpi_global->sender_processes[rank]);
- }
-
- return retval;
-}
-
-int smpi_irecv(smpi_mpi_request_t *request)
-{
- int retval = MPI_SUCCESS;
- int rank = smpi_mpi_comm_rank_self(smpi_mpi_global->mpi_comm_world);
-
- if (NULL != request) {
- SIMIX_mutex_lock(smpi_global->pending_recv_request_queues_mutexes[rank]);
- xbt_fifo_push(smpi_global->pending_recv_request_queues[rank], request);
- SIMIX_mutex_unlock(smpi_global->pending_recv_request_queues_mutexes[rank]);
- }
-
- if (SIMIX_process_is_suspended(smpi_global->receiver_processes[rank])) {
- SIMIX_process_resume(smpi_global->receiver_processes[rank]);
- }
-
- return retval;
-}
-
-void smpi_wait(smpi_mpi_request_t *request, smpi_mpi_status_t *status)
-{
- if (NULL != request) {
- SIMIX_mutex_lock(request->mutex);
- if (!request->completed) {
- SIMIX_cond_wait(request->cond, request->mutex);
- }
- if (NULL != status) {
- status->MPI_SOURCE = request->src;
- }
- SIMIX_mutex_unlock(request->mutex);
- }
-}
-
-// FIXME: move into own file
-int smpi_gettimeofday(struct timeval *tv, struct timezone *tz)
-{
- double now;
- int retval = 0;
- smpi_bench_end();
- if (NULL == tv) {
- retval = -1;
- } else {
- now = SIMIX_get_clock();
- tv->tv_sec = now;
- tv->tv_usec = ((now - (double)tv->tv_sec) * 1000000.0);
- }
- smpi_bench_begin();
- return retval;
-}
-
-unsigned int smpi_sleep(unsigned int seconds)
-{
- smx_mutex_t mutex;
- smx_cond_t cond;
- smx_host_t host;
- smx_action_t sleep_action;
-
- smpi_bench_end();
- host = SIMIX_host_self();
- sleep_action = SIMIX_action_sleep(host, seconds);
- mutex = SIMIX_mutex_init();
- cond = SIMIX_cond_init();
-
- SIMIX_mutex_lock(mutex);
- SIMIX_register_action_to_condition(sleep_action, cond);
- SIMIX_cond_wait(cond, mutex);
- //SIMIX_unregister_action_to_condition(sleep_action, cond);
- SIMIX_mutex_unlock(mutex);
-
- SIMIX_mutex_destroy(mutex);
- SIMIX_cond_destroy(cond);
-
- // FIXME: check for success/failure?
-
- smpi_bench_begin();
- return 0;
-}
-
-void smpi_exit(int status)
-{
- smpi_bench_end();
- SIMIX_mutex_lock(smpi_global->running_hosts_count_mutex);
- smpi_global->running_hosts_count--;
- SIMIX_mutex_unlock(smpi_global->running_hosts_count_mutex);
- SIMIX_process_kill(SIMIX_process_self());
- return;
+void smpi_mpi_scan(void* sendbuf, void* recvbuf, int count, MPI_Datatype datatype, MPI_Op op, MPI_Comm comm) {
+ int system_tag = 666;
+ int rank, size, other, index, datasize;
+ int total;
+ MPI_Request* requests;
+ void** tmpbufs;
+
+ rank = smpi_comm_rank(comm);
+ size = smpi_comm_size(comm);
+ datasize = smpi_datatype_size(datatype);
+ // Local copy from self
+ memcpy(recvbuf, sendbuf, count * datasize * sizeof(char));
+ // Send/Recv buffers to/from others;
+ total = rank + (size - (rank + 1));
+ requests = xbt_new(MPI_Request, total);
+ tmpbufs = xbt_new(void*, rank);
+ index = 0;
+ for(other = 0; other < rank; other++) {
+ tmpbufs[index] = xbt_malloc(count * datasize);
+ requests[index] = smpi_irecv_init(tmpbufs[index], count, datatype, other, system_tag, comm);
+ index++;
+ }
+ for(other = rank + 1; other < size; other++) {
+ requests[index] = smpi_isend_init(sendbuf, count, datatype, other, system_tag, comm);
+ index++;
+ }
+ // Wait for completion of all comms.
+ smpi_mpi_startall(size - 1, requests);
+ for(other = 0; other < total; other++) {
+ index = smpi_mpi_waitany(size - 1, requests, MPI_STATUS_IGNORE);
+ if(index == MPI_UNDEFINED) {
+ break;
+ }
+ if(index < rank) {
+ // #Request is below rank: it's a irecv
+ smpi_op_apply(op, tmpbufs[index], recvbuf, &count, &datatype);
+ }
+ }
+ for(index = 0; index < size - 1; index++) {
+ xbt_free(tmpbufs[index]);
+ }
+ xbt_free(tmpbufs);
+ xbt_free(requests);