X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/7ce270272856fcab2790a35bcefb9d6e0fb52423..0dc6f803fc980dade7792b7dfb4fa699073604d4:/src/smpi/colls/alltoall-pair-one-barrier.c diff --git a/src/smpi/colls/alltoall-pair-one-barrier.c b/src/smpi/colls/alltoall-pair-one-barrier.c index 478b881d8e..209fa81c29 100644 --- a/src/smpi/colls/alltoall-pair-one-barrier.c +++ b/src/smpi/colls/alltoall-pair-one-barrier.c @@ -1,4 +1,4 @@ -#include "colls.h" +#include "colls_private.h" /***************************************************************************** * Function: alltoall_pair @@ -21,35 +21,35 @@ ****************************************************************************/ int -smpi_coll_tuned_alltoall_pair_one_barrier(void * send_buff, int send_count, MPI_Datatype send_type, - void * recv_buff, int recv_count, MPI_Datatype recv_type, - MPI_Comm comm) +smpi_coll_tuned_alltoall_pair_one_barrier(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; /*, failure = 0, pof2 = 1; */ + int tag = COLL_TAG_ALLTOALL; - 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); send_chunk *= send_count; - recv_chunk *= recv_count; - - MPI_Barrier(comm); - 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); - } - - return success; + recv_chunk *= recv_count; + + mpi_coll_barrier_fun(comm); + 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 MPI_SUCCESS; }