Stokhos Development
Loading...
Searching...
No Matches
Stokhos::MultiplyImpl< ValueType, MatrixValue, VectorValue, Device > Class Template Reference
Collaboration diagram for Stokhos::MultiplyImpl< ValueType, MatrixValue, VectorValue, Device >:

Public Types

typedef Device execution_space
typedef CrsProductTensor< ValueType, execution_space > tensor_type
typedef StochasticProductTensor< ValueType, tensor_type, execution_space > BlockSpec
typedef BlockSpec::size_type size_type
typedef Kokkos::View< VectorValue **, Kokkos::LayoutLeft, execution_space > block_vector_type
typedef BlockCrsMatrix< BlockSpec, MatrixValue, execution_space > matrix_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_type > compute_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

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