|
typedef Traits_ | Traits |
| Traits. More...
|
|
typedef Traits::FragmentStream | Base |
| Base class. More...
|
|
typedef Traits::Scalar | Scalar |
| Scalar type. More...
|
|
typedef Base::LoadIterator | LoadIterator |
| Defines the load iterator. More...
|
|
typedef Base::StoreIterator | StoreIterator |
| Defines the store iterator. More...
|
|
typedef Base::Convert | Convert |
| Converts between tiles. More...
|
|
typedef Base::Fragment | Fragment |
| Loaded fragment type. More...
|
|
typedef Base::StoreFragment | StoreFragment |
| Stored fragment type. More...
|
|
typedef Base::Storage | Storage |
| Destination storage. More...
|
|
typedef Traits::Index | Index |
| Index type. More...
|
|
typedef Traits::Scalar const * | Pointer |
| The pointer. More...
|
|
typedef Traits_ | Traits |
| Defines traits of WMMA GEMM tile stream. More...
|
|
typedef TileLoadIterator< Traits_::TileTraits, Traits_::Scalar, Traits_::MultiplicandTraits::kKstrided ? IteratorAdvance::kH :IteratorAdvance::kW, MemorySpace::kGlobal, Traits_::Index > | LoadIterator |
| Defines the load iterator. More...
|
|
typedef TileStoreIterator< Traits_::TileTraits, Traits_::Scalar, Traits_::MultiplicandTraits::kKstrided ? IteratorAdvance::kH :IteratorAdvance::kW, MemorySpace::kShared, Traits_::Index, Traits_::Scalar, IteratorFragment::kScalar, Traits_::DestinationSkew > | StoreIterator |
| Defines the store iterator. More...
|
|
typedef FragmentCopy< typename TileStoreIterator< Traits_::TileTraits, Traits_::Scalar, Traits_::MultiplicandTraits::kKstrided ? IteratorAdvance::kH :IteratorAdvance::kW, MemorySpace::kShared, Traits_::Index, Traits_::Scalar, IteratorFragment::kScalar, Traits_::DestinationSkew > ::Fragment, typename TileLoadIterator< Traits_::TileTraits, Traits_::Scalar, Traits_::MultiplicandTraits::kKstrided ? IteratorAdvance::kH :IteratorAdvance::kW, MemorySpace::kGlobal, Traits_::Index > ::Fragment > | Convert |
| Converts between tiles. More...
|
|
typedef int | Index |
| Index type. More...
|
|
typedef LoadIterator::Fragment | Fragment |
| Loaded fragment type. More...
|
|
typedef StoreIterator::Fragment | StoreFragment |
| Stored fragment type. More...
|
|
typedef StoreIterator::Storage | Storage |
| Destination storage. More...
|
|
typedef StoreIterator::Storage | SharedStoreStorage |
| The storage in shared memory. More...
|
|
|
CUTLASS_DEVICE | GemmFragmentStream () |
|
CUTLASS_DEVICE | GemmFragmentStream (Params const ¶ms, Coord< 3 > const &bounds, Coord< 3 > const &block_offset=make_Coord(0, 0, 0)) |
| Constructor - bounds and block offset are aligned to GEMM coordinates (K, N, M) More...
|
|
CUTLASS_DEVICE void | load () |
| Loads the fragment. More...
|
|
CUTLASS_DEVICE void | commit () |
| Commits the fragment. More...
|
|
CUTLASS_DEVICE void | residue (Coord< 3 > const &bounds, Coord< 3 > const &block_offset) |
| TODO - Recomputes predicates and clears fetch registers. More...
|
|
CUTLASS_DEVICE void | initialize_predicates (Coord< 3 > const &bounds, Coord< 3 > const &block_offset) |
| Recomputes predicates aligned to GEMM coordinates (K, N, M) More...
|
|
CUTLASS_DEVICE | FragmentStream () |
|
CUTLASS_DEVICE | FragmentStream (Params const ¶ms, Coord< 3 > const &bounds, Coord< 3 > const &block_offset=make_Coord(0, 0, 0)) |
| Constructor. More...
|
|
CUTLASS_DEVICE void | load () |
| Loads the fragment. More...
|
|
CUTLASS_DEVICE void | commit () |
| Commits the fragment. More...
|
|
CUTLASS_DEVICE void | initialize_predicates (Coord< 3 > const &bounds, Coord< 3 > const &block_offset) |
| Recomputes predicates. More...
|
|