Stokhos Development
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
Collaboration diagram for Stokhos::MultiplyImpl< ValueType, MatrixValue, VectorValue, Device >:
Collaboration graph
[legend]

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: