|
void | convert_to (Ell< next_precision< ValueType >, IndexType > *result) const override |
|
void | move_to (Ell< next_precision< ValueType >, IndexType > *result) override |
|
void | convert_to (Ell< next_precision< next_precision< ValueType > >, IndexType > *result) const override |
|
void | move_to (Ell< next_precision< next_precision< ValueType > >, IndexType > *result) override |
|
std::unique_ptr< unbatch_type > | create_view_for_item (size_type item_id) |
| Creates a mutable view (of matrix::Ell type) of one item of the batch::matrix::Ell<value_type> object.
|
|
std::unique_ptr< const unbatch_type > | create_const_view_for_item (size_type item_id) const |
| Creates a mutable view (of matrix::Ell type) of one item of the batch::matrix::Ell<value_type> object.
|
|
value_type * | get_values () noexcept |
| Returns a pointer to the array of values of the matrix.
|
|
const value_type * | get_const_values () const noexcept |
| Returns a pointer to the array of values of the matrix.
|
|
index_type * | get_col_idxs () noexcept |
| Returns a pointer to the array of column indices of the matrix.
|
|
const index_type * | get_const_col_idxs () const noexcept |
| Returns a pointer to the array of column indices of the matrix.
|
|
index_type | get_num_stored_elements_per_row () const noexcept |
| Returns the number of elements per row explicitly stored.
|
|
size_type | get_num_stored_elements () const noexcept |
| Returns the number of elements explicitly stored in the batch matrix, cumulative across all the batch items.
|
|
size_type | get_num_elements_per_item () const noexcept |
| Returns the number of stored elements in each batch item.
|
|
index_type * | get_col_idxs_for_item (size_type batch_id) noexcept |
| Returns a pointer to the array of col_idxs of the matrix.
|
|
const index_type * | get_const_col_idxs_for_item (size_type batch_id) const noexcept |
| Returns a pointer to the array of col_idxs of the matrix.
|
|
value_type * | get_values_for_item (size_type batch_id) noexcept |
| Returns a pointer to the array of values of the matrix for a specific batch item.
|
|
const value_type * | get_const_values_for_item (size_type batch_id) const noexcept |
| Returns a pointer to the array of values of the matrix for a specific batch item.
|
|
Ell * | apply (ptr_param< const MultiVector< value_type > > b, ptr_param< MultiVector< value_type > > x) |
| Apply the matrix to a multi-vector.
|
|
Ell * | apply (ptr_param< const MultiVector< value_type > > alpha, ptr_param< const MultiVector< value_type > > b, ptr_param< const MultiVector< value_type > > beta, ptr_param< MultiVector< value_type > > x) |
| Apply the matrix to a multi-vector with a linear combination of the given input vector.
|
|
const Ell * | apply (ptr_param< const MultiVector< value_type > > b, ptr_param< MultiVector< value_type > > x) const |
|
const Ell * | apply (ptr_param< const MultiVector< value_type > > alpha, ptr_param< const MultiVector< value_type > > b, ptr_param< const MultiVector< value_type > > beta, ptr_param< MultiVector< value_type > > x) const |
|
void | scale (const array< value_type > &row_scale, const array< value_type > &col_scale) |
| Performs in-place row and column scaling for this matrix.
|
|
void | add_scaled_identity (ptr_param< const MultiVector< value_type > > alpha, ptr_param< const MultiVector< value_type > > beta) |
| Performs the operation this = alpha*I + beta*this.
|
|
std::unique_ptr< Ell< default_precision, int32 > > | create_default (std::shared_ptr< const Executor > exec) const |
|
std::unique_ptr< Ell< default_precision, int32 > > | create_default () const |
|
std::unique_ptr< Ell< default_precision, int32 > > | clone (std::shared_ptr< const Executor > exec) const |
|
std::unique_ptr< Ell< default_precision, int32 > > | clone () const |
|
Ell< default_precision, int32 > * | copy_from (const PolymorphicObject *other) |
|
std::enable_if_t< std::is_base_of< PolymorphicObject, std::decay_t< Derived > >::value, Ell< default_precision, int32 > > * | copy_from (std::unique_ptr< Derived > &&other) |
|
std::enable_if_t< std::is_base_of< PolymorphicObject, std::decay_t< Derived > >::value, Ell< default_precision, int32 > > * | copy_from (const std::unique_ptr< Derived > &other) |
|
Ell< default_precision, int32 > * | copy_from (const std::shared_ptr< const PolymorphicObject > &other) |
|
Ell< default_precision, int32 > * | move_from (ptr_param< PolymorphicObject > other) |
|
Ell< default_precision, int32 > * | clear () |
|
void | convert_to (result_type *result) const override |
|
void | move_to (result_type *result) override |
|
void | convert_to (ptr_param< result_type > result) const |
|
void | move_to (ptr_param< result_type > result) |
|
virtual void | convert_to (result_type *result) const=0 |
| Converts the implementer to an object of type result_type.
|
|
void | convert_to (ptr_param< result_type > result) const |
|
virtual void | move_to (result_type *result)=0 |
| Converts the implementer to an object of type result_type by moving data from this object.
|
|
void | move_to (ptr_param< result_type > result) |
|
virtual void | convert_to (result_type *result) const=0 |
| Converts the implementer to an object of type result_type.
|
|
void | convert_to (ptr_param< result_type > result) const |
|
virtual void | move_to (result_type *result)=0 |
| Converts the implementer to an object of type result_type by moving data from this object.
|
|
void | move_to (ptr_param< result_type > result) |
|
|
static std::unique_ptr< Ell > | create (std::shared_ptr< const Executor > exec, const batch_dim< 2 > &size=batch_dim< 2 >{}, const IndexType num_elems_per_row=0) |
| Creates an uninitialized Ell matrix of the specified size.
|
|
static std::unique_ptr< Ell > | create (std::shared_ptr< const Executor > exec, const batch_dim< 2 > &size, const IndexType num_elems_per_row, array< value_type > values, array< index_type > col_idxs) |
| Creates a Ell matrix from an already allocated (and initialized) array.
|
|
template<typename InputValueType, typename ColIndexType> |
static std::unique_ptr< Ell > | create (std::shared_ptr< const Executor > exec, const batch_dim< 2 > &size, const IndexType num_elems_per_row, std::initializer_list< InputValueType > values, std::initializer_list< ColIndexType > col_idxs) |
| create(std::shared_ptr<const Executor>,const batch_dim<2>&, const IndexType, array<value_type>,array<index_type>)
|
|
static std::unique_ptr< const Ell > | create_const (std::shared_ptr< const Executor > exec, const batch_dim< 2 > &sizes, const index_type num_elems_per_row, gko::detail::const_array_view< value_type > &&values, gko::detail::const_array_view< index_type > &&col_idxs) |
| Creates a constant (immutable) batch ell matrix from a constant array.
|
|
template<typename ValueType = default_precision, typename IndexType = int32>
class gko::batch::matrix::Ell< ValueType, IndexType >
Ell is a sparse matrix format that stores the same number of nonzeros in each row, enabling coalesced accesses.
It is suitable for sparsity patterns that have a similar number of nonzeros in every row. The values are stored in a column-major fashion similar to the monolithic gko::matrix::Ell class.
Similar to the monolithic gko::matrix::Ell class, invalid_index<IndexType> is used as the column index for padded zero entries.
- Note
- It is also assumed that the sparsity pattern of all the items in the batch is the same and therefore only a single copy of the sparsity pattern is stored.
-
Currently only IndexType of int32 is supported.
- Template Parameters
-
ValueType | value precision of matrix elements |
IndexType | index precision of matrix elements |