CSRLocalMatrixOperationsDevice.h

Go to the documentation of this file.

#ifndef included_CSRLocalMatrixOperationsDevice_H_
#define included_CSRLocalMatrixOperationsDevice_H_
 
namespace AMP::LinearAlgebra {
 
template<typename Config, typename LocalMatrixData = CSRLocalMatrixData<Config>>
class CSRLocalMatrixOperationsDevice
{
public:
    using gidx_t         = typename Config::gidx_t;
    using lidx_t         = typename Config::lidx_t;
    using scalar_t       = typename Config::scalar_t;
    using allocator_type = typename Config::allocator_type;
 
    static void mult( const scalar_t *in, std::shared_ptr<LocalMatrixData> A, scalar_t *out );
 
    static void multTranspose( const scalar_t *in,
                               std::shared_ptr<LocalMatrixData> A,
                               std::vector<scalar_t> &vvals,
                               std::vector<size_t> &rcols );
 
    static void scale( scalar_t alpha, std::shared_ptr<LocalMatrixData> A );
 
    static void
    axpy( scalar_t alpha, std::shared_ptr<LocalMatrixData> X, std::shared_ptr<LocalMatrixData> Y );
 
    static void setScalar( scalar_t alpha, std::shared_ptr<LocalMatrixData> A );
 
    static void zero( std::shared_ptr<LocalMatrixData> A );
 
    static void setDiagonal( const scalar_t *in, std::shared_ptr<LocalMatrixData> A );
 
    static void extractDiagonal( std::shared_ptr<LocalMatrixData> A, scalar_t *buf );
 
    static void setIdentity( std::shared_ptr<LocalMatrixData> A );
 
    static void LinfNorm( std::shared_ptr<LocalMatrixData> A, scalar_t *rowSums );
 
    static void copy( std::shared_ptr<const LocalMatrixData> X,
                      std::shared_ptr<LocalMatrixData> Y );
 
    template<typename ConfigIn>
    static void
    copyCast( std::shared_ptr<
                  CSRLocalMatrixData<typename ConfigIn::template set_alloc_t<Config::allocator>>> X,
              std::shared_ptr<LocalMatrixData> Y );
};
 
} // namespace AMP::LinearAlgebra
 
#endif