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

diff --git a/src/smpi/colls/bcast-flattree.c b/src/smpi/colls/bcast-flattree.c
index 693d83d555..5d4708d1e3 100644
--- a/src/smpi/colls/bcast-flattree.c
+++ b/src/smpi/colls/bcast-flattree.c
@@ -1,4 +1,10 @@
-#include "colls.h"
+/* 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"
 
 int
 smpi_coll_tuned_bcast_flattree(void *buff, int count, MPI_Datatype data_type,
@@ -8,28 +14,28 @@ smpi_coll_tuned_bcast_flattree(void *buff, int count, MPI_Datatype data_type,
   MPI_Request *reqs;
 
   int i, rank, num_procs;
-  int tag = 1;
+  int tag = COLL_TAG_BCAST;
 
-  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);
   }