X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/482cc3463d7b94d6c39816c23f0d1e972a289728..a00f94a0c4dab6793ebd7f61961ef04d15873c23:/src/smpi/colls/alltoall-pair.c diff --git a/src/smpi/colls/alltoall-pair.c b/src/smpi/colls/alltoall-pair.c index 83658e02c0..24ef5f769c 100644 --- a/src/smpi/colls/alltoall-pair.c +++ b/src/smpi/colls/alltoall-pair.c @@ -1,4 +1,5 @@ -#include "smpi/mpi.h" +#include "colls_private.h" + /***************************************************************************** * Function: alltoall_pair @@ -21,10 +22,9 @@ ****************************************************************************/ /* -int -alltoall_pair(void * send_buff, int send_count, MPI_Datatype send_type, - void * recv_buff, int recv_count, MPI_Datatype recv_type, - MPI_Comm comm) +int alltoall_pair(void *send_buff, int send_count, MPI_Datatype send_type, + void *recv_buff, int recv_count, MPI_Datatype recv_type, + MPI_Comm comm) { MPI_Aint send_chunk, recv_chunk; @@ -34,62 +34,58 @@ alltoall_pair(void * send_buff, int send_count, MPI_Datatype send_type, int i, src, dst, rank, num_procs; int tag = 1, success = 1, failure = 0, pof2 = 1; - char * send_ptr = (char *) send_buff; - char * recv_ptr = (char *) recv_buff; - - MPI_Comm_rank(comm, &rank); - MPI_Comm_size(comm, &num_procs); - MPI_Type_extent(send_type, &send_chunk); - MPI_Type_extent(recv_type, &recv_chunk); + char *send_ptr = (char *) send_buff; + char *recv_ptr = (char *) recv_buff; + + rank = smpi_comm_rank(comm); + num_procs = smpi_comm_size(comm); + send_chunk = smpi_datatype_get_extent(send_type); + recv_chunk = smpi_datatype_get_extent(recv_type); - MPI_Win_create(recv_buff, num_procs*recv_chunk*send_count,recv_chunk,0, - comm, &win); + MPI_Win_create(recv_buff, num_procs * recv_chunk * send_count, recv_chunk, 0, + comm, &win); send_chunk *= send_count; - recv_chunk *= recv_count; + recv_chunk *= recv_count; MPI_Win_fence(assert, win); - for (i = 0; i < num_procs; i++) - { - src = dst = rank ^ i; - MPI_Put(send_ptr + dst * send_chunk, send_count, send_type, dst, - rank*send_chunk, send_count, send_type, win); - } - MPI_Win_fence (assert, win); + for (i = 0; i < num_procs; i++) { + src = dst = rank ^ i; + MPI_Put(send_ptr + dst * send_chunk, send_count, send_type, dst, + rank * send_chunk, send_count, send_type, win); + } + MPI_Win_fence(assert, win); MPI_Win_free(&win); return 0; } */ -int -smpi_coll_tuned_alltoall_pair(void * send_buff, int send_count, MPI_Datatype send_type, - void * recv_buff, int recv_count, MPI_Datatype recv_type, - MPI_Comm comm) +int smpi_coll_tuned_alltoall_pair(void *send_buff, int send_count, + MPI_Datatype send_type, + void *recv_buff, int recv_count, + MPI_Datatype recv_type, MPI_Comm comm) { MPI_Aint send_chunk, recv_chunk; MPI_Status s; int i, src, dst, rank, num_procs; - int tag = 1, success = 1; + int tag = 1; + char *send_ptr = (char *) send_buff; + char *recv_ptr = (char *) recv_buff; - char * send_ptr = (char *) send_buff; - char * recv_ptr = (char *) recv_buff; - - MPI_Comm_rank(comm, &rank); - MPI_Comm_size(comm, &num_procs); - MPI_Type_extent(send_type, &send_chunk); - MPI_Type_extent(recv_type, &recv_chunk); + rank = smpi_comm_rank(comm); + num_procs = smpi_comm_size(comm); + send_chunk = smpi_datatype_get_extent(send_type); + recv_chunk = smpi_datatype_get_extent(recv_type); send_chunk *= send_count; - recv_chunk *= recv_count; + recv_chunk *= recv_count; - for (i = 0; i < num_procs; i++) - { - src = dst = rank ^ i; - MPI_Sendrecv(send_ptr + dst * send_chunk, send_count, send_type, dst, - tag, recv_ptr + src * recv_chunk, recv_count, recv_type, - src, tag, comm, &s); - } + for (i = 0; i < num_procs; i++) { + src = dst = rank ^ i; + smpi_mpi_sendrecv(send_ptr + dst * send_chunk, send_count, send_type, dst, tag, + recv_ptr + src * recv_chunk, recv_count, recv_type, src, tag, + comm, &s); + } - return success; + return MPI_SUCCESS; } -