X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/2a9d2f09c3a11c11dbce827e48c9177a911854ac..e7c0c67af63b3979a597a66e5e1c8b0435fc6e19:/src/smpi/colls/bcast-flattree.c

diff --git a/src/smpi/colls/bcast-flattree.c b/src/smpi/colls/bcast-flattree.c
index 693d83d555..626c20381c 100644
--- a/src/smpi/colls/bcast-flattree.c
+++ b/src/smpi/colls/bcast-flattree.c
@@ -1,4 +1,4 @@
-#include "colls.h"
+#include "colls_private.h"
 
 int
 smpi_coll_tuned_bcast_flattree(void *buff, int count, MPI_Datatype data_type,
@@ -10,26 +10,26 @@ smpi_coll_tuned_bcast_flattree(void *buff, int count, MPI_Datatype data_type,
   int i, rank, num_procs;
   int tag = 1;
 
-  MPI_Comm_rank(comm, &rank);
-  MPI_Comm_size(comm, &num_procs);
+  rank = smpi_comm_rank(comm);
+  num_procs = smpi_comm_size(comm);
 
   if (rank != root) {
-    MPI_Recv(buff, count, data_type, root, tag, comm, MPI_STATUS_IGNORE);
+    smpi_mpi_recv(buff, count, data_type, root, tag, comm, MPI_STATUS_IGNORE);
   }
 
   else {
-    reqs = (MPI_Request *) malloc((num_procs - 1) * sizeof(MPI_Request));
+    reqs = (MPI_Request *) xbt_malloc((num_procs - 1) * sizeof(MPI_Request));
     req_ptr = reqs;
 
     // Root sends data to all others
     for (i = 0; i < num_procs; i++) {
       if (i == rank)
         continue;
-      MPI_Isend(buff, count, data_type, i, tag, comm, req_ptr++);
+      *(req_ptr++) = smpi_mpi_isend(buff, count, data_type, i, tag, comm);
     }
 
     // wait on all requests
-    MPI_Waitall(num_procs - 1, reqs, MPI_STATUSES_IGNORE);
+    smpi_mpi_waitall(num_procs - 1, reqs, MPI_STATUSES_IGNORE);
 
     free(reqs);
   }