Teuchos_MpiReductionOpSetter.hpp

Go to the documentation of this file.

// @HEADER
// *****************************************************************************
//                    Teuchos: Common Tools Package
//
// Copyright 2004 NTESS and the Teuchos contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER

#ifndef TEUCHOS_MPI_REDUCTION_OP_SETTER_HPP
#define TEUCHOS_MPI_REDUCTION_OP_SETTER_HPP


#include <Teuchos_ConfigDefs.hpp>

#ifdef HAVE_MPI
#include "Teuchos_ReductionOp.hpp"
#include <mpi.h>

namespace Teuchos {
namespace Details {

class TEUCHOSCOMM_LIB_DLL_EXPORT MpiReductionOpBase :
    virtual public Describable {
public:
  virtual void
  reduce (void* invec, void* inoutvec,
          int* len, MPI_Datatype* datatype) const = 0;
};

template<typename OrdinalType>
class MpiReductionOp : public MpiReductionOpBase {
public:
  MpiReductionOp (const ValueTypeReductionOp<OrdinalType,char>& reductOp)
    : reductOp_ (reductOp)
  {}

  void
  reduce (void* invec, void* inoutvec, int* len, MPI_Datatype* datatype) const
  {
#ifdef TEUCHOS_DEBUG
    TEUCHOS_TEST_FOR_EXCEPT(!len);
    TEUCHOS_TEST_FOR_EXCEPT(!datatype);
#endif
    // mfh 23 Nov 2014: Ross made the unfortunate decision initially
    // to mash everything into MPI_CHAR.  This means that we have to
    // do a lot of type casting here.  Of course, most implementations
    // of ValueTypeReductionOp will immediately cast back from char*
    // to the actual type of interest.
    int sz;
    MPI_Type_size (*datatype, &sz);
    (void) datatype;
    reductOp_.reduce ((*len) * sz, reinterpret_cast<char*> (invec),
                      reinterpret_cast<char*> (inoutvec));
  }

private:
  const ValueTypeReductionOp<OrdinalType, char>& reductOp_;
  // Not defined and not to be called
  MpiReductionOp ();
  MpiReductionOp (const MpiReductionOp&);
  MpiReductionOp& operator= (const MpiReductionOp&);
};

TEUCHOSCOMM_LIB_DLL_EXPORT MPI_Op setMpiReductionOp (const MpiReductionOpBase& reductOp);

} // namespace Details
} // namespace Teuchos

#endif // HAVE_MPI

#endif // TEUCHOS_MPI_REDUCTION_OP_SETTER_HPP