Constructs an iterator from a pair of iterators.
|
CUTLASS_DEVICE | ZipTileIterator () |
| Default constructor. More...
|
|
CUTLASS_DEVICE | ZipTileIterator (Params const &_params, Coord< 3 > const &threadblock_offset=make_Coord(0, 0, 0)) |
| Constructs a zip iterator from params. More...
|
|
CUTLASS_DEVICE | ZipTileIterator (First const &_first, Second const &_second) |
| Constructs a zip iterator from iterator instances. More...
|
|
CUTLASS_DEVICE | ZipTileIterator (TensorRef const &ref) |
| Constructs a zip iterator from iterator instances. More...
|
|
CUTLASS_DEVICE | ZipTileIterator (Params const &_params, TensorRef const &ref) |
| Constructs a zip iterator from iterator instances. 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...
|
|
template<typename Fragment > |
CUTLASS_DEVICE void | load_post_increment (Fragment &fragment) |
| Loads a fragment and increments without predicates. More...
|
|
template<typename Fragment > |
CUTLASS_DEVICE void | load_post_increment (Fragment &fragment, Coord< 4 > const &offset) |
| Loads a fragment and increments without predicates. More...
|
|
template<typename Fragment > |
CUTLASS_DEVICE void | load (Fragment &fragment) const |
| Loads a fragment without predicates. More...
|
|
template<typename Fragment > |
CUTLASS_DEVICE void | load (Fragment &fragment, Coord< 4 > const &offset) const |
| Loads a fragment without predicates. More...
|
|
template<typename Fragment > |
CUTLASS_DEVICE void | store_post_increment (Fragment const &fragment) |
| Stores a fragment and increments without predicates. More...
|
|
template<typename Fragment > |
CUTLASS_DEVICE void | store_post_increment (Fragment const &fragment, Coord< 4 > const &offset) |
| Stores a fragment and increments without predicates. More...
|
|
template<typename Fragment > |
CUTLASS_DEVICE void | store (Fragment const &fragment) const |
| Stores a fragment without predicates. More...
|
|
template<typename Fragment > |
CUTLASS_DEVICE void | store (Fragment const &fragment, Coord< 4 > const &offset) const |
| Stores a fragment without predicates. More...
|
|
template<typename Fragment , typename PredicateIterator > |
CUTLASS_DEVICE void | load_post_increment (Fragment &fragment, PredicateIterator pred_it) |
| Loads a fragment and increments, using predicates. More...
|
|
template<typename Fragment , typename PredicateIterator > |
CUTLASS_DEVICE void | load (Fragment &fragment, PredicateIterator pred_it) const |
| Loads a fragment with predicates. More...
|
|
template<typename Fragment , typename PredicateIterator > |
CUTLASS_DEVICE void | store_post_increment (Fragment const &fragment, PredicateIterator pred_it) |
| Loads a fragment and increments, using predicates. More...
|
|
template<typename Fragment , typename PredicateIterator > |
CUTLASS_DEVICE void | store (Fragment const &fragment, PredicateIterator pred_it) const |
| Loads a fragment with predicates. More...
|
|
CUTLASS_DEVICE ZipTileIterator & | increment (int count=1) |
| Increments store iterator to next tile. More...
|
|
CUTLASS_DEVICE ZipTileIterator & | operator++ () |
| Increments to next tile. More...
|
|
CUTLASS_DEVICE ZipTileIterator & | operator+= (int count) |
|
CUTLASS_DEVICE ZipTileIterator & | operator+= (Coord< 3 > const &offset) |
| Adds a vector offset to the underlying iterators. More...
|
|
CUTLASS_DEVICE ZipTileIterator & | decrement (int count=1) |
| Increments store iterator to previous tile. More...
|
|
CUTLASS_DEVICE ZipTileIterator & | operator-- () |
| Increments to subsequent tile. More...
|
|
CUTLASS_DEVICE ZipTileIterator & | operator-= (int count) |
| Decrements to previous tile. More...
|
|
CUTLASS_DEVICE void | add_pointer_offset (LongIndex offset) |
| Adds an offset to both iterators. More...
|
|