|
typedef TileIteratorBase< Traits_, Scalar_, Advance_, MemorySpace, Index_, FragmentElement_, FragmentElementType_, Skew_ > | Base |
| Base class. More...
|
|
typedef Base::Traits | Traits |
| concept TileTraits More...
|
|
typedef Base::Scalar | Scalar |
| Scalar element. More...
|
|
typedef Base::FragmentElement | FragmentElement |
| Fragment element. More...
|
|
typedef Base::Index | Index |
| Index type. More...
|
|
typedef Base::LongIndex | LongIndex |
| Long index type. More...
|
|
typedef Base::Skew | Skew |
| Skew quantity. More...
|
|
typedef Base::Tile | Tile |
| Tile shape. More...
|
|
typedef Base::Delta | Delta |
| Delta. More...
|
|
typedef Base::Iterations | Iterations |
| Iterations. More...
|
|
typedef Base::ThreadOffset | ThreadOffset |
| ThreadOffset functor. More...
|
|
typedef Base::FragmentShape | FragmentShape |
| Fragment type. More...
|
|
typedef Base::AccessType | AccessType |
| Memory access type. More...
|
|
typedef Base::Fragment | Fragment |
| Fragment definition. More...
|
|
typedef Base::FragmentIterator | FragmentIterator |
| Fragment iterator definition. More...
|
|
typedef Base::FragmentConstIterator | FragmentConstIterator |
| Fragment const iterator definition. More...
|
|
typedef Base::PredicateVector | PredicateVector |
| Default predicate mask type. More...
|
|
typedef Base::Storage | SharedStorage |
| Storage object which may be stored to. More...
|
|
typedef Base::Params | BaseParams |
| IteratorBase parameters. More...
|
|
typedef Scalar * | Pointer |
| Pointer to underlying type. More...
|
|
typedef TensorRef< Scalar, 4 > | TensorRef |
| Tensor reference for the store iterator. More...
|
|
typedef Traits_ | Traits |
| concept TileTraits More...
|
|
typedef Scalar_ | Scalar |
| Scalar element. More...
|
|
typedef FragmentElement_ | FragmentElement |
| Fragment element. More...
|
|
typedef Index_ | Index |
| Index type. More...
|
|
typedef long long | LongIndex |
| Long index. More...
|
|
typedef Skew_ | Skew |
| Skew quantity. More...
|
|
typedef Traits::Tile | Tile |
| Tile shape. More...
|
|
typedef Traits::Delta | Delta |
| Distance along each dimension. More...
|
|
typedef Traits::ImmediateOffsetStrides | ImmediateOffsetStrides |
| The strides in each dimension between different loads/stores. More...
|
|
typedef Traits::Iterations | Iterations |
| Iterations. More...
|
|
typedef Traits::ThreadOffset | ThreadOffset |
| Thread offset. More...
|
|
typedef Vectorize< FragmentElement, kAccessSize >::Type | AccessType |
| The elements loaded/store by one instruction. More...
|
|
typedef Fragment< Scalar, ShapeCount< Tile >::kCount, kFragmentSize > | Storage |
| The storage. More...
|
|
typedef Fragment< FragmentElement, ShapeCount< Iterations >::kCount *kAccessSize > | Fragment |
| The fragment. More...
|
|
typedef FragmentIterator< Fragment, Iterations, AccessType > | FragmentIterator |
| The fragment iterator. More...
|
|
typedef FragmentConstIterator< Fragment, Iterations, AccessType > | FragmentConstIterator |
| The fragment const iterator. More...
|
|
typedef FragmentIterator::FragmentShape | FragmentShape |
| The shape of the fragment. More...
|
|
typedef PredicateVector< ShapeCount< Iterations >::kCount > | PredicateVector |
| Default predicate mask type. More...
|
|
|
template<typename PredicateIterator > |
CUTLASS_HOST_DEVICE void | initialize_predicates (PredicateIterator predicate_it, Coord< 3 > const &bounds, Coord< 3 > const &block_offset=make_Coord(0, 0, 0)) |
| Initializes a predicate vector using a RegularTilePredicateFunctor. More...
|
|
template<typename PredicateIterator , typename PredicateFunctor > |
CUTLASS_HOST_DEVICE void | initialize_predicates (PredicateIterator predicate_it, PredicateFunctor const &functor, Coord< 3 > const &block_offset) |
| Initializes a predicate vector using an arbitrary predicate functor. More...
|
|
CUTLASS_HOST_DEVICE | TileStoreIterator () |
| Default constructor. More...
|
|
CUTLASS_HOST_DEVICE | TileStoreIterator (Params const &_params, Coord< 3 > const &block_offset=make_Coord(0, 0, 0), ThreadOffset thread_offset_func=ThreadOffset()) |
| Constructs a tile store iterator. More...
|
|
CUTLASS_HOST_DEVICE | TileStoreIterator (Params const &, Scalar *ptr, ThreadOffset thread_offset_func=ThreadOffset()) |
| Constructs a tile store iterator. More...
|
|
CUTLASS_HOST_DEVICE void | inc_d () |
| Increment in the D dimension. More...
|
|
CUTLASS_HOST_DEVICE void | inc_h () |
| Increment in the H dimension. More...
|
|
CUTLASS_HOST_DEVICE void | inc_w () |
| Increment in the W dimension. More...
|
|
CUTLASS_HOST_DEVICE void | inc_advance () |
| Increment in the next dimension. More...
|
|
CUTLASS_HOST_DEVICE void | inc_stage () |
| Increment the stage. More...
|
|
CUTLASS_HOST_DEVICE TileStoreIterator & | operator+= (Coord< 3 > const &offset) |
| Adds a vector offset to the iterator. More...
|
|
CUTLASS_HOST_DEVICE void | add_pointer_offset (LongIndex offset) |
| Adds a raw offset to the pointer. More...
|
|
CUTLASS_HOST_DEVICE void | store_element (AccessType const &value, int d, int h, int w, int c) |
| Stores a single fragment element into memory. More...
|
|
template<typename Fragment , typename PredicateIterator > |
CUTLASS_HOST_DEVICE void | store_post_increment (Fragment const &fragment, PredicateIterator pred_it) |
| Stores a fragment and advances to the next tile. More...
|
|
template<typename Fragment > |
CUTLASS_HOST_DEVICE void | store_post_increment (Fragment const &fragment) |
| Stores a fragment and advances to the next tile. More...
|
|
template<typename Fragment , typename PredicateIterator > |
CUTLASS_HOST_DEVICE void | store (Fragment const &fragment, PredicateIterator pred_it) const |
| Stores a fragment without advancing the iterator. More...
|
|
template<typename Fragment > |
CUTLASS_HOST_DEVICE void | store (Fragment const &fragment) const |
| Stores a fragment without advancing the iterator. More...
|
|
CUTLASS_HOST_DEVICE void | load_element (AccessType &value, int d, int h, int w, int c) const |
| Loads a single fragment element from memory. More...
|
|
template<typename Fragment , typename PredicateIterator > |
CUTLASS_HOST_DEVICE void | load_post_increment (Fragment &fragment, PredicateIterator pred_it) |
| Loads a fragment and advances the iterator to the next tile. More...
|
|
template<typename Fragment > |
CUTLASS_HOST_DEVICE void | load_post_increment (Fragment &fragment) |
| Loads a fragment and advances the iterator to the next tile. More...
|
|
template<typename Fragment , typename PredicateIterator > |
CUTLASS_HOST_DEVICE void | load (Fragment &fragment, PredicateIterator pred_it) const |
| Loads a fragment without advancing the iterator.. More...
|
|
template<typename Fragment > |
CUTLASS_HOST_DEVICE void | load (Fragment &fragment) const |
| Loads a fragment without advancing the iterator.. More...
|
|
template<typename Fragment > |
CUTLASS_HOST_DEVICE void | load (Fragment &fragment, int d) |
| Loads a fragment without advancing the iterator.. More...
|
|
CUTLASS_HOST_DEVICE bool | valid (int d, int h, int w, int c) const |
| Is the iterator valid? More...
|
|
template<typename Traits_ , typename Scalar_ , IteratorAdvance::Kind Advance_ = IteratorAdvance::kH, MemorySpace::Kind MemorySpace = MemorySpace::kGeneric, typename Index_ = int, typename FragmentElement_ = Scalar_, FragmentElementType::Kind FragmentElementType_ = FragmentElementType::kScalar, typename Skew_ = Shape<0, 0, 0, 0>>
template<typename PredicateIterator >
CUTLASS_HOST_DEVICE void cutlass::TileStoreIterator< Traits_, Scalar_, Advance_, MemorySpace, Index_, FragmentElement_, FragmentElementType_, Skew_ >::initialize_predicates |
( |
PredicateIterator |
predicate_it, |
|
|
Coord< 3 > const & |
bounds, |
|
|
Coord< 3 > const & |
block_offset = make_Coord(0, 0, 0) |
|
) |
| |
|
inline |
template<typename Traits_ , typename Scalar_ , IteratorAdvance::Kind Advance_ = IteratorAdvance::kH, MemorySpace::Kind MemorySpace = MemorySpace::kGeneric, typename Index_ = int, typename FragmentElement_ = Scalar_, FragmentElementType::Kind FragmentElementType_ = FragmentElementType::kScalar, typename Skew_ = Shape<0, 0, 0, 0>>
template<typename PredicateIterator , typename PredicateFunctor >
template<typename Traits_ , typename Scalar_ , IteratorAdvance::Kind Advance_ = IteratorAdvance::kH, MemorySpace::Kind MemorySpace = MemorySpace::kGeneric, typename Index_ = int, typename FragmentElement_ = Scalar_, FragmentElementType::Kind FragmentElementType_ = FragmentElementType::kScalar, typename Skew_ = Shape<0, 0, 0, 0>>
template<typename Fragment , typename PredicateIterator >
template<typename Traits_ , typename Scalar_ , IteratorAdvance::Kind Advance_ = IteratorAdvance::kH, MemorySpace::Kind MemorySpace = MemorySpace::kGeneric, typename Index_ = int, typename FragmentElement_ = Scalar_, FragmentElementType::Kind FragmentElementType_ = FragmentElementType::kScalar, typename Skew_ = Shape<0, 0, 0, 0>>
template<typename Fragment >
template<typename Traits_ , typename Scalar_ , IteratorAdvance::Kind Advance_ = IteratorAdvance::kH, MemorySpace::Kind MemorySpace = MemorySpace::kGeneric, typename Index_ = int, typename FragmentElement_ = Scalar_, FragmentElementType::Kind FragmentElementType_ = FragmentElementType::kScalar, typename Skew_ = Shape<0, 0, 0, 0>>
template<typename Fragment >
template<typename Traits_ , typename Scalar_ , IteratorAdvance::Kind Advance_ = IteratorAdvance::kH, MemorySpace::Kind MemorySpace = MemorySpace::kGeneric, typename Index_ = int, typename FragmentElement_ = Scalar_, FragmentElementType::Kind FragmentElementType_ = FragmentElementType::kScalar, typename Skew_ = Shape<0, 0, 0, 0>>
template<typename Fragment , typename PredicateIterator >
template<typename Traits_ , typename Scalar_ , IteratorAdvance::Kind Advance_ = IteratorAdvance::kH, MemorySpace::Kind MemorySpace = MemorySpace::kGeneric, typename Index_ = int, typename FragmentElement_ = Scalar_, FragmentElementType::Kind FragmentElementType_ = FragmentElementType::kScalar, typename Skew_ = Shape<0, 0, 0, 0>>
template<typename Fragment >
template<typename Traits_ , typename Scalar_ , IteratorAdvance::Kind Advance_ = IteratorAdvance::kH, MemorySpace::Kind MemorySpace = MemorySpace::kGeneric, typename Index_ = int, typename FragmentElement_ = Scalar_, FragmentElementType::Kind FragmentElementType_ = FragmentElementType::kScalar, typename Skew_ = Shape<0, 0, 0, 0>>
template<typename Fragment , typename PredicateIterator >
template<typename Traits_ , typename Scalar_ , IteratorAdvance::Kind Advance_ = IteratorAdvance::kH, MemorySpace::Kind MemorySpace = MemorySpace::kGeneric, typename Index_ = int, typename FragmentElement_ = Scalar_, FragmentElementType::Kind FragmentElementType_ = FragmentElementType::kScalar, typename Skew_ = Shape<0, 0, 0, 0>>
template<typename Fragment >
template<typename Traits_ , typename Scalar_ , IteratorAdvance::Kind Advance_ = IteratorAdvance::kH, MemorySpace::Kind MemorySpace = MemorySpace::kGeneric, typename Index_ = int, typename FragmentElement_ = Scalar_, FragmentElementType::Kind FragmentElementType_ = FragmentElementType::kScalar, typename Skew_ = Shape<0, 0, 0, 0>>
template<typename Fragment , typename PredicateIterator >
template<typename Traits_ , typename Scalar_ , IteratorAdvance::Kind Advance_ = IteratorAdvance::kH, MemorySpace::Kind MemorySpace = MemorySpace::kGeneric, typename Index_ = int, typename FragmentElement_ = Scalar_, FragmentElementType::Kind FragmentElementType_ = FragmentElementType::kScalar, typename Skew_ = Shape<0, 0, 0, 0>>
template<typename Fragment >
template<typename Traits_ , typename Scalar_ , IteratorAdvance::Kind Advance_ = IteratorAdvance::kH, MemorySpace::Kind MemorySpace = MemorySpace::kGeneric, typename Index_ = int, typename FragmentElement_ = Scalar_, FragmentElementType::Kind FragmentElementType_ = FragmentElementType::kScalar, typename Skew_ = Shape<0, 0, 0, 0>>
template<typename Traits_ , typename Scalar_ , IteratorAdvance::Kind Advance_ = IteratorAdvance::kH, MemorySpace::Kind MemorySpace = MemorySpace::kGeneric, typename Index_ = int, typename FragmentElement_ = Scalar_, FragmentElementType::Kind FragmentElementType_ = FragmentElementType::kScalar, typename Skew_ = Shape<0, 0, 0, 0>>