Stokhos Package Browser (Single Doxygen Collection) Version of the Day
Loading...
Searching...
No Matches
Public Types | Public Member Functions | Static Public Member Functions | Public Attributes | List of all members
Stokhos::MultiplyImpl< ValueType, MatrixValue, VectorValue, Device > Class Template Reference

#include <Stokhos_CrsProductTensor.hpp>

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 &device) 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 >
static 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: