4d8a187450
* New "op" MPI layer framework * Addition of the MPI_REDUCE_LOCAL proposed function (for MPI-2.2) = Op framework = Add new "op" framework in the ompi layer. This framework replaces the hard-coded MPI_Op back-end functions for (MPI_Op, MPI_Datatype) tuples for pre-defined MPI_Ops, allowing components and modules to provide the back-end functions. The intent is that components can be written to take advantage of hardware acceleration (GPU, FPGA, specialized CPU instructions, etc.). Similar to other frameworks, components are intended to be able to discover at run-time if they can be used, and if so, elect themselves to be selected (or disqualify themselves from selection if they cannot run). If specialized hardware is not available, there is a default set of functions that will automatically be used. This framework is ''not'' used for user-defined MPI_Ops. The new op framework is similar to the existing coll framework, in that the final set of function pointers that are used on any given intrinsic MPI_Op can be a mixed bag of function pointers, potentially coming from multiple different op modules. This allows for hardware that only supports some of the operations, not all of them (e.g., a GPU that only supports single-precision operations). All the hard-coded back-end MPI_Op functions for (MPI_Op, MPI_Datatype) tuples still exist, but unlike coll, they're in the framework base (vs. being in a separate "basic" component) and are automatically used if no component is found at runtime that provides a module with the necessary function pointers. There is an "example" op component that will hopefully be useful to those writing meaningful op components. It is currently .ompi_ignore'd so that it doesn't impinge on other developers (it's somewhat chatty in terms of opal_output() so that you can tell when its functions have been invoked). See the README file in the example op component directory. Developers of new op components are encouraged to look at the following wiki pages: https://svn.open-mpi.org/trac/ompi/wiki/devel/Autogen https://svn.open-mpi.org/trac/ompi/wiki/devel/CreateComponent https://svn.open-mpi.org/trac/ompi/wiki/devel/CreateFramework = MPI_REDUCE_LOCAL = Part of the MPI-2.2 proposal listed here: https://svn.mpi-forum.org/trac/mpi-forum-web/ticket/24 is to add a new function named MPI_REDUCE_LOCAL. It is very easy to implement, so I added it (also because it makes testing the op framework pretty easy -- you can do it in serial rather than via parallel reductions). There's even a man page! This commit was SVN r20280.
141 строка
3.4 KiB
C++
141 строка
3.4 KiB
C++
// -*- c++ -*-
|
|
//
|
|
// Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
|
|
// University Research and Technology
|
|
// Corporation. All rights reserved.
|
|
// Copyright (c) 2004-2005 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 (c) 2006-2009 Cisco Systems, Inc. All rights reserved.
|
|
// $COPYRIGHT$
|
|
//
|
|
// Additional copyrights may follow
|
|
//
|
|
// $HEADER$
|
|
//
|
|
|
|
#if 0 /* OMPI_ENABLE_MPI_PROFILING */
|
|
|
|
inline
|
|
MPI::Op::Op() { }
|
|
|
|
inline
|
|
MPI::Op::Op(const MPI::Op& o) : pmpi_op(o.pmpi_op) { }
|
|
|
|
inline
|
|
MPI::Op::Op(MPI_Op o) : pmpi_op(o) { }
|
|
|
|
inline
|
|
MPI::Op::~Op() { }
|
|
|
|
inline
|
|
MPI::Op& MPI::Op::operator=(const MPI::Op& op) {
|
|
pmpi_op = op.pmpi_op; return *this;
|
|
}
|
|
|
|
// comparison
|
|
inline bool
|
|
MPI::Op::operator== (const MPI::Op &a) {
|
|
return (bool)(pmpi_op == a.pmpi_op);
|
|
}
|
|
|
|
inline bool
|
|
MPI::Op::operator!= (const MPI::Op &a) {
|
|
return (bool)!(*this == a);
|
|
}
|
|
|
|
// inter-language operability
|
|
inline MPI::Op&
|
|
MPI::Op::operator= (const MPI_Op &i) { pmpi_op = i; return *this; }
|
|
|
|
inline
|
|
MPI::Op::operator MPI_Op () const { return pmpi_op; }
|
|
|
|
//inline
|
|
//MPI::Op::operator MPI_Op* () { return pmpi_op; }
|
|
|
|
|
|
#else // ============= NO PROFILING ===================================
|
|
|
|
// construction
|
|
inline
|
|
MPI::Op::Op() : mpi_op(MPI_OP_NULL) { }
|
|
|
|
inline
|
|
MPI::Op::Op(MPI_Op i) : mpi_op(i) { }
|
|
|
|
inline
|
|
MPI::Op::Op(const MPI::Op& op)
|
|
: mpi_op(op.mpi_op) { }
|
|
|
|
inline
|
|
MPI::Op::~Op()
|
|
{
|
|
#if 0
|
|
mpi_op = MPI_OP_NULL;
|
|
op_user_function = 0;
|
|
#endif
|
|
}
|
|
|
|
inline MPI::Op&
|
|
MPI::Op::operator=(const MPI::Op& op) {
|
|
mpi_op = op.mpi_op;
|
|
return *this;
|
|
}
|
|
|
|
// comparison
|
|
inline bool
|
|
MPI::Op::operator== (const MPI::Op &a) { return (bool)(mpi_op == a.mpi_op); }
|
|
|
|
inline bool
|
|
MPI::Op::operator!= (const MPI::Op &a) { return (bool)!(*this == a); }
|
|
|
|
// inter-language operability
|
|
inline MPI::Op&
|
|
MPI::Op::operator= (const MPI_Op &i) { mpi_op = i; return *this; }
|
|
|
|
inline
|
|
MPI::Op::operator MPI_Op () const { return mpi_op; }
|
|
|
|
//inline
|
|
//MPI::Op::operator MPI_Op* () { return &mpi_op; }
|
|
|
|
#endif
|
|
|
|
// Extern this function here rather than include an internal Open MPI
|
|
// header file (and therefore force installing the internal Open MPI
|
|
// header file so that user apps can #include it)
|
|
|
|
extern "C" void ompi_op_set_cxx_callback(MPI_Op op, MPI_User_function*);
|
|
|
|
// There is a lengthy comment in ompi/mpi/cxx/intercepts.cc explaining
|
|
// what this function is doing. Please read it before modifying this
|
|
// function.
|
|
inline void
|
|
MPI::Op::Init(MPI::User_function *func, bool commute)
|
|
{
|
|
(void)MPI_Op_create((MPI_User_function*) ompi_mpi_cxx_op_intercept,
|
|
(int) commute, &mpi_op);
|
|
ompi_op_set_cxx_callback(mpi_op, (MPI_User_function*) func);
|
|
}
|
|
|
|
|
|
inline void
|
|
MPI::Op::Free()
|
|
{
|
|
(void)MPI_Op_free(&mpi_op);
|
|
}
|
|
|
|
|
|
inline void
|
|
MPI::Op::Reduce_local(const void *inbuf, void *inoutbuf, int count,
|
|
const MPI::Datatype& datatype) const
|
|
{
|
|
(void)MPI_Reduce_local(const_cast<void*>(inbuf), inoutbuf, count,
|
|
datatype, mpi_op);
|
|
}
|