Stokhos
Development
|
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 |