Coo< ValueType, IndexType > Class Template Reference

Coo&lt; ValueType, IndexType &gt; Class Template Reference#

Reference API: gko::matrix::Coo< ValueType, IndexType > Class Template Reference
Reference API

#include <ginkgo/core/matrix/coo.hpp>

Inheritance diagram for gko::matrix::Coo< ValueType, IndexType >:
[legend]

Public Types

using value_type = ValueType
 
using index_type = IndexType
 
using mat_data = matrix_data< ValueType, IndexType >
 
using device_mat_data = device_matrix_data< ValueType, IndexType >
 
using absolute_type = remove_complex< Coo >
 
- Public Types inherited from gko::EnablePolymorphicAssignment< ConcreteLinOp >
using result_type = ConcreteLinOp
 
- Public Types inherited from gko::ConvertibleTo< ResultType >
using result_type = ResultType
 
- Public Types inherited from gko::DiagonalExtractable< ValueType >
using value_type = ValueType
 
- Public Types inherited from gko::ReadableFromMatrixData< ValueType, IndexType >
using value_type = ValueType
 
using index_type = IndexType
 
- Public Types inherited from gko::WritableToMatrixData< ValueType, IndexType >
using value_type = ValueType
 
using index_type = IndexType
 
- Public Types inherited from gko::EnableAbsoluteComputation< AbsoluteLinOp >
using absolute_type = AbsoluteLinOp
 

Public Member Functions

void convert_to (Coo< next_precision< ValueType >, IndexType > *result) const override
 
void move_to (Coo< next_precision< ValueType >, IndexType > *result) override
 
void convert_to (Coo< next_precision< next_precision< ValueType > >, IndexType > *result) const override
 
void move_to (Coo< next_precision< next_precision< ValueType > >, IndexType > *result) override
 
void convert_to (Csr< ValueType, IndexType > *other) const override
 
void move_to (Csr< ValueType, IndexType > *other) override
 
void convert_to (Dense< ValueType > *other) const override
 
void move_to (Dense< ValueType > *other) override
 
void read (const mat_data &data) override
 
void read (const device_mat_data &data) override
 
void read (device_mat_data &&data) override
 
void write (mat_data &data) const override
 
std::unique_ptr< Diagonal< ValueType > > extract_diagonal () const override
 
std::unique_ptr< absolute_type > compute_absolute () const override
 
void compute_absolute_inplace () override
 
value_type * get_values () noexcept
 
const value_type * get_const_values () const noexcept
 
index_type * get_col_idxs () noexcept
 
const index_type * get_const_col_idxs () const noexcept
 
index_type * get_row_idxs () noexcept
 
const index_type * get_const_row_idxs () const noexcept
 
size_type get_num_stored_elements () const noexcept
 
LinOpapply2 (ptr_param< const LinOp > b, ptr_param< LinOp > x)
 
const LinOpapply2 (ptr_param< const LinOp > b, ptr_param< LinOp > x) const
 
LinOpapply2 (ptr_param< const LinOp > alpha, ptr_param< const LinOp > b, ptr_param< LinOp > x)
 
const LinOpapply2 (ptr_param< const LinOp > alpha, ptr_param< const LinOp > b, ptr_param< LinOp > x) const
 
- Public Member Functions inherited from gko::EnableLinOp< ConcreteLinOp, PolymorphicBase >
const ConcreteLinOp * apply (ptr_param< const LinOp > b, ptr_param< LinOp > x) const
 
ConcreteLinOp * apply (ptr_param< const LinOp > b, ptr_param< LinOp > x)
 
const ConcreteLinOp * apply (ptr_param< const LinOp > alpha, ptr_param< const LinOp > b, ptr_param< const LinOp > beta, ptr_param< LinOp > x) const
 
ConcreteLinOp * apply (ptr_param< const LinOp > alpha, ptr_param< const LinOp > b, ptr_param< const LinOp > beta, ptr_param< LinOp > x)
 
- Public Member Functions inherited from gko::EnableAbstractPolymorphicObject< AbstractObject, PolymorphicBase >
std::unique_ptr< AbstractObject > create_default (std::shared_ptr< const Executor > exec) const
 
std::unique_ptr< AbstractObject > create_default () const
 
std::unique_ptr< AbstractObject > clone (std::shared_ptr< const Executor > exec) const
 
std::unique_ptr< AbstractObject > clone () const
 
AbstractObject * copy_from (const PolymorphicObject *other)
 
template<typename Derived >
std::enable_if_t< std::is_base_of< PolymorphicObject, std::decay_t< Derived > >::value, AbstractObject > * copy_from (std::unique_ptr< Derived > &&other)
 
template<typename Derived >
std::enable_if_t< std::is_base_of< PolymorphicObject, std::decay_t< Derived > >::value, AbstractObject > * copy_from (const std::unique_ptr< Derived > &other)
 
AbstractObject * copy_from (const std::shared_ptr< const PolymorphicObject > &other)
 
AbstractObject * move_from (ptr_param< PolymorphicObject > other)
 
AbstractObject * clear ()
 
- Public Member Functions inherited from gko::PolymorphicObject
PolymorphicObjectoperator= (const PolymorphicObject &)
 
std::unique_ptr< PolymorphicObjectcreate_default (std::shared_ptr< const Executor > exec) const
 
std::unique_ptr< PolymorphicObjectcreate_default () const
 
std::unique_ptr< PolymorphicObjectclone (std::shared_ptr< const Executor > exec) const
 
std::unique_ptr< PolymorphicObjectclone () const
 
PolymorphicObjectcopy_from (const PolymorphicObject *other)
 
template<typename Derived , typename Deleter >
std::enable_if_t< std::is_base_of< PolymorphicObject, std::decay_t< Derived > >::value, PolymorphicObject > * copy_from (std::unique_ptr< Derived, Deleter > &&other)
 
template<typename Derived , typename Deleter >
std::enable_if_t< std::is_base_of< PolymorphicObject, std::decay_t< Derived > >::value, PolymorphicObject > * copy_from (const std::unique_ptr< Derived, Deleter > &other)
 
PolymorphicObjectcopy_from (const std::shared_ptr< const PolymorphicObject > &other)
 
PolymorphicObjectmove_from (ptr_param< PolymorphicObject > other)
 
PolymorphicObjectclear ()
 
std::shared_ptr< const Executorget_executor () const noexcept
 
- Public Member Functions inherited from gko::log::EnableLogging< PolymorphicObject >
void add_logger (std::shared_ptr< const Logger > logger) override
 
void remove_logger (const Logger *logger) override
 
void remove_logger (ptr_param< const Logger > logger)
 
const std::vector< std::shared_ptr< const Logger > > & get_loggers () const override
 
void clear_loggers () override
 
- Public Member Functions inherited from gko::log::Loggable
void remove_logger (ptr_param< const Logger > logger)
 
- Public Member Functions inherited from gko::EnablePolymorphicAssignment< ConcreteLinOp >
void convert_to (result_type *result) const override
 
void move_to (result_type *result) override
 
- Public Member Functions inherited from gko::ConvertibleTo< ResultType >
virtual void convert_to (result_type *result) const =0
 
void convert_to (ptr_param< result_type > result) const
 
virtual void move_to (result_type *result)=0
 
void move_to (ptr_param< result_type > result)
 
- Public Member Functions inherited from gko::DiagonalExtractable< ValueType >
std::unique_ptr< LinOpextract_diagonal_linop () const override
 
- Public Member Functions inherited from gko::ReadableFromMatrixData< ValueType, IndexType >
void read (const matrix_assembly_data< ValueType, IndexType > &data)
 
- Public Member Functions inherited from gko::EnableAbsoluteComputation< AbsoluteLinOp >
std::unique_ptr< LinOpcompute_absolute_linop () const override
 

Static Public Member Functions

static std::unique_ptr< Coocreate (std::shared_ptr< const Executor > exec, const dim< 2 > &size=dim< 2 >{}, size_type num_nonzeros={})
 
static std::unique_ptr< Coocreate (std::shared_ptr< const Executor > exec, const dim< 2 > &size, array< value_type > values, array< index_type > col_idxs, array< index_type > row_idxs)
 
template<typename InputValueType , typename InputColumnIndexType , typename InputRowIndexType >
static std::unique_ptr< Coocreate (std::shared_ptr< const Executor > exec, const dim< 2 > &size, std::initializer_list< InputValueType > values, std::initializer_list< InputColumnIndexType > col_idxs, std::initializer_list< InputRowIndexType > row_idxs)
 create(std::shared_ptr<const Executor>,const dim<2>&, array<value_type>, array<index_type>, array<index_type>)
 
static std::unique_ptr< const Coocreate_const (std::shared_ptr< const Executor > exec, const dim< 2 > &size, gko::detail::const_array_view< ValueType > &&values, gko::detail::const_array_view< IndexType > &&col_idxs, gko::detail::const_array_view< IndexType > &&row_idxs)
 

Protected Member Functions

 Coo (std::shared_ptr< const Executor > exec, const dim< 2 > &size=dim< 2 >{}, size_type num_nonzeros={})
 
 Coo (std::shared_ptr< const Executor > exec, const dim< 2 > &size, array< value_type > values, array< index_type > col_idxs, array< index_type > row_idxs)
 
void resize (dim< 2 > new_size, size_type nnz)
 
void apply_impl (const LinOp *b, LinOp *x) const override
 
void apply_impl (const LinOp *alpha, const LinOp *b, const LinOp *beta, LinOp *x) const override
 
void apply2_impl (const LinOp *b, LinOp *x) const
 
void apply2_impl (const LinOp *alpha, const LinOp *b, LinOp *x) const
 
- Protected Member Functions inherited from gko::EnableLinOp< ConcreteLinOp, PolymorphicBase >
ConcreteLinOp * self () noexcept
 
const ConcreteLinOp * self () const noexcept
 
- Protected Member Functions inherited from gko::EnablePolymorphicObject< ConcreteObject, PolymorphicBase >
std::unique_ptr< PolymorphicObjectcreate_default_impl (std::shared_ptr< const Executor > exec) const override
 
PolymorphicObjectcopy_from_impl (const PolymorphicObject *other) override
 
PolymorphicObjectcopy_from_impl (std::unique_ptr< PolymorphicObject > other) override
 
PolymorphicObjectmove_from_impl (PolymorphicObject *other) override
 
PolymorphicObjectmove_from_impl (std::unique_ptr< PolymorphicObject > other) override
 
PolymorphicObjectclear_impl () override
 
- Protected Member Functions inherited from gko::PolymorphicObject
 PolymorphicObject (std::shared_ptr< const Executor > exec)
 
 PolymorphicObject (const PolymorphicObject &other)
 
- Protected Member Functions inherited from gko::log::EnableLogging< PolymorphicObject >
void log (Params &&... params) const
 

Friends

class EnablePolymorphicObject< Coo, LinOp >
 
class Csr< ValueType, IndexType >
 
class Dense< ValueType >
 
class CooBuilder< ValueType, IndexType >
 
class Coo< to_complex< ValueType >, IndexType >
 
class Hybrid< ValueType, IndexType >
 
class Coo< previous_precision< ValueType >, IndexType >
 
class Coo< previous_precision< previous_precision< ValueType > >, IndexType >
 

Additional Inherited Members

- Protected Attributes inherited from gko::log::EnableLogging< PolymorphicObject >
std::vector< std::shared_ptr< const Logger > > loggers_
 

Detailed Description

template<typename ValueType = default_precision, typename IndexType = int32>
class gko::matrix::Coo< ValueType, IndexType >

COO stores a matrix in the coordinate matrix format.

The nonzero elements are stored in an array row-wise (but not necessarily sorted by column index within a row). Two extra arrays contain the row and column indexes of each nonzero element of the matrix.

Template Parameters
ValueTypeprecision of matrix elements
IndexTypeprecision of matrix indexes

Member Function Documentation

◆ apply2() [1/4]

template<typename ValueType = default_precision, typename IndexType = int32>
LinOp * gko::matrix::Coo< ValueType, IndexType >::apply2 ( ptr_param< const LinOp alpha,
ptr_param< const LinOp b,
ptr_param< LinOp x 
)

Performs the operation x = alpha * Coo * b + x.

Parameters
alphascaling of the result of Coo * b
bvector(s) on which the operator is applied
xoutput vector(s)
Returns
this

◆ apply2() [2/4]

template<typename ValueType = default_precision, typename IndexType = int32>
const LinOp * gko::matrix::Coo< ValueType, IndexType >::apply2 ( ptr_param< const LinOp alpha,
ptr_param< const LinOp b,
ptr_param< LinOp x 
) const

◆ apply2() [3/4]

template<typename ValueType = default_precision, typename IndexType = int32>
LinOp * gko::matrix::Coo< ValueType, IndexType >::apply2 ( ptr_param< const LinOp b,
ptr_param< LinOp x 
)

Applies Coo matrix axpy to a vector (or a sequence of vectors).

Performs the operation x = Coo * b + x

Parameters
bthe input vector(s) on which the operator is applied
xthe output vector(s) where the result is stored
Returns
this

◆ apply2() [4/4]

template<typename ValueType = default_precision, typename IndexType = int32>
const LinOp * gko::matrix::Coo< ValueType, IndexType >::apply2 ( ptr_param< const LinOp b,
ptr_param< LinOp x 
) const

◆ compute_absolute()

template<typename ValueType = default_precision, typename IndexType = int32>
std::unique_ptr< absolute_type > gko::matrix::Coo< ValueType, IndexType >::compute_absolute ( ) const
overridevirtual

Gets the AbsoluteLinOp

Returns
a pointer to the new absolute object

Implements gko::EnableAbsoluteComputation< AbsoluteLinOp >.

◆ compute_absolute_inplace()

template<typename ValueType = default_precision, typename IndexType = int32>
void gko::matrix::Coo< ValueType, IndexType >::compute_absolute_inplace ( )
overridevirtual

Compute absolute inplace on each element.

Implements gko::AbsoluteComputable.

◆ create() [1/3]

template<typename ValueType = default_precision, typename IndexType = int32>
static std::unique_ptr< Coo > gko::matrix::Coo< ValueType, IndexType >::create ( std::shared_ptr< const Executor exec,
const dim< 2 > &  size,
array< value_type >  values,
array< index_type >  col_idxs,
array< index_type >  row_idxs 
)
static

Creates a COO matrix from already allocated (and initialized) row index, column index and value arrays.

Parameters
execExecutor associated to the matrix
sizesize of the matrix
valuesarray of matrix values
col_idxsarray of column indexes
row_idxsarray of row pointers
Note
If one of row_idxs, col_idxs or values is not an rvalue, not an array of IndexType, IndexType and ValueType, respectively, or is on the wrong executor, an internal copy of that array will be created, and the original array data will not be used in the matrix.
Returns
A smart pointer to the matrix wrapping the input arrays (if they reside on the same executor as the matrix) or a copy of these arrays on the correct executor.

◆ create() [2/3]

template<typename ValueType = default_precision, typename IndexType = int32>
template<typename InputValueType , typename InputColumnIndexType , typename InputRowIndexType >
static std::unique_ptr< Coo > gko::matrix::Coo< ValueType, IndexType >::create ( std::shared_ptr< const Executor exec,
const dim< 2 > &  size,
std::initializer_list< InputValueType >  values,
std::initializer_list< InputColumnIndexType >  col_idxs,
std::initializer_list< InputRowIndexType >  row_idxs 
)
inlinestatic

◆ create() [3/3]

template<typename ValueType = default_precision, typename IndexType = int32>
static std::unique_ptr< Coo > gko::matrix::Coo< ValueType, IndexType >::create ( std::shared_ptr< const Executor exec,
const dim< 2 > &  size = dim< 2 >{},
size_type  num_nonzeros = {} 
)
static

Creates an uninitialized COO matrix of the specified size.

Parameters
execExecutor associated to the matrix
sizesize of the matrix
num_nonzerosnumber of nonzeros
Returns
A smart pointer to the newly created matrix.

Referenced by gko::matrix::Coo< ValueType, IndexType >::create().

◆ create_const()

template<typename ValueType = default_precision, typename IndexType = int32>
static std::unique_ptr< const Coo > gko::matrix::Coo< ValueType, IndexType >::create_const ( std::shared_ptr< const Executor exec,
const dim< 2 > &  size,
gko::detail::const_array_view< ValueType > &&  values,
gko::detail::const_array_view< IndexType > &&  col_idxs,
gko::detail::const_array_view< IndexType > &&  row_idxs 
)
static

Creates a constant (immutable) Coo matrix from a set of constant arrays.

Parameters
execthe executor to create the matrix on
sizethe dimensions of the matrix
valuesthe value array of the matrix
col_idxsthe column index array of the matrix
row_ptrsthe row index array of the matrix
Returns
A smart pointer to the constant matrix wrapping the input arrays (if they reside on the same executor as the matrix) or a copy of these arrays on the correct executor.

◆ extract_diagonal()

template<typename ValueType = default_precision, typename IndexType = int32>
std::unique_ptr< Diagonal< ValueType > > gko::matrix::Coo< ValueType, IndexType >::extract_diagonal ( ) const
overridevirtual

Extracts the diagonal entries of the matrix into a vector.

Parameters
diagthe vector into which the diagonal will be written

Implements gko::DiagonalExtractable< ValueType >.

◆ get_col_idxs()

template<typename ValueType = default_precision, typename IndexType = int32>
index_type * gko::matrix::Coo< ValueType, IndexType >::get_col_idxs ( )
inlinenoexcept

Returns the column indexes of the matrix.

Returns
the column indexes of the matrix.

References gko::array< ValueType >::get_data().

◆ get_const_col_idxs()

template<typename ValueType = default_precision, typename IndexType = int32>
const index_type * gko::matrix::Coo< ValueType, IndexType >::get_const_col_idxs ( ) const
inlinenoexcept

Returns the column indexes of the matrix.

Returns
the column indexes of the matrix.
Note
This is the constant version of the function, which can be significantly more memory efficient than the non-constant version, so always prefer this version.

References gko::array< ValueType >::get_const_data().

◆ get_const_row_idxs()

template<typename ValueType = default_precision, typename IndexType = int32>
const index_type * gko::matrix::Coo< ValueType, IndexType >::get_const_row_idxs ( ) const
inlinenoexcept

Note
This is the constant version of the function, which can be significantly more memory efficient than the non-constant version, so always prefer this version.

References gko::array< ValueType >::get_const_data().

◆ get_const_values()

template<typename ValueType = default_precision, typename IndexType = int32>
const value_type * gko::matrix::Coo< ValueType, IndexType >::get_const_values ( ) const
inlinenoexcept

Returns the values of the matrix.

Returns
the values of the matrix.
Note
This is the constant version of the function, which can be significantly more memory efficient than the non-constant version, so always prefer this version.

References gko::array< ValueType >::get_const_data().

◆ get_num_stored_elements()

template<typename ValueType = default_precision, typename IndexType = int32>
size_type gko::matrix::Coo< ValueType, IndexType >::get_num_stored_elements ( ) const
inlinenoexcept

Returns the number of elements explicitly stored in the matrix.

Returns
the number of elements explicitly stored in the matrix

References gko::array< ValueType >::get_size().

◆ get_row_idxs()

template<typename ValueType = default_precision, typename IndexType = int32>
index_type * gko::matrix::Coo< ValueType, IndexType >::get_row_idxs ( )
inlinenoexcept

Returns the row indexes of the matrix.

Returns
the row indexes of the matrix.

References gko::array< ValueType >::get_data().

◆ get_values()

template<typename ValueType = default_precision, typename IndexType = int32>
value_type * gko::matrix::Coo< ValueType, IndexType >::get_values ( )
inlinenoexcept

Returns the values of the matrix.

Returns
the values of the matrix.

References gko::array< ValueType >::get_data().

◆ read() [1/3]

template<typename ValueType = default_precision, typename IndexType = int32>
void gko::matrix::Coo< ValueType, IndexType >::read ( const device_mat_data data)
overridevirtual

Reads a matrix from a device_matrix_data structure.

Parameters
datathe device_matrix_data structure.

Reimplemented from gko::ReadableFromMatrixData< ValueType, IndexType >.

◆ read() [2/3]

template<typename ValueType = default_precision, typename IndexType = int32>
void gko::matrix::Coo< ValueType, IndexType >::read ( const mat_data data)
overridevirtual

Reads a matrix from a matrix_data structure.

Parameters
datathe matrix_data structure

Implements gko::ReadableFromMatrixData< ValueType, IndexType >.

◆ read() [3/3]

template<typename ValueType = default_precision, typename IndexType = int32>
void gko::matrix::Coo< ValueType, IndexType >::read ( device_mat_data &&  data)
overridevirtual

Reads a matrix from a device_matrix_data structure. The structure may be emptied by this function.

Parameters
datathe device_matrix_data structure.

Reimplemented from gko::ReadableFromMatrixData< ValueType, IndexType >.

◆ resize()

template<typename ValueType = default_precision, typename IndexType = int32>
void gko::matrix::Coo< ValueType, IndexType >::resize ( dim< 2 >  new_size,
size_type  nnz 
)
protected

Resizes the matrix and associated storage to the given sizes. Internal storage may be reallocated if they don't match the old values.

Parameters
new_sizethe new matrix dimensions.
nnzthe new number of nonzeros.

◆ write()

template<typename ValueType = default_precision, typename IndexType = int32>
void gko::matrix::Coo< ValueType, IndexType >::write ( mat_data data) const
overridevirtual

Writes a matrix to a matrix_data structure.

Parameters
datathe matrix_data structure

Implements gko::WritableToMatrixData< ValueType, IndexType >.


The documentation for this class was generated from the following file: