1 /*
2 * Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
3 * University Research and Technology
4 * Corporation. All rights reserved.
5 * Copyright (c) 2004-2005 The University of Tennessee and The University
6 * of Tennessee Research Foundation. All rights
7 * reserved.
8 * Copyright (c) 2004-2005 High Performance Computing Center Stuttgart,
9 * University of Stuttgart. All rights reserved.
10 * Copyright (c) 2004-2005 The Regents of the University of California.
11 * All rights reserved.
12 * $COPYRIGHT$
13 *
14 * Additional copyrights may follow
15 *
16 * $HEADER$
17 */
18
19 #include "ompi_config.h"
20 #include "opal/util/output.h"
21
22 #include "mpi.h"
23 #include "ompi/constants.h"
24 #include "ompi/mca/coll/coll.h"
25 #include "ompi/mca/coll/base/base.h"
26 #include "coll_demo.h"
27
28
29 /*
30 * reduce_scatter
31 *
32 * Function: - reduce then scatter
33 * Accepts: - same as MPI_Reduce_scatter()
34 * Returns: - MPI_SUCCESS or error code
35 */
36 int mca_coll_demo_reduce_scatter_intra(void *sbuf, void *rbuf, int *rcounts,
37 struct ompi_datatype_t *dtype,
38 struct ompi_op_t *op,
39 struct ompi_communicator_t *comm,
40 mca_coll_base_module_t *module)
41 {
42 mca_coll_demo_module_t *demo_module = (mca_coll_demo_module_t*) module;
43 opal_output_verbose(10, ompi_coll_base_framework.framework_output, "In demo scatter_intra");
44 return demo_module->underlying.coll_reduce_scatter(sbuf, rbuf, rcounts,
45 dtype, op, comm,
46 demo_module->underlying.coll_reduce_scatter_module);
47 }
48
49
50 /*
51 * reduce_scatter_inter
52 *
53 * Function: - reduce/scatter operation
54 * Accepts: - same arguments as MPI_Reduce_scatter()
55 * Returns: - MPI_SUCCESS or error code
56 */
57 int mca_coll_demo_reduce_scatter_inter(void *sbuf, void *rbuf, int *rcounts,
58 struct ompi_datatype_t *dtype,
59 struct ompi_op_t *op,
60 struct ompi_communicator_t *comm,
61 mca_coll_base_module_t *module)
62 {
63 mca_coll_demo_module_t *demo_module = (mca_coll_demo_module_t*) module;
64 opal_output_verbose(10, ompi_coll_base_framework.framework_output, "In demo scatter_inter");
65 return demo_module->underlying.coll_reduce_scatter(sbuf, rbuf, rcounts,
66 dtype, op, comm,
67 demo_module->underlying.coll_reduce_scatter_module);
68 }