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

Inheritance diagram for gko::batch::matrix::Dense< ValueType >:

Public Types
using	value_type = ValueType

using	index_type = int32

using	transposed_type = Dense< ValueType >

using	unbatch_type = gko::matrix::Dense< ValueType >

using	absolute_type = remove_complex< Dense >

using	complex_type = to_complex< Dense >

Public Types inherited from gko::EnablePolymorphicAssignment< ConcreteBatchLinOp >
using	result_type = ConcreteBatchLinOp

Public Types inherited from gko::ConvertibleTo< ResultType >
using	result_type = ResultType

Public Member Functions
void	convert_to (Dense< next_precision< ValueType > > *result) const override

void	move_to (Dense< next_precision< ValueType > > *result) override

void	convert_to (Dense< next_precision< next_precision< ValueType > > > *result) const override

void	move_to (Dense< next_precision< next_precision< ValueType > > > *result) override

std::unique_ptr< unbatch_type >	create_view_for_item (size_type item_id)

std::unique_ptr< const unbatch_type >	create_const_view_for_item (size_type item_id) const

size_type	get_cumulative_offset (size_type batch_id) const

value_type *	get_values () noexcept

const value_type *	get_const_values () const noexcept

value_type &	at (size_type batch_id, size_type row, size_type col)

value_type	at (size_type batch_id, size_type row, size_type col) const

ValueType &	at (size_type batch_id, size_type idx) noexcept

ValueType	at (size_type batch_id, size_type idx) const noexcept

value_type *	get_values_for_item (size_type batch_id) noexcept

const value_type *	get_const_values_for_item (size_type batch_id) const noexcept

size_type	get_num_stored_elements () const noexcept

size_type	get_num_elements_per_item () const noexcept

Dense *	apply (ptr_param< const MultiVector< value_type > > b, ptr_param< MultiVector< value_type > > x)

Dense *	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 Dense *	apply (ptr_param< const MultiVector< value_type > > b, ptr_param< MultiVector< value_type > > x) const

const Dense *	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)

void	scale_add (ptr_param< const MultiVector< value_type > > alpha, ptr_param< const batch::matrix::Dense< value_type > > b)

void	add_scaled_identity (ptr_param< const MultiVector< value_type > > alpha, ptr_param< const MultiVector< value_type > > beta)

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
PolymorphicObject &	operator= (const PolymorphicObject &)

std::unique_ptr< PolymorphicObject >	create_default (std::shared_ptr< const Executor > exec) const

std::unique_ptr< PolymorphicObject >	create_default () const

std::unique_ptr< PolymorphicObject >	clone (std::shared_ptr< const Executor > exec) const

std::unique_ptr< PolymorphicObject >	clone () const

PolymorphicObject *	copy_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)

PolymorphicObject *	copy_from (const std::shared_ptr< const PolymorphicObject > &other)

PolymorphicObject *	move_from (ptr_param< PolymorphicObject > other)

PolymorphicObject *	clear ()

std::shared_ptr< const Executor >	get_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< ConcreteBatchLinOp >
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)

Static Public Member Functions
static std::unique_ptr< Dense >	create (std::shared_ptr< const Executor > exec, const batch_dim< 2 > &size=batch_dim< 2 >{})

static std::unique_ptr< Dense >	create (std::shared_ptr< const Executor > exec, const batch_dim< 2 > &size, array< value_type > values)

template<typename InputValueType >
static std::unique_ptr< Dense >	create (std::shared_ptr< const Executor > exec, const batch_dim< 2 > &size, std::initializer_list< InputValueType > values)
	create(std::shared_ptr<const Executor>,const batch_dim<2>&, array<value_type>)

static std::unique_ptr< const Dense >	create_const (std::shared_ptr< const Executor > exec, const batch_dim< 2 > &sizes, gko::detail::const_array_view< ValueType > &&values)

Friends
class	EnablePolymorphicObject< Dense, BatchLinOp >

class	Dense< to_complex< ValueType > >

class	Dense< previous_precision< ValueType > >

class	Dense< previous_precision< previous_precision< ValueType > > >

Additional Inherited Members
Protected Member Functions inherited from gko::EnablePolymorphicObject< ConcreteObject, PolymorphicBase >
std::unique_ptr< PolymorphicObject >	create_default_impl (std::shared_ptr< const Executor > exec) const override

PolymorphicObject *	copy_from_impl (const PolymorphicObject *other) override

PolymorphicObject *	copy_from_impl (std::unique_ptr< PolymorphicObject > other) override

PolymorphicObject *	move_from_impl (PolymorphicObject *other) override

PolymorphicObject *	move_from_impl (std::unique_ptr< PolymorphicObject > other) override

PolymorphicObject *	clear_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

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

Detailed Description

template<typename ValueType = default_precision>
class gko::batch::matrix::Dense< ValueType >

Dense is a batch matrix format which explicitly stores all values of the matrix in each of the batches.

The values in each of the batches are stored in row-major format (values belonging to the same row appear consecutive in the memory and the values of each batch item are also stored consecutively in memory).

Note: Though the storage layout is the same as the multi-vector object, the class semantics and the operations it aims to provide are different. Hence it is recommended to create multi-vector objects if the user means to view the data as a set of vectors.

Template Parameters

ValueType precision of matrix elements

Member Function Documentation

◆ add_scaled_identity()

template<typename ValueType = default_precision>

void gko::batch::matrix::Dense< ValueType >::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.

Parameters

alpha	the scalar for identity
beta	the scalar to multiply this matrix

Note: Performs the operation in-place for this batch matrix

◆ apply() [1/4]

template<typename ValueType = default_precision>

Dense * gko::batch::matrix::Dense< ValueType >::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. Represents the matrix vector multiplication, x = alpha * A

b + beta * x, where x and b are both multi-vectors.

Parameters

alpha	the scalar to scale the matrix-vector product with
b	the multi-vector to be applied to
beta	the scalar to scale the x vector with
x	the output multi-vector

◆ apply() [2/4]

template<typename ValueType = default_precision>

const Dense * gko::batch::matrix::Dense< ValueType >::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

◆ apply() [3/4]

template<typename ValueType = default_precision>

Dense * gko::batch::matrix::Dense< ValueType >::apply	(	ptr_param< const MultiVector< value_type > >	b,
		ptr_param< MultiVector< value_type > >	x
	)

Apply the matrix to a multi-vector. Represents the matrix vector multiplication, x = A * b, where x and b are both multi-vectors.

Parameters

b	the multi-vector to be applied to
x	the output multi-vector

◆ apply() [4/4]

template<typename ValueType = default_precision>

const Dense * gko::batch::matrix::Dense< ValueType >::apply	(	ptr_param< const MultiVector< value_type > >	b,
		ptr_param< MultiVector< value_type > >	x
	)		const

◆ at() [1/4]

template<typename ValueType = default_precision>

ValueType gko::batch::matrix::Dense< ValueType >::at	(	size_type	batch_id,
		size_type	idx
	)		const

inlinenoexcept

Returns a single element for a particular batch item.

Parameters

batch_id	the batch item index to be queried
row	the row of the requested element
col	the column of the requested element

Note: the method has to be called on the same Executor the matrix is stored at (e.g. trying to call this method on a GPU Dense object from the OMP may result in incorrect behaviour)

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

◆ at() [2/4]

template<typename ValueType = default_precision>

ValueType & gko::batch::matrix::Dense< ValueType >::at	(	size_type	batch_id,
		size_type	idx
	)

inlinenoexcept

Returns a single element for a particular batch item.

Useful for iterating across all elements of the matrix. However, it is less efficient than the two-parameter variant of this method.

Parameters

batch_id	the batch item index to be queried
idx	a linear index of the requested element

Note: the method has to be called on the same Executor the matrix is stored at (e.g. trying to call this method on a GPU Dense object from the OMP may result in incorrect behaviour)

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

◆ at() [3/4]

template<typename ValueType = default_precision>

value_type & gko::batch::matrix::Dense< ValueType >::at	(	size_type	batch_id,
		size_type	row,
		size_type	col
	)

inline

Returns a single element for a particular batch item.

Parameters

batch_id	the batch item index to be queried
row	the row of the requested element
col	the column of the requested element

Note: the method has to be called on the same Executor the matrix is stored at (e.g. trying to call this method on a GPU Dense object from the OMP may result in incorrect behaviour)

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

◆ at() [4/4]

template<typename ValueType = default_precision>

value_type gko::batch::matrix::Dense< ValueType >::at	(	size_type	batch_id,
		size_type	row,
		size_type	col
	)		const

inline

Returns a single element for a particular batch item.

Parameters

batch_id	the batch item index to be queried
row	the row of the requested element
col	the column of the requested element

Note: the method has to be called on the same Executor the matrix is stored at (e.g. trying to call this method on a GPU Dense object from the OMP may result in incorrect behaviour)

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

◆ create() [1/3]

template<typename ValueType = default_precision>

static std::unique_ptr< Dense > gko::batch::matrix::Dense< ValueType >::create	(	std::shared_ptr< const Executor >	exec,
		const batch_dim< 2 > &	size,
		array< value_type >	values
	)

static

Creates a Dense matrix from an already allocated (and initialized) array.

Parameters

exec	Executor associated to the matrix
size	sizes of the batch matrices in a batch_dim object
values	array of matrix values

Note: If values is not an rvalue, not an array of ValueType, or is on the wrong executor, an internal copy will be created, and the original array data will not be used in the matrix.

Returns: A smart pointer to the newly created matrix.

◆ create() [2/3]

template<typename ValueType = default_precision>

template<typename InputValueType >

static std::unique_ptr< Dense > gko::batch::matrix::Dense< ValueType >::create	(	std::shared_ptr< const Executor >	exec,
		const batch_dim< 2 > &	size,
		std::initializer_list< InputValueType >	values
	)

inlinestatic

create(std::shared_ptr<const Executor>,const batch_dim<2>&, array<value_type>)

References gko::batch::matrix::Dense< ValueType >::create().

◆ create() [3/3]

template<typename ValueType = default_precision>

static std::unique_ptr< Dense > gko::batch::matrix::Dense< ValueType >::create	(	std::shared_ptr< const Executor >	exec,
		const batch_dim< 2 > &	size = `batch_dim< 2 >{}`
	)

static

Creates an uninitialized Dense matrix of the specified size.

Parameters

exec	Executor associated to the matrix
size	size of the matrix

Returns: A smart pointer to the newly created matrix.

Referenced by gko::batch::matrix::Dense< ValueType >::create().

◆ create_const()

template<typename ValueType = default_precision>

static std::unique_ptr< const Dense > gko::batch::matrix::Dense< ValueType >::create_const	(	std::shared_ptr< const Executor >	exec,
		const batch_dim< 2 > &	sizes,
		gko::detail::const_array_view< ValueType > &&	values
	)

static

Creates a constant (immutable) batch dense matrix from a constant array.

Parameters

exec	the executor to create the matrix on
size	the dimensions of the matrix
values	the value array of the matrix

Returns: A smart pointer to the constant matrix wrapping the input array (if it resides on the same executor as the matrix) or a copy of the array on the correct executor.; A smart pointer to the newly created matrix.

◆ create_const_view_for_item()

template<typename ValueType = default_precision>

std::unique_ptr< const unbatch_type > gko::batch::matrix::Dense< ValueType >::create_const_view_for_item ( size_type item_id ) const

Creates a mutable view (of gko::matrix::Dense type) of one item of the batch::matrix::Dense<value_type> object. Does not perform any deep copies, but only returns a view of the data.

Parameters

item_id The index of the batch item

Returns: a gko::matrix::Dense object with the data from the batch item at the given index.

◆ create_view_for_item()

template<typename ValueType = default_precision>

std::unique_ptr< unbatch_type > gko::batch::matrix::Dense< ValueType >::create_view_for_item ( size_type item_id )

Creates a mutable view (of gko::matrix::Dense type) of one item of the batch::matrix::Dense<value_type> object. Does not perform any deep copies, but only returns a view of the data.

Parameters

item_id The index of the batch item

Returns: a gko::matrix::Dense object with the data from the batch item at the given index.

◆ get_const_values()

template<typename ValueType = default_precision>

const value_type * gko::batch::matrix::Dense< ValueType >::get_const_values ( ) const

inlinenoexcept

Returns a pointer to the array of values of the multi-vector

Returns: the pointer to the array of values

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_for_item()

template<typename ValueType = default_precision>

const value_type * gko::batch::matrix::Dense< ValueType >::get_const_values_for_item ( size_type batch_id ) const

inlinenoexcept

Returns a pointer to the array of values of the matrix for a specific batch item.

Parameters

batch_id the id of the batch item.

Returns: the pointer to the array of values

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(), and gko::batch::matrix::Dense< ValueType >::get_cumulative_offset().

◆ get_cumulative_offset()

template<typename ValueType = default_precision>

size_type gko::batch::matrix::Dense< ValueType >::get_cumulative_offset ( size_type batch_id ) const

inline

Get the cumulative storage size offset

Parameters

batch_id the batch id

Returns: the cumulative offset

Referenced by gko::batch::matrix::Dense< ValueType >::get_const_values_for_item(), and gko::batch::matrix::Dense< ValueType >::get_values_for_item().

◆ get_num_elements_per_item()

template<typename ValueType = default_precision>

size_type gko::batch::matrix::Dense< ValueType >::get_num_elements_per_item ( ) const

inlinenoexcept

Returns the number of stored elements in each batch item.

Returns: the number of stored elements per batch item.

References gko::batch::matrix::Dense< ValueType >::get_num_stored_elements().

◆ get_num_stored_elements()

template<typename ValueType = default_precision>

size_type gko::batch::matrix::Dense< ValueType >::get_num_stored_elements ( ) const

inlinenoexcept

Returns the number of elements explicitly stored in the batch matrix, cumulative across all the batch items.

Returns: the number of elements explicitly stored in the vector, cumulative across all the batch items

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

Referenced by gko::batch::matrix::Dense< ValueType >::get_num_elements_per_item().

◆ get_values()

template<typename ValueType = default_precision>

value_type * gko::batch::matrix::Dense< ValueType >::get_values ( )

inlinenoexcept

Returns a pointer to the array of values of the multi-vector

Returns: the pointer to the array of values

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

◆ get_values_for_item()

template<typename ValueType = default_precision>

value_type * gko::batch::matrix::Dense< ValueType >::get_values_for_item ( size_type batch_id )

inlinenoexcept

Returns a pointer to the array of values of the matrix for a specific batch item.

Parameters

batch_id the id of the batch item.

Returns: the pointer to the array of values

References gko::batch::matrix::Dense< ValueType >::get_cumulative_offset(), and gko::array< ValueType >::get_data().

◆ scale()

template<typename ValueType = default_precision>

void gko::batch::matrix::Dense< ValueType >::scale	(	const array< value_type > &	row_scale,
		const array< value_type > &	col_scale
	)

Performs in-place row and column scaling for this matrix.

Parameters

row_scale	the row scalars
col_scale	the column scalars

◆ scale_add()

template<typename ValueType = default_precision>

void gko::batch::matrix::Dense< ValueType >::scale_add	(	ptr_param< const MultiVector< value_type > >	alpha,
		ptr_param< const batch::matrix::Dense< value_type > >	b
	)

Performs the operation this = alpha*this + b.

Parameters

alpha	the scalar to multiply this matrix
b	the matrix to add

Note: Performs the operation in-place for this batch matrix

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

ginkgo/core/matrix/batch_dense.hpp

Dense< ValueType > Class Template Reference

Dense< ValueType > Class Template Reference#

Public Types

Public Member Functions

Static Public Member Functions

Friends

Additional Inherited Members

Detailed Description

Member Function Documentation

◆ add_scaled_identity()

◆ apply() [1/4]

◆ apply() [2/4]

◆ apply() [3/4]

◆ apply() [4/4]

◆ at() [1/4]

◆ at() [2/4]

◆ at() [3/4]

◆ at() [4/4]

◆ create() [1/3]

◆ create() [2/3]

◆ create() [3/3]

◆ create_const()

◆ create_const_view_for_item()

◆ create_view_for_item()

◆ get_const_values()

◆ get_const_values_for_item()

◆ get_cumulative_offset()

◆ get_num_elements_per_item()

◆ get_num_stored_elements()

◆ get_values()

◆ get_values_for_item()

◆ scale()

◆ scale_add()

Dense< ValueType > Class Template Reference

Dense&lt; ValueType &gt; Class Template Reference#

Public Types

Public Member Functions

Static Public Member Functions

Friends

Additional Inherited Members

Detailed Description

Member Function Documentation

◆ add_scaled_identity()

◆ apply() [1/4]

◆ apply() [2/4]

◆ apply() [3/4]

◆ apply() [4/4]

◆ at() [1/4]

◆ at() [2/4]

◆ at() [3/4]

◆ at() [4/4]

◆ create() [1/3]

◆ create() [2/3]

◆ create() [3/3]

◆ create_const()

◆ create_const_view_for_item()

◆ create_view_for_item()

◆ get_const_values()

◆ get_const_values_for_item()

◆ get_cumulative_offset()

◆ get_num_elements_per_item()

◆ get_num_stored_elements()

◆ get_values()

◆ get_values_for_item()

◆ scale()

◆ scale_add()

Dense< ValueType > Class Template Reference#