|
Cutlass
CUDA Templates for Linear Algebra Subroutines and Solvers
|
#include <gemm_global_tile.h>
Classes | |
| struct | Params |
| The params. More... | |
Public Types | |
| 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 TileTraits_::ThreadOffset | ThreadOffset |
| The thread offset. More... | |
Public Types inherited from cutlass::TileIteratorBase< TileTraits_, TileTraits_::Scalar, IteratorAdvance::kH, MemorySpace::kGlobal, Index_ > | |
| 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 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... | |
Public Member Functions | |
| CUTLASS_DEVICE | GemmGlobalIteratorCd () |
| Ctor. More... | |
| CUTLASS_DEVICE | GemmGlobalIteratorCd (Params const ¶ms, const Coord< 3 > &bounds, const Coord< 3 > &block, int offset=0, int pred_offset=0, ThreadOffset thread_offset_func=ThreadOffset()) |
| Ctor. More... | |
| CUTLASS_DEVICE void | inc_c () |
| Increment the pointer in the C dimension. More... | |
| CUTLASS_DEVICE void | inc_w () |
| Increment the pointer in the W dimension. More... | |
| CUTLASS_DEVICE void | inc_h () |
| Increment the pointer in the H dimension. More... | |
| CUTLASS_DEVICE void | inc_d () |
| Increment the pointer in the D dimension. More... | |
| CUTLASS_DEVICE void | inc_advance () |
| Increment the pointer to move to the next iteration. More... | |
| CUTLASS_DEVICE bool | valid (int d, int h, int w, int c) const |
| Test the validity of the iterator. More... | |
| CUTLASS_HOST_DEVICE Pointer | data () |
| Returns the raw pointer. More... | |
| CUTLASS_HOST_DEVICE Pointer const | data () const |
Public Member Functions inherited from cutlass::TileIteratorBase< TileTraits_, TileTraits_::Scalar, IteratorAdvance::kH, MemorySpace::kGlobal, Index_ > | |
| CUTLASS_DEVICE bool | valid (int d, int h, int w, int c) const |
| Is the iterator valid? More... | |
Public Attributes | |
| Params | params |
| Coord< 4 > | thread_offset |
| Offset of an individual lane from the start of the tile. More... | |
| cutlass::PredicateVector< Base::Iterations::kW > | predicates |
| The predicates for the row. More... | |
Static Public Attributes | |
| static MatrixLayout::Kind const | kLayout = TileTraits_::kLayout |
| The layout. More... | |
Static Public Attributes inherited from cutlass::TileIteratorBase< TileTraits_, TileTraits_::Scalar, IteratorAdvance::kH, MemorySpace::kGlobal, Index_ > | |
| static IteratorAdvance::Kind const | kAdvance |
| Specifies dimension in which post-increment accesses advance. More... | |
| static IteratorFragment::Kind const | kIteratorFragment |
| Specifies iterator storage fragment type (Scalar or WmmaMatrix) More... | |
| static MemorySpace::Kind const | kMemorySpace |
| Source or destination memory space. More... | |
| static int const | kAccessSize |
| The number of scalars accessed per load/store. More... | |
| static int const | kFragmentSize |
| The size of storage needed per fragment. More... | |
Additional Inherited Members | |
Static Public Member Functions inherited from cutlass::TileIteratorBase< TileTraits_, TileTraits_::Scalar, IteratorAdvance::kH, MemorySpace::kGlobal, Index_ > | |
| static CUTLASS_DEVICE void | initialize_predicates (PredicateIterator predicate_it, Coord< 3 > const &bounds, Coord< 3 > const &offset=make_Coord(0, 0, 0)) |
| Initializes a predicate vector. More... | |
| typedef TileIteratorBase<TileTraits_, typename TileTraits_::Scalar, IteratorAdvance::kH, MemorySpace::kGlobal, Index_> cutlass::gemm::GemmGlobalIteratorCd< TileTraits_, Index_ >::Base |
| typedef Index_ cutlass::gemm::GemmGlobalIteratorCd< TileTraits_, Index_ >::Index |
| typedef TileTraits_::Pointer cutlass::gemm::GemmGlobalIteratorCd< TileTraits_, Index_ >::Pointer |
| typedef TileTraits_::Scalar cutlass::gemm::GemmGlobalIteratorCd< TileTraits_, Index_ >::Scalar |
| typedef GemmGlobalIteratorCd<TileTraits_, Index_> cutlass::gemm::GemmGlobalIteratorCd< TileTraits_, Index_ >::This_ |
| typedef TileTraits_::ThreadOffset cutlass::gemm::GemmGlobalIteratorCd< TileTraits_, Index_ >::ThreadOffset |
| typedef TileTraits_::Threads cutlass::gemm::GemmGlobalIteratorCd< TileTraits_, Index_ >::Threads |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
static |
| Params cutlass::gemm::GemmGlobalIteratorCd< TileTraits_, Index_ >::params |
| cutlass::PredicateVector<Base::Iterations::kW> cutlass::gemm::GemmGlobalIteratorCd< TileTraits_, Index_ >::predicates |
| Coord<4> cutlass::gemm::GemmGlobalIteratorCd< TileTraits_, Index_ >::thread_offset |
1.8.14