|
typedef GemmGlobalIteratorCd< TileTraits_, Index_ > | This_ |
| This class. More...
|
|
typedef TileIteratorBase< TileTraits_, typename TileTraits_::Scalar, IteratorAdvance::kH, MemorySpace::kGlobal, Index_ > | Base |
| The base class. More...
|
|
typedef TileTraits_::Scalar | Scalar |
| The scalar. More...
|
|
typedef TileTraits_::Pointer | Pointer |
| The pointer. More...
|
|
typedef TileTraits_::Threads | Threads |
| The threads. More...
|
|
typedef Index_ | Index |
| The index. More...
|
|
typedef long long | LongIndex |
| The index. More...
|
|
typedef TileTraits_::ThreadOffset | ThreadOffset |
| The thread offset. More...
|
|
typedef TileTraits_ | Traits |
| concept TileTraits More...
|
|
typedef TileTraits_::Scalar | Scalar |
| Scalar element. More...
|
|
typedef TileTraits_::Scalar | FragmentElement |
| Fragment element. More...
|
|
typedef Index_ | Index |
| Index type. More...
|
|
typedef long long | LongIndex |
| Long index. More...
|
|
typedef Shape< 0, 0, 0, 0 > | 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...
|
|
|
CUTLASS_HOST_DEVICE | GemmGlobalIteratorCd (Params const &_params, const Coord< 3 > &bounds, const Coord< 3 > &block, int offset=0, int pred_offset=0, ThreadOffset thread_offset_func=ThreadOffset()) |
| Ctor. More...
|
|
CUTLASS_HOST_DEVICE void | inc_c () |
| Increment the pointer in the C dimension. More...
|
|
CUTLASS_HOST_DEVICE void | inc_w () |
| Increment the pointer in the W dimension. More...
|
|
CUTLASS_HOST_DEVICE void | inc_h () |
| Increment the pointer in the H dimension. More...
|
|
CUTLASS_HOST_DEVICE void | inc_d () |
| Increment the pointer in the D dimension. More...
|
|
CUTLASS_HOST_DEVICE void | inc_advance () |
| Increment the pointer to move to the next iteration. More...
|
|
CUTLASS_HOST_DEVICE GemmGlobalIteratorCd & | operator+= (Coord< 3 > const &offset) |
| Adds a vector offset to the iterator. More...
|
|
CUTLASS_HOST_DEVICE void | load_element (typename Base::AccessType &value, int d, int h, int w, int c) const |
| Loads a single fragment element from memory. More...
|
|
CUTLASS_HOST_DEVICE void | store_element (typename Base::AccessType const &value, int d, int h, int w, int c) |
| Stores a single fragment element into memory. More...
|
|
CUTLASS_HOST_DEVICE bool | valid (int d, int h, int w, int c) const |
| Test the validity of the. More...
|
|
CUTLASS_HOST_DEVICE void | add_pointer_offset (LongIndex offset) |
| add pointer offset More...
|
|
template<typename Fragment > |
CUTLASS_HOST_DEVICE void | load_post_increment (Fragment &fragment) |
| Loads and increments iterator. More...
|
|
template<typename Fragment > |
CUTLASS_HOST_DEVICE void | store_post_increment (Fragment &fragment) |
|
CUTLASS_HOST_DEVICE bool | valid (int d, int h, int w, int c) const |
| Is the iterator valid? More...
|
|