1 /* Copyright (c) 2013-2014. The SimGrid Team.
2 * All rights reserved. */
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. */
7 #include "colls_private.h"
9 int flattree_segment_in_byte = 8192;
12 smpi_coll_tuned_bcast_flattree_pipeline(void *buff, int count,
13 MPI_Datatype data_type, int root,
16 int i, j, rank, num_procs;
17 int tag = COLL_TAG_BCAST;
20 extent = smpi_datatype_get_extent(data_type);
22 int segment = flattree_segment_in_byte / extent;
23 segment = segment == 0 ? 1 :segment;
24 int pipe_length = count / segment;
25 int increment = segment * extent;
27 XBT_WARN("MPI_bcast_flattree_pipeline use default MPI_bcast_flattree.");
28 return smpi_coll_tuned_bcast_flattree(buff, count, data_type, root, comm);
30 rank = smpi_comm_rank(comm);
31 num_procs = smpi_comm_size(comm);
33 MPI_Request *request_array;
34 MPI_Status *status_array;
36 request_array = (MPI_Request *) xbt_malloc(pipe_length * sizeof(MPI_Request));
37 status_array = (MPI_Status *) xbt_malloc(pipe_length * sizeof(MPI_Status));
40 for (i = 0; i < pipe_length; i++) {
41 request_array[i] = smpi_mpi_irecv((char *)buff + (i * increment), segment, data_type, root, tag, comm);
43 smpi_mpi_waitall(pipe_length, request_array, status_array);
47 // Root sends data to all others
48 for (j = 0; j < num_procs; j++) {
52 for (i = 0; i < pipe_length; i++) {
53 smpi_mpi_send((char *)buff + (i * increment), segment, data_type, j, tag, comm);