Stokhos Development
Loading...
Searching...
No Matches
Stokhos::SimpleTiledCrsProductTensor< ValueType, ExecutionSpace > Class Template Reference

Public Types

typedef ExecutionSpace execution_space
typedef int size_type
typedef ValueType value_type

Public Member Functions

 SimpleTiledCrsProductTensor (const SimpleTiledCrsProductTensor &rhs)
SimpleTiledCrsProductTensor & operator= (const SimpleTiledCrsProductTensor &rhs)
KOKKOS_INLINE_FUNCTION size_type dimension () const
 Dimension of the tensor.
KOKKOS_INLINE_FUNCTION size_type entry_count () const
 Number of sparse entries.
KOKKOS_INLINE_FUNCTION size_type num_i_tiles () const
 Number i-tiles.
KOKKOS_INLINE_FUNCTION size_type i_begin (const size_type i) const
 Begin entries with for tile 'i'.
KOKKOS_INLINE_FUNCTION size_type i_size (const size_type i) const
 Number of entries with for tile 'i'.
KOKKOS_INLINE_FUNCTION size_type num_j_tiles (const size_type i) const
 Number j-tiles.
KOKKOS_INLINE_FUNCTION size_type j_begin (const size_type i, const size_type j) const
 Begin entries with for tile 'i,j'.
KOKKOS_INLINE_FUNCTION size_type j_size (const size_type i, const size_type j) const
 Number of entries with for tile 'i,j'.
KOKKOS_INLINE_FUNCTION size_type num_k_tiles (const size_type i, const size_type j) const
 Number k-tiles.
KOKKOS_INLINE_FUNCTION size_type k_begin (const size_type i, const size_type j, const size_type k) const
 Begin entries with for tile 'i,j,k'.
KOKKOS_INLINE_FUNCTION size_type k_size (const size_type i, const size_type j, const size_type k) const
 Number of entries with for tile 'i,j'.
KOKKOS_INLINE_FUNCTION size_type num_entry (const size_type i, const size_type j, const size_type k, const size_type r) const
 Number of entries for tile (i,j,k) and row r.
KOKKOS_INLINE_FUNCTION size_type entry_begin (const size_type i, const size_type j, const size_type k, const size_type r) const
 Begin entries for tile (i,j,k) and row r.
KOKKOS_INLINE_FUNCTION size_type entry_end (const size_type i, const size_type j, const size_type k, const size_type r) const
 End entries for tile (i,j,k) and row r.
KOKKOS_INLINE_FUNCTION const size_type & coord (const size_type entry, const size_type c) const
 Coordinates of an entry.
KOKKOS_INLINE_FUNCTION const size_type & coord (const size_type entry) const
 Coordinates of an entry.
KOKKOS_INLINE_FUNCTION const value_type & value (const size_type entry) const
 Value of an entry.
KOKKOS_INLINE_FUNCTION size_type num_non_zeros () const
 Number of non-zero's.
KOKKOS_INLINE_FUNCTION size_type num_flops () const
 Number flop's per multiply-add.
KOKKOS_INLINE_FUNCTION size_type max_i_tile_size () const
 Max size of any i tile.
KOKKOS_INLINE_FUNCTION size_type max_jk_tile_size () const
 Max size of any j/k tile.

Static Public Member Functions

template<typename OrdinalType>
static SimpleTiledCrsProductTensor create (const Stokhos::ProductBasis< OrdinalType, ValueType > &basis, const Stokhos::Sparse3Tensor< OrdinalType, ValueType > &Cijk, const Teuchos::ParameterList &params)

Static Public Attributes

static const size_type host_vectorsize = 2
static const bool use_intrinsics = false
static const size_type cuda_vectorsize = 32
static const bool is_cuda
static const size_type vectorsize = is_cuda ? cuda_vectorsize : host_vectorsize
static const size_type tensor_align = vectorsize

Member Data Documentation

◆ is_cuda

template<typename ValueType, class ExecutionSpace>
const bool Stokhos::SimpleTiledCrsProductTensor< ValueType, ExecutionSpace >::is_cuda
static
Initial value:
=
false

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