388 строки
13 KiB
C
388 строки
13 KiB
C
|
/*
|
||
|
|
* Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
|
||
|
|
* University Research and Technology
|
||
|
|
* Corporation. All rights reserved.
|
||
|
|
* Copyright (c) 2004-2006 The University of Tennessee and The University
|
||
|
|
* of Tennessee Research Foundation. All rights
|
||
|
|
* reserved.
|
||
|
|
* Copyright (c) 2004-2005 High Performance Computing Center Stuttgart,
|
||
|
|
* 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 "ompi_config.h"
|
||
|
|
|
||
|
|
#include "mpi.h"
|
||
|
|
#include "ompi/constants.h"
|
||
|
|
#include "ompi/datatype/datatype.h"
|
||
|
|
#include "ompi/communicator/communicator.h"
|
||
|
|
#include "ompi/mca/coll/coll.h"
|
||
|
|
#include "ompi/mca/coll/base/coll_tags.h"
|
||
|
|
#include "ompi/mca/pml/pml.h"
|
||
|
|
#include "ompi/op/op.h"
|
||
|
|
#include "coll_tuned.h"
|
||
|
|
#include "coll_tuned_topo.h"
|
||
|
|
#include "coll_tuned_util.h"
|
||
|
|
|
||
|
|
|
||
|
|
int
|
||
|
|
ompi_coll_tuned_scatter_intra_binomial(void *sbuf, int scount,
|
||
|
|
struct ompi_datatype_t *sdtype,
|
||
|
|
void *rbuf, int rcount,
|
||
|
|
struct ompi_datatype_t *rdtype,
|
||
|
|
int root, struct ompi_communicator_t *comm)
|
||
|
|
{
|
||
|
|
int line = -1;
|
||
|
|
int i;
|
||
|
|
int rank;
|
||
|
|
int vrank;
|
||
|
|
int size;
|
||
|
|
int total_recv = 0, total_send = 0;
|
||
|
|
char *ptmp = NULL;
|
||
|
|
char *tempbuf = NULL;
|
||
|
|
int err;
|
||
|
|
ompi_coll_tree_t* bmtree;
|
||
|
|
MPI_Status status;
|
||
|
|
MPI_Aint sextent, slb, strue_lb, strue_extent;
|
||
|
|
MPI_Aint rextent, rlb, rtrue_lb, rtrue_extent;
|
||
|
|
|
||
|
|
size = ompi_comm_size(comm);
|
||
|
|
rank = ompi_comm_rank(comm);
|
||
|
|
|
||
|
|
OPAL_OUTPUT((ompi_coll_tuned_stream,
|
||
|
|
"ompi_coll_tuned_scatter_intra_binomial rank %d", rank));
|
||
|
|
|
||
|
|
/* create the binomial tree */
|
||
|
|
COLL_TUNED_UPDATE_IN_ORDER_BMTREE( comm, root );
|
||
|
|
bmtree = comm->c_coll_selected_data->cached_in_order_bmtree;
|
||
|
|
|
||
|
|
ompi_ddt_get_extent(sdtype, &slb, &sextent);
|
||
|
|
ompi_ddt_get_true_extent(sdtype, &strue_lb, &strue_extent);
|
||
|
|
ompi_ddt_get_extent(rdtype, &rlb, &rextent);
|
||
|
|
ompi_ddt_get_true_extent(rdtype, &rtrue_lb, &rtrue_extent);
|
||
|
|
|
||
|
|
vrank = (rank - root + size) % size;
|
||
|
|
|
||
|
|
if (rank == root) {
|
||
|
|
if (0 == root) {
|
||
|
|
/* root on 0, just use the send buffer */
|
||
|
|
ptmp = sbuf;
|
||
|
|
if (rbuf != MPI_IN_PLACE) {
|
||
|
|
/* local copy to rbuf */
|
||
|
|
err = ompi_ddt_sndrcv(sbuf, scount, sdtype,
|
||
|
|
rbuf, rcount, rdtype);
|
||
|
|
if (MPI_SUCCESS != err) { line = __LINE__; goto err_hndl; }
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
/* root is not on 0, allocate temp buffer for send */
|
||
|
|
tempbuf = malloc(strue_extent + (scount*size - 1) * sextent);
|
||
|
|
if (NULL == tempbuf) {
|
||
|
|
err = OMPI_ERR_OUT_OF_RESOURCE; line = __LINE__; goto err_hndl;
|
||
|
|
}
|
||
|
|
|
||
|
|
ptmp = tempbuf - slb;
|
||
|
|
|
||
|
|
/* and rotate data so they will eventually in the right place */
|
||
|
|
err = ompi_ddt_copy_content_same_ddt(sdtype, scount*(size - root),
|
||
|
|
ptmp, (char *) sbuf + sextent*root*scount);
|
||
|
|
if (MPI_SUCCESS != err) { line = __LINE__; goto err_hndl; }
|
||
|
|
|
||
|
|
|
||
|
|
err = ompi_ddt_copy_content_same_ddt(sdtype, scount*root,
|
||
|
|
ptmp + sextent*scount*(size - root), sbuf);
|
||
|
|
if (MPI_SUCCESS != err) { line = __LINE__; goto err_hndl; }
|
||
|
|
|
||
|
|
if (rbuf != MPI_IN_PLACE) {
|
||
|
|
/* local copy to rbuf */
|
||
|
|
err = ompi_ddt_sndrcv(ptmp, scount, sdtype,
|
||
|
|
rbuf, rcount, rdtype);
|
||
|
|
if (MPI_SUCCESS != err) { line = __LINE__; goto err_hndl; }
|
||
|
|
}
|
||
|
|
}
|
||
|
|
total_send = scount;
|
||
|
|
} else if (!(vrank % 2)) {
|
||
|
|
/* non-root, non-leaf nodes, allocte temp buffer for recv
|
||
|
|
* the most we need is rcount*size/2 */
|
||
|
|
tempbuf = malloc(rtrue_extent + (rcount*size - 1) * rextent);
|
||
|
|
if (NULL == tempbuf) {
|
||
|
|
err= OMPI_ERR_OUT_OF_RESOURCE; line = __LINE__; goto err_hndl;
|
||
|
|
}
|
||
|
|
|
||
|
|
ptmp = tempbuf - rlb;
|
||
|
|
|
||
|
|
sdtype = rdtype;
|
||
|
|
scount = rcount;
|
||
|
|
sextent = rextent;
|
||
|
|
total_send = scount;
|
||
|
|
} else {
|
||
|
|
/* leaf nodes, just use rbuf */
|
||
|
|
ptmp = rbuf;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!(vrank % 2)) {
|
||
|
|
if (rank != root) {
|
||
|
|
/* recv from parent on non-root */
|
||
|
|
err = MCA_PML_CALL(recv(ptmp, rcount*size, rdtype, bmtree->tree_prev,
|
||
|
|
MCA_COLL_BASE_TAG_SCATTER, comm, &status));
|
||
|
|
if (MPI_SUCCESS != err) { line = __LINE__; goto err_hndl; }
|
||
|
|
/* local copy to rbuf */
|
||
|
|
err = ompi_ddt_sndrcv(ptmp, scount, sdtype,
|
||
|
|
rbuf, rcount, rdtype);
|
||
|
|
if (MPI_SUCCESS != err) { line = __LINE__; goto err_hndl; }
|
||
|
|
}
|
||
|
|
/* send to children on all non-leaf */
|
||
|
|
for (i = 0; i < bmtree->tree_nextsize; i++) {
|
||
|
|
int mycount = 0, vkid;
|
||
|
|
/* figure out how much data I have to send to this child */
|
||
|
|
vkid = (bmtree->tree_next[i] - root + size) % size;
|
||
|
|
mycount = vkid - vrank;
|
||
|
|
if (mycount > (size - vkid))
|
||
|
|
mycount = size - vkid;
|
||
|
|
mycount *= scount;
|
||
|
|
|
||
|
|
err = MCA_PML_CALL(send(ptmp + total_send*sextent, mycount, sdtype,
|
||
|
|
bmtree->tree_next[i],
|
||
|
|
MCA_COLL_BASE_TAG_SCATTER,
|
||
|
|
MCA_PML_BASE_SEND_STANDARD, comm));
|
||
|
|
if (MPI_SUCCESS != err) { line = __LINE__; goto err_hndl; }
|
||
|
|
|
||
|
|
total_send += mycount;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (NULL != tempbuf)
|
||
|
|
free(tempbuf);
|
||
|
|
} else {
|
||
|
|
/* recv from parent on leaf nodes */
|
||
|
|
err = MCA_PML_CALL(recv(ptmp, rcount, rdtype, bmtree->tree_prev,
|
||
|
|
MCA_COLL_BASE_TAG_SCATTER, comm, &status));
|
||
|
|
if (MPI_SUCCESS != err) { line = __LINE__; goto err_hndl; }
|
||
|
|
}
|
||
|
|
|
||
|
|
return MPI_SUCCESS;
|
||
|
|
|
||
|
|
err_hndl:
|
||
|
|
if (NULL != tempbuf)
|
||
|
|
free(tempbuf);
|
||
|
|
|
||
|
|
OPAL_OUTPUT((ompi_coll_tuned_stream, "%s:%4d\tError occurred %d, rank %2d",
|
||
|
|
__FILE__, line, err, rank));
|
||
|
|
return err;
|
||
|
|
}
|
||
|
|
|
||
|
|
/*
|
||
|
|
* Linear functions are copied from the BASIC coll module
|
||
|
|
* they do not segment the message and are simple implementations
|
||
|
|
* but for some small number of nodes and/or small data sizes they
|
||
|
|
* are just as fast as tuned/tree based segmenting operations
|
||
|
|
* and as such may be selected by the decision functions
|
||
|
|
* These are copied into this module due to the way we select modules
|
||
|
|
* in V1. i.e. in V2 we will handle this differently and so will not
|
||
|
|
* have to duplicate code.
|
||
|
|
* JPG following the examples from other coll_tuned implementations. Dec06.
|
||
|
|
*/
|
||
|
|
|
||
|
|
/* copied function (with appropriate renaming) starts here */
|
||
|
|
/*
|
||
|
|
* scatter_intra
|
||
|
|
*
|
||
|
|
* Function: - basic scatter operation
|
||
|
|
* Accepts: - same arguments as MPI_Scatter()
|
||
|
|
* Returns: - MPI_SUCCESS or error code
|
||
|
|
*/
|
||
|
|
int
|
||
|
|
ompi_coll_tuned_scatter_intra_basic_linear(void *sbuf, int scount,
|
||
|
|
struct ompi_datatype_t *sdtype,
|
||
|
|
void *rbuf, int rcount,
|
||
|
|
struct ompi_datatype_t *rdtype,
|
||
|
|
int root, struct ompi_communicator_t *comm)
|
||
|
|
{
|
||
|
|
int i, rank, size, err;
|
||
|
|
char *ptmp;
|
||
|
|
ptrdiff_t lb, incr;
|
||
|
|
|
||
|
|
/* Initialize */
|
||
|
|
|
||
|
|
rank = ompi_comm_rank(comm);
|
||
|
|
size = ompi_comm_size(comm);
|
||
|
|
|
||
|
|
/* If not root, receive data. */
|
||
|
|
|
||
|
|
if (rank != root) {
|
||
|
|
err = MCA_PML_CALL(recv(rbuf, rcount, rdtype, root,
|
||
|
|
MCA_COLL_BASE_TAG_SCATTER,
|
||
|
|
comm, MPI_STATUS_IGNORE));
|
||
|
|
return err;
|
||
|
|
}
|
||
|
|
|
||
|
|
/* I am the root, loop sending data. */
|
||
|
|
|
||
|
|
err = ompi_ddt_get_extent(sdtype, &lb, &incr);
|
||
|
|
if (OMPI_SUCCESS != err) {
|
||
|
|
return OMPI_ERROR;
|
||
|
|
}
|
||
|
|
|
||
|
|
incr *= scount;
|
||
|
|
for (i = 0, ptmp = (char *) sbuf; i < size; ++i, ptmp += incr) {
|
||
|
|
|
||
|
|
/* simple optimization */
|
||
|
|
|
||
|
|
if (i == rank) {
|
||
|
|
if (MPI_IN_PLACE != rbuf) {
|
||
|
|
err =
|
||
|
|
ompi_ddt_sndrcv(ptmp, scount, sdtype, rbuf, rcount,
|
||
|
|
rdtype);
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
err = MCA_PML_CALL(send(ptmp, scount, sdtype, i,
|
||
|
|
MCA_COLL_BASE_TAG_SCATTER,
|
||
|
|
MCA_PML_BASE_SEND_STANDARD, comm));
|
||
|
|
}
|
||
|
|
if (MPI_SUCCESS != err) {
|
||
|
|
return err;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/* All done */
|
||
|
|
|
||
|
|
return MPI_SUCCESS;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
/* copied function (with appropriate renaming) ends here */
|
||
|
|
|
||
|
|
/* The following are used by dynamic and forced rules */
|
||
|
|
|
||
|
|
/* publish details of each algorithm and if its forced/fixed/locked in */
|
||
|
|
/* as you add methods/algorithms you must update this and the query/map
|
||
|
|
routines */
|
||
|
|
|
||
|
|
/* this routine is called by the component only */
|
||
|
|
/* this makes sure that the mca parameters are set to their initial values
|
||
|
|
and perms */
|
||
|
|
/* module does not call this they call the forced_getvalues routine instead */
|
||
|
|
|
||
|
|
int
|
||
|
|
ompi_coll_tuned_scatter_intra_check_forced_init(coll_tuned_force_algorithm_mca_param_indices_t *mca_param_indices)
|
||
|
|
{
|
||
|
|
int rc, max_alg = 2, requested_alg;
|
||
|
|
|
||
|
|
ompi_coll_tuned_forced_max_algorithms[SCATTER] = max_alg;
|
||
|
|
|
||
|
|
rc = mca_base_param_reg_int (&mca_coll_tuned_component.super.collm_version,
|
||
|
|
"scatter_algorithm_count",
|
||
|
|
"Number of scatter algorithms available",
|
||
|
|
false, true, max_alg, NULL);
|
||
|
|
|
||
|
|
mca_param_indices->algorithm_param_index
|
||
|
|
= mca_base_param_reg_int(&mca_coll_tuned_component.super.collm_version,
|
||
|
|
"scatter_algorithm",
|
||
|
|
"Which scatter algorithm is used. Can be locked down to choice of: 0 ignore, 1 basic linear, 2 binomial.",
|
||
|
|
false, false, 0, NULL);
|
||
|
|
mca_base_param_lookup_int(mca_param_indices->algorithm_param_index,
|
||
|
|
&(requested_alg));
|
||
|
|
if( requested_alg > max_alg ) {
|
||
|
|
if( 0 == ompi_comm_rank( MPI_COMM_WORLD ) ) {
|
||
|
|
opal_output( 0, "Scatter algorithm #%d is not available (range [0..%d]). Switching back to ignore(0)\n",
|
||
|
|
requested_alg, max_alg );
|
||
|
|
}
|
||
|
|
mca_base_param_set_int( mca_param_indices->algorithm_param_index, 0);
|
||
|
|
}
|
||
|
|
|
||
|
|
mca_param_indices->segsize_param_index
|
||
|
|
= mca_base_param_reg_int(&mca_coll_tuned_component.super.collm_version,
|
||
|
|
"Scatter_algorithm_segmentsize",
|
||
|
|
"Segment size in bytes used by default for scatter algorithms. Only has meaning if algorithm is forced and supports segmenting. 0 bytes means no segmentation. Currently, available algorithms do not support segmentation.",
|
||
|
|
false, false, 0, NULL);
|
||
|
|
|
||
|
|
mca_param_indices->tree_fanout_param_index
|
||
|
|
= mca_base_param_reg_int(&mca_coll_tuned_component.super.collm_version,
|
||
|
|
"scatter_algorithm_tree_fanout",
|
||
|
|
"Fanout for n-tree used for scatter algorithms. Only has meaning if algorithm is forced and supports n-tree topo based operation. Currently, available algorithms do not support n-tree topologies.",
|
||
|
|
false, false,
|
||
|
|
ompi_coll_tuned_init_tree_fanout, /* get system wide default */
|
||
|
|
NULL);
|
||
|
|
|
||
|
|
mca_param_indices->chain_fanout_param_index
|
||
|
|
= mca_base_param_reg_int(&mca_coll_tuned_component.super.collm_version,
|
||
|
|
"scatter_algorithm_chain_fanout",
|
||
|
|
"Fanout for chains used for scatter algorithms. Only has meaning if algorithm is forced and supports chain topo based operation. Currently, available algorithms do not support chain topologies.",
|
||
|
|
false, false,
|
||
|
|
ompi_coll_tuned_init_chain_fanout, /* get system wide default */
|
||
|
|
NULL);
|
||
|
|
|
||
|
|
return (MPI_SUCCESS);
|
||
|
|
}
|
||
|
|
|
||
|
|
int
|
||
|
|
ompi_coll_tuned_scatter_intra_do_forced(void *sbuf, int scount,
|
||
|
|
struct ompi_datatype_t *sdtype,
|
||
|
|
void* rbuf, int rcount,
|
||
|
|
struct ompi_datatype_t *rdtype,
|
||
|
|
int root, struct ompi_communicator_t *comm)
|
||
|
|
{
|
||
|
|
OPAL_OUTPUT((ompi_coll_tuned_stream,
|
||
|
|
"coll:tuned:scatter_intra_do_forced selected algorithm %d",
|
||
|
|
comm->c_coll_selected_data->user_forced[SCATTER].algorithm));
|
||
|
|
|
||
|
|
switch (comm->c_coll_selected_data->user_forced[SCATTER].algorithm) {
|
||
|
|
case (0):
|
||
|
|
return ompi_coll_tuned_scatter_intra_dec_fixed (sbuf, scount, sdtype,
|
||
|
|
rbuf, rcount, rdtype,
|
||
|
|
root, comm);
|
||
|
|
case (1):
|
||
|
|
return ompi_coll_tuned_scatter_intra_basic_linear (sbuf, scount, sdtype,
|
||
|
|
rbuf, rcount, rdtype,
|
||
|
|
root, comm);
|
||
|
|
case (2):
|
||
|
|
return ompi_coll_tuned_scatter_intra_binomial(sbuf, scount, sdtype,
|
||
|
|
rbuf, rcount, rdtype,
|
||
|
|
root, comm);
|
||
|
|
default:
|
||
|
|
OPAL_OUTPUT((ompi_coll_tuned_stream,
|
||
|
|
"coll:tuned:scatter_intra_do_forced attempt to select algorithm %d when only 0-%d is valid?",
|
||
|
|
comm->c_coll_selected_data->user_forced[SCATTER].algorithm,
|
||
|
|
ompi_coll_tuned_forced_max_algorithms[SCATTER]));
|
||
|
|
return (MPI_ERR_ARG);
|
||
|
|
} /* switch */
|
||
|
|
}
|
||
|
|
|
||
|
|
int
|
||
|
|
ompi_coll_tuned_scatter_intra_do_this(void *sbuf, int scount,
|
||
|
|
struct ompi_datatype_t *sdtype,
|
||
|
|
void* rbuf, int rcount,
|
||
|
|
struct ompi_datatype_t *rdtype,
|
||
|
|
int root, struct ompi_communicator_t *comm,
|
||
|
|
int algorithm, int faninout, int segsize)
|
||
|
|
{
|
||
|
|
OPAL_OUTPUT((ompi_coll_tuned_stream,
|
||
|
|
"coll:tuned:scatter_intra_do_this selected algorithm %d topo faninout %d segsize %d",
|
||
|
|
algorithm, faninout, segsize));
|
||
|
|
|
||
|
|
switch (algorithm) {
|
||
|
|
case (0):
|
||
|
|
return ompi_coll_tuned_scatter_intra_dec_fixed (sbuf, scount, sdtype,
|
||
|
|
rbuf, rcount, rdtype,
|
||
|
|
root, comm);
|
||
|
|
case (1):
|
||
|
|
return ompi_coll_tuned_scatter_intra_basic_linear (sbuf, scount, sdtype,
|
||
|
|
rbuf, rcount, rdtype,
|
||
|
|
root, comm);
|
||
|
|
case (2):
|
||
|
|
return ompi_coll_tuned_scatter_intra_binomial(sbuf, scount, sdtype,
|
||
|
|
rbuf, rcount, rdtype,
|
||
|
|
root, comm);
|
||
|
|
default:
|
||
|
|
OPAL_OUTPUT((ompi_coll_tuned_stream,
|
||
|
|
"coll:tuned:scatter_intra_do_this attempt to select algorithm %d when only 0-%d is valid?",
|
||
|
|
algorithm,
|
||
|
|
ompi_coll_tuned_forced_max_algorithms[SCATTER]));
|
||
|
|
return (MPI_ERR_ARG);
|
||
|
|
} /* switch */
|
||
|
|
}
|