+/* 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. */
+
/*
* Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
* University Research and Technology
* University of Stuttgart. All rights reserved.
* Copyright (c) 2004-2005 The Regents of the University of California.
* All rights reserved.
- * $COPYRIGHT$
*
* Additional copyrights may follow
- *
- * $HEADER$
*/
#include "colls_private.h"
#include "coll_tuned_topo.h"
-#define MCA_COLL_BASE_TAG_GATHER 333
/* Todo: gather_intra_generic, gather_intra_binary, gather_intra_chain,
* gather_intra_pipeline, segmentation? */
int
"smpi_coll_tuned_gather_ompi_binomial rank %d recv %d mycount = %d",
rank, bmtree->tree_next[i], mycount);
- smpi_mpi_recv(ptmp + total_recv*rextent, rcount*size-total_recv, rdtype,
- bmtree->tree_next[i], MCA_COLL_BASE_TAG_GATHER,
+ smpi_mpi_recv(ptmp + total_recv*rextent, mycount, rdtype,
+ bmtree->tree_next[i], COLL_TAG_GATHER,
comm, &status);
total_recv += mycount;
smpi_mpi_send(ptmp, total_recv, sdtype,
bmtree->tree_prev,
- MCA_COLL_BASE_TAG_GATHER,
+ COLL_TAG_GATHER,
comm);
}
if (rank == root) {
/* other non-leaf nodes */
free(tempbuf);
}
+ xbt_free(bmtree);
return MPI_SUCCESS;
err_hndl:
first_segment_count );
smpi_mpi_recv(sbuf, 0, MPI_BYTE, root,
- MCA_COLL_BASE_TAG_GATHER,
+ COLL_TAG_GATHER,
comm, MPI_STATUS_IGNORE);
smpi_mpi_send(sbuf, first_segment_count, sdtype, root,
- MCA_COLL_BASE_TAG_GATHER,
+ COLL_TAG_GATHER,
comm);
smpi_mpi_send((char*)sbuf + extent * first_segment_count,
(scount - first_segment_count), sdtype,
- root, MCA_COLL_BASE_TAG_GATHER,
+ root, COLL_TAG_GATHER,
comm);
}
COLL_TUNED_COMPUTED_SEGCOUNT( (size_t)first_segment_size, typelng,
first_segment_count );
- ptmp = (char *) rbuf;
for (i = 0; i < size; ++i) {
if (i == rank) {
/* skip myself */
/* irecv for the first segment from i */
ptmp = (char*)rbuf + i * rcount * extent;
first_segment_req = smpi_mpi_irecv(ptmp, first_segment_count, rdtype, i,
- MCA_COLL_BASE_TAG_GATHER, comm
+ COLL_TAG_GATHER, comm
);
/* send sync message */
smpi_mpi_send(rbuf, 0, MPI_BYTE, i,
- MCA_COLL_BASE_TAG_GATHER,
+ COLL_TAG_GATHER,
comm);
/* irecv for the second segment */
ptmp = (char*)rbuf + (i * rcount + first_segment_count) * extent;
reqs[i]=smpi_mpi_irecv(ptmp, (rcount - first_segment_count),
- rdtype, i, MCA_COLL_BASE_TAG_GATHER, comm
+ rdtype, i, COLL_TAG_GATHER, comm
);
/* wait on the first segment to complete */
/* wait all second segments to complete */
ret = smpi_mpi_waitall(size, reqs, MPI_STATUSES_IGNORE);
+ if (ret != MPI_SUCCESS) { line = __LINE__; goto error_hndl; }
free(reqs);
}
if (rank != root) {
smpi_mpi_send(sbuf, scount, sdtype, root,
- MCA_COLL_BASE_TAG_GATHER,
+ COLL_TAG_GATHER,
comm);
return MPI_SUCCESS;
}
}
} else {
smpi_mpi_recv(ptmp, rcount, rdtype, i,
- MCA_COLL_BASE_TAG_GATHER,
+ COLL_TAG_GATHER,
comm, MPI_STATUS_IGNORE);
err = MPI_SUCCESS;
}