CSRLocalMatrixOperationsDefault.h

Go to the documentation of this file.

#ifndef included_CSRLocalMatrixOperationsDefault_H_
#define included_CSRLocalMatrixOperationsDefault_H_
 
namespace AMP::LinearAlgebra {
 
template<typename Config>
class CSRLocalMatrixOperationsDefault
{
public:
    using config_type       = Config;
    using allocator_type    = typename Config::allocator_type;
    using localmatrixdata_t = CSRLocalMatrixData<Config>;
    using gidx_t            = typename Config::gidx_t;
    using lidx_t            = typename Config::lidx_t;
    using scalar_t          = typename Config::scalar_t;
 
    static_assert( std::is_same_v<typename allocator_type::value_type, void> );
 
    static void mult( const scalar_t *in, std::shared_ptr<localmatrixdata_t> A, scalar_t *out );
 
    static void multTranspose( const scalar_t *in,
                               std::shared_ptr<localmatrixdata_t> A,
                               std::vector<scalar_t> &vvals,
                               std::vector<size_t> &rcols );
 
    static void scale( scalar_t alpha, std::shared_ptr<localmatrixdata_t> A );
 
    static void scale( scalar_t alpha, const scalar_t *D, std::shared_ptr<localmatrixdata_t> A );
 
    static void scaleInv( scalar_t alpha, const scalar_t *D, std::shared_ptr<localmatrixdata_t> A );
 
    static void axpy( scalar_t alpha,
                      std::shared_ptr<localmatrixdata_t> X,
                      std::shared_ptr<localmatrixdata_t> Y );
 
    static void setScalar( scalar_t alpha, std::shared_ptr<localmatrixdata_t> A );
 
    static void zero( std::shared_ptr<localmatrixdata_t> A );
 
    static void setDiagonal( const scalar_t *in, std::shared_ptr<localmatrixdata_t> A );
 
    static void extractDiagonal( std::shared_ptr<localmatrixdata_t> A, scalar_t *buf );
 
    static void getRowSums( std::shared_ptr<localmatrixdata_t> A, scalar_t *buf );
 
    static void getRowSumsAbsolute( std::shared_ptr<localmatrixdata_t> A, scalar_t *buf );
 
    static void setIdentity( std::shared_ptr<localmatrixdata_t> A );
 
    static void copy( std::shared_ptr<const localmatrixdata_t> X,
                      std::shared_ptr<localmatrixdata_t> Y );
 
    template<typename ConfigIn>
    static void
    copyCast( std::shared_ptr<
                  CSRLocalMatrixData<typename ConfigIn::template set_alloc_t<Config::allocator>>> X,
              std::shared_ptr<localmatrixdata_t> Y );
};
 
} // namespace AMP::LinearAlgebra
 
#endif