Stokhos Package Browser (Single Doxygen Collection) Version of the Day
Loading...
Searching...
No Matches
Stokhos::MultiplyImpl< ValueType, MatrixValue, VectorValue, Device > Class Template Reference

#include <Stokhos_CrsProductTensor.hpp>

Inheritance diagram for Stokhos::MultiplyImpl< ValueType, MatrixValue, VectorValue, Device >:

Public Types

typedef Device execution_space
typedef CrsProductTensor< ValueType, execution_spacetensor_type
typedef StochasticProductTensor< ValueType, tensor_type, execution_spaceBlockSpec
typedef BlockSpec::size_type size_type
typedef Kokkos::View< VectorValue **, Kokkos::LayoutLeft, execution_spaceblock_vector_type
typedef BlockCrsMatrix< BlockSpec, MatrixValue, execution_spacematrix_type

Public Member Functions

 MultiplyImpl (const matrix_type &A, const block_vector_type &x, const block_vector_type &y)
KOKKOS_INLINE_FUNCTION void operator() (const size_type iBlockRow) const
KOKKOS_INLINE_FUNCTION std::pair< size_type, size_typecompute_work_range (const size_type work_count, const size_type thread_count, const size_type thread_rank) const
KOKKOS_INLINE_FUNCTION void operator() (const typename Kokkos::TeamPolicy< execution_space >::member_type &devicedevice) const

Static Public Member Functions

static void apply (const matrix_type &A, const block_vector_type &x, const block_vector_type &y)

Public Attributes

const matrix_type m_A
const block_vector_type m_x
const block_vector_type m_y

Detailed Description

template<typename ValueType, typename MatrixValue, typename VectorValue, typename Device>
class Stokhos::MultiplyImpl< ValueType, MatrixValue, VectorValue, Device >

Definition at line 757 of file Stokhos_CrsProductTensor.hpp.

Member Typedef Documentation

◆ execution_space

template<typename ValueType, typename MatrixValue, typename VectorValue, typename Device>
typedef Device Stokhos::MultiplyImpl< ValueType, MatrixValue, VectorValue, Device >::execution_space

Definition at line 760 of file Stokhos_CrsProductTensor.hpp.

◆ tensor_type

template<typename ValueType, typename MatrixValue, typename VectorValue, typename Device>
typedef CrsProductTensor< ValueType , execution_space > Stokhos::MultiplyImpl< ValueType, MatrixValue, VectorValue, Device >::tensor_type

Definition at line 761 of file Stokhos_CrsProductTensor.hpp.

◆ BlockSpec

template<typename ValueType, typename MatrixValue, typename VectorValue, typename Device>
typedef StochasticProductTensor< ValueType, tensor_type, execution_space > Stokhos::MultiplyImpl< ValueType, MatrixValue, VectorValue, Device >::BlockSpec

Definition at line 762 of file Stokhos_CrsProductTensor.hpp.

◆ size_type

template<typename ValueType, typename MatrixValue, typename VectorValue, typename Device>
typedef BlockSpec::size_type Stokhos::MultiplyImpl< ValueType, MatrixValue, VectorValue, Device >::size_type

Definition at line 763 of file Stokhos_CrsProductTensor.hpp.

◆ block_vector_type

template<typename ValueType, typename MatrixValue, typename VectorValue, typename Device>
typedef Kokkos::View< VectorValue** , Kokkos::LayoutLeft , execution_space > Stokhos::MultiplyImpl< ValueType, MatrixValue, VectorValue, Device >::block_vector_type

Definition at line 764 of file Stokhos_CrsProductTensor.hpp.

◆ matrix_type

template<typename ValueType, typename MatrixValue, typename VectorValue, typename Device>
typedef BlockCrsMatrix< BlockSpec , MatrixValue , execution_space > Stokhos::MultiplyImpl< ValueType, MatrixValue, VectorValue, Device >::matrix_type

Definition at line 765 of file Stokhos_CrsProductTensor.hpp.

Constructor & Destructor Documentation

◆ MultiplyImpl()

template<typename ValueType, typename MatrixValue, typename VectorValue, typename Device>
Stokhos::MultiplyImpl< ValueType, MatrixValue, VectorValue, Device >::MultiplyImpl ( const matrix_type & A,
const block_vector_type & x,
const block_vector_type & y )
inline

Definition at line 771 of file Stokhos_CrsProductTensor.hpp.

Member Function Documentation

◆ operator()() [1/2]

template<typename ValueType, typename MatrixValue, typename VectorValue, typename Device>
KOKKOS_INLINE_FUNCTION void Stokhos::MultiplyImpl< ValueType, MatrixValue, VectorValue, Device >::operator() ( const size_type iBlockRow) const
inline

Definition at line 786 of file Stokhos_CrsProductTensor.hpp.

◆ compute_work_range()

template<typename ValueType, typename MatrixValue, typename VectorValue, typename Device>
KOKKOS_INLINE_FUNCTION std::pair< size_type, size_type > Stokhos::MultiplyImpl< ValueType, MatrixValue, VectorValue, Device >::compute_work_range ( const size_type work_count,
const size_type thread_count,
const size_type thread_rank ) const
inline

Definition at line 813 of file Stokhos_CrsProductTensor.hpp.

◆ operator()() [2/2]

template<typename ValueType, typename MatrixValue, typename VectorValue, typename Device>
KOKKOS_INLINE_FUNCTION void Stokhos::MultiplyImpl< ValueType, MatrixValue, VectorValue, Device >::operator() ( const typename Kokkos::TeamPolicy< execution_space >::member_type & device) const
inline

Definition at line 958 of file Stokhos_CrsProductTensor.hpp.

◆ apply()

template<typename ValueType, typename MatrixValue, typename VectorValue, typename Device>
void Stokhos::MultiplyImpl< ValueType, MatrixValue, VectorValue, Device >::apply ( const matrix_type & A,
const block_vector_type & x,
const block_vector_type & y )
inlinestatic

Definition at line 1065 of file Stokhos_CrsProductTensor.hpp.

Member Data Documentation

◆ m_A

template<typename ValueType, typename MatrixValue, typename VectorValue, typename Device>
const matrix_type Stokhos::MultiplyImpl< ValueType, MatrixValue, VectorValue, Device >::m_A

Definition at line 767 of file Stokhos_CrsProductTensor.hpp.

◆ m_x

template<typename ValueType, typename MatrixValue, typename VectorValue, typename Device>
const block_vector_type Stokhos::MultiplyImpl< ValueType, MatrixValue, VectorValue, Device >::m_x

Definition at line 768 of file Stokhos_CrsProductTensor.hpp.

◆ m_y

template<typename ValueType, typename MatrixValue, typename VectorValue, typename Device>
const block_vector_type Stokhos::MultiplyImpl< ValueType, MatrixValue, VectorValue, Device >::m_y

Definition at line 769 of file Stokhos_CrsProductTensor.hpp.


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