src/smpi/colls/allgatherv/allgatherv-GB.cpp

   1 /* Copyright (c) 2013-2022. The SimGrid Team.
   2  * All rights reserved.                                                     */
   3
   4 /* This program is free software; you can redistribute it and/or modify it
   5  * under the terms of the license (GNU LGPL) which comes with this package. */
   6
   7 #include "../colls_private.hpp"
   8
   9 namespace simgrid::smpi {
  10
  11 // Allgather - gather/bcast algorithm
  12 int allgatherv__GB(const void *send_buff, int send_count,
  13                    MPI_Datatype send_type, void *recv_buff,
  14                    const int *recv_counts, const int *recv_disps, MPI_Datatype recv_type,
  15                    MPI_Comm comm)
  16 {
  17   colls::gatherv(send_buff, send_count, send_type, recv_buff, recv_counts, recv_disps, recv_type, 0, comm);
  18   int num_procs, i, current, max = 0;
  19   num_procs = comm->size();
  20   for (i = 0; i < num_procs; i++) {
  21     current = recv_disps[i] + recv_counts[i];
  22     if (current > max)
  23       max = current;
  24   }
  25   colls::bcast(recv_buff, max, recv_type, 0, comm);
  26
  27   return MPI_SUCCESS;
  28 }
  29
  30 } // namespace simgrid::smpi