X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/89e44dce958ad9625570b126f11d299b829d4932..ec3e4ee5f1a7ffeb96e044057809944f364014e6:/src/smpi/colls/allreduce-rdb.c diff --git a/src/smpi/colls/allreduce-rdb.c b/src/smpi/colls/allreduce-rdb.c index f1fe9f6a5d..85a31efd54 100644 --- a/src/smpi/colls/allreduce-rdb.c +++ b/src/smpi/colls/allreduce-rdb.c @@ -1,10 +1,16 @@ +/* Copyright (c) 2013-2014. The SimGrid Team. + * All rights reserved. */ + +/* This program is free software; you can redistribute it and/or modify it + * under the terms of the license (GNU LGPL) which comes with this package. */ + #include "colls_private.h" //#include int smpi_coll_tuned_allreduce_rdb(void *sbuff, void *rbuff, int count, MPI_Datatype dtype, MPI_Op op, MPI_Comm comm) { - int nprocs, rank, type_size, tag = 543; + int nprocs, rank, tag = COLL_TAG_ALLREDUCE; int mask, dst, pof2, newrank, rem, newdst; MPI_Aint extent, lb; MPI_Status status; @@ -23,13 +29,11 @@ int smpi_coll_tuned_allreduce_rdb(void *sbuff, void *rbuff, int count, rank=smpi_comm_rank(comm); smpi_datatype_extent(dtype, &lb, &extent); - tmp_buf = (void *) xbt_malloc(count * extent); + tmp_buf = (void *) smpi_get_tmp_sendbuffer(count * extent); smpi_mpi_sendrecv(sbuff, count, dtype, rank, 500, rbuff, count, dtype, rank, 500, comm, &status); - type_size=smpi_datatype_size(dtype); - // find nearest power-of-two less than or equal to comm_size pof2 = 1; while (pof2 <= nprocs) @@ -121,6 +125,6 @@ int smpi_coll_tuned_allreduce_rdb(void *sbuff, void *rbuff, int count, smpi_mpi_recv(rbuff, count, dtype, rank + 1, tag, comm, &status); } - free(tmp_buf); + smpi_free_tmp_buffer(tmp_buf); return MPI_SUCCESS; }