TpetraVectorOperations.h

Go to the documentation of this file.

#ifndef included_AMP_NativeTpetraVectorOperations_H_
#define included_AMP_NativeTpetraVectorOperations_H_
 
#include "AMP/vectors/Scalar.h"
#include "AMP/vectors/Vector.h"
#include "AMP/vectors/operations/VectorOperations.h"
 
namespace AMP::LinearAlgebra {
 
 
template<typename ST = double,
         typename LO = int32_t,
         typename GO = int64_t,
         typename NT = Tpetra::Vector<>::node_type>
class TpetraVectorOperations : public VectorOperations
{
public:
    TpetraVectorOperations()          = default;
    virtual ~TpetraVectorOperations() = default;
 
    std::string VectorOpName() const override { return "TpetraVectorOperations"; }
    std::shared_ptr<VectorOperations> cloneOperations() const override
    {
        return std::make_shared<TpetraVectorOperations<ST, LO, GO, NT>>();
    }
    //  function that operate on VectorData
    void setToScalar( const Scalar &alpha, VectorData &z ) override;
    void setMin( const Scalar &alpha, VectorData &z ) override;
    void setMax( const Scalar &alpha, VectorData &z ) override;
    void addScalar( const VectorData &, const Scalar &, VectorData & ) override;
    void zero( VectorData &x ) override { setToScalar( static_cast<ST>( 0.0 ), x ); }
    void setRandomValues( VectorData &x ) override;
    void copy( const VectorData &x, VectorData &z ) override;
    void copyCast( const VectorData &, VectorData & ) override { AMP_ERROR( "Not implemented!" ); }
    void scale( const Scalar &alpha, const VectorData &x, VectorData &y ) override;
    void scale( const Scalar &alpha, VectorData &x ) override;
    void add( const VectorData &x, const VectorData &y, VectorData &z ) override;
    void subtract( const VectorData &x, const VectorData &y, VectorData &z ) override;
    void multiply( const VectorData &x, const VectorData &y, VectorData &z ) override;
    void divide( const VectorData &x, const VectorData &y, VectorData &z ) override;
    void reciprocal( const VectorData &x, VectorData &y ) override;
    void linearSum( const Scalar &alpha,
                    const VectorData &x,
                    const Scalar &beta,
                    const VectorData &y,
                    VectorData &z ) override;
    void
    axpy( const Scalar &alpha, const VectorData &x, const VectorData &y, VectorData &z ) override;
    void
    axpby( const Scalar &alpha, const Scalar &beta, const VectorData &x, VectorData &y ) override;
    void abs( const VectorData &x, VectorData &z ) override;
 
    Scalar min( const VectorData &x ) const override;
    Scalar max( const VectorData &x ) const override;
    Scalar L1Norm( const VectorData &x ) const override;
    Scalar L2Norm( const VectorData &x ) const override;
    Scalar maxNorm( const VectorData &x ) const override;
    Scalar dot( const VectorData &x, const VectorData &y ) const override;
 
    Scalar localL1Norm( const AMP::LinearAlgebra::VectorData &x ) const override;
    Scalar localL2Norm( const AMP::LinearAlgebra::VectorData &x ) const override;
    Scalar localMaxNorm( const AMP::LinearAlgebra::VectorData &x ) const override;
    Scalar localDot( const AMP::LinearAlgebra::VectorData &x, const VectorData &y ) const override;
 
    Scalar localMin( const AMP::LinearAlgebra::VectorData & ) const override;
    Scalar localMax( const AMP::LinearAlgebra::VectorData & ) const override;
    Scalar localSum( const AMP::LinearAlgebra::VectorData & ) const override;
    Scalar localMinQuotient( const AMP::LinearAlgebra::VectorData &,
                             const AMP::LinearAlgebra::VectorData & ) const override;
    Scalar localWrmsNorm( const AMP::LinearAlgebra::VectorData &,
                          const AMP::LinearAlgebra::VectorData & ) const override;
    Scalar localWrmsNormMask( const AMP::LinearAlgebra::VectorData &,
                              const AMP::LinearAlgebra::VectorData &,
                              const AMP::LinearAlgebra::VectorData & ) const override;
    bool localEquals( const AMP::LinearAlgebra::VectorData &,
                      const AMP::LinearAlgebra::VectorData &,
                      const AMP::Scalar & ) const override;
};
 
 
} // namespace AMP::LinearAlgebra
 
#endif