CSRMatrix.h

Go to the documentation of this file.

#ifndef included_AMP_CSRMatrix
#define included_AMP_CSRMatrix
 
#include "AMP/matrices/CSRConfig.h"
#include "AMP/matrices/Matrix.h"
#include "AMP/utils/Memory.h"
#include "AMP/vectors/Vector.h"
#include <memory>
 
namespace AMP::IO {
class RestartManager;
}
 
namespace AMP::LinearAlgebra {
 
template<typename Policy>
class CSRMatrixData;
 
template<class Config>
class CSRMatrix : public Matrix
{
public:
    using config_type    = Config;
    using allocator_type = typename Config::allocator_type;
    using matrixdata_t   = CSRMatrixData<Config>;
    static_assert( std::is_same_v<typename allocator_type::value_type, void> );
 
    CSRMatrix() = delete;
 
    explicit CSRMatrix( std::shared_ptr<MatrixParametersBase> params );
 
 
    explicit CSRMatrix( std::shared_ptr<MatrixData> data );
 
    CSRMatrix( const CSRMatrix & ) = delete;
 
    CSRMatrix &operator=( const CSRMatrix & ) = delete;
 
    virtual ~CSRMatrix();
 
    virtual std::string type() const override { return "CSRMatrix"; }
 
    virtual std::uint16_t mode() const override
    {
        return static_cast<std::uint16_t>( Config::mode );
    }
 
    shared_ptr transpose() const override;
 
    shared_ptr clone() const override;
 
    shared_ptr migrate( AMP::Utilities::MemoryType memType ) const;
 
    shared_ptr migrate( AMP::Utilities::MemoryType memType, AMP::Utilities::Backend backend ) const;
 
    template<typename ConfigOut>
    shared_ptr migrate() const;
 
    template<typename ConfigOut>
    shared_ptr migrate( AMP::Utilities::Backend backend ) const;
 
    void setBackend( AMP::Utilities::Backend backend ) override;
 
    Vector::shared_ptr
    extractDiagonal( Vector::shared_ptr buf = Vector::shared_ptr() ) const override;
 
    Vector::shared_ptr createInputVector() const override;
 
    Vector::shared_ptr createOutputVector() const override;
 
    CSRMatrix( int64_t fid, AMP::IO::RestartManager *manager );
 
protected:
    void multiply( shared_ptr other_op, shared_ptr &result ) override;
};
} // namespace AMP::LinearAlgebra
 
 
#endif