Use simgrid function instead of MPI in collectives

[simgrid.git] / src / smpi / colls / allgather-GB.c

diff --git a/src/smpi/colls/allgather-GB.c b/src/smpi/colls/allgather-GB.c
index b9df40e..f9a1e07 100644 (file)
--- a/src/smpi/colls/allgather-GB.c
+++ b/src/smpi/colls/allgather-GB.c
@@ -1,4 +1,4 @@
-#include "colls.h"
+#include "colls_private.h"
 
 // Allgather - gather/bcast algorithm
 int smpi_coll_tuned_allgather_GB(void *send_buff, int send_count,
@@ -7,10 +7,10 @@ int smpi_coll_tuned_allgather_GB(void *send_buff, int send_count,
                                  MPI_Comm comm)
 {
   int num_procs;
-  MPI_Comm_size(comm, &num_procs);
-  MPI_Gather(send_buff, send_count, send_type, recv_buff, recv_count, recv_type,
+  num_procs = smpi_comm_size(comm);
+  smpi_mpi_gather(send_buff, send_count, send_type, recv_buff, recv_count, recv_type,
              0, comm);
-  MPI_Bcast(recv_buff, (recv_count * num_procs), recv_type, 0, comm);
+  mpi_coll_bcast_fun(recv_buff, (recv_count * num_procs), recv_type, 0, comm);
 
   return MPI_SUCCESS;
 }