![]() |
CUTLASS
CUDA Templates for Linear Algebra Subroutines and Solvers
|
Partial specialization for row-major shared memory.
#include <fragment_iterator_simt.h>
Public Types | |
| using | WarpShape = WarpShape_ |
| using | Operator = Operator_ |
| using | Layout = layout::RowMajor |
| using | Policy = SimtPolicy< WarpShape, Operator, Layout, MmaSimtPolicy_ > |
| Policy for warp-level epilogue components. More... | |
| using | Fragment = Array< typename Operator::ElementC, Policy::kElementsPerIteration > |
| This is the fragment size produced by one access of the iterator. More... | |
| using | AccumulatorTile = Array< typename Operator::ElementC, Policy::kAccumulatorElementCount > |
| This is the complete warp-level accumulator tile. More... | |
| using | OutputAccumulatorTile = AccumulatorTile |
Public Member Functions | |
| CUTLASS_HOST_DEVICE | FragmentIteratorSimt (AccumulatorTile const &accum) |
| Constructs an iterator. More... | |
| CUTLASS_HOST_DEVICE FragmentIteratorSimt & | operator++ () |
| Increments. More... | |
| CUTLASS_HOST_DEVICE FragmentIteratorSimt & | operator-- () |
| Decrements. More... | |
| CUTLASS_HOST_DEVICE void | load (Fragment &frag, int index_offset=0) const |
| Loads a fragment from the referenced part of the accumulator tile. More... | |
Static Public Attributes | |
| static int const | kIterations = Policy::kIterations |
| Number of times this iterator can be incremented. More... | |
| using cutlass::epilogue::warp::FragmentIteratorSimt< WarpShape_, Operator_, layout::RowMajor, MmaSimtPolicy_ >::AccumulatorTile = Array< typename Operator::ElementC, Policy::kAccumulatorElementCount> |
| using cutlass::epilogue::warp::FragmentIteratorSimt< WarpShape_, Operator_, layout::RowMajor, MmaSimtPolicy_ >::Fragment = Array< typename Operator::ElementC, Policy::kElementsPerIteration> |
| using cutlass::epilogue::warp::FragmentIteratorSimt< WarpShape_, Operator_, layout::RowMajor, MmaSimtPolicy_ >::Layout = layout::RowMajor |
| using cutlass::epilogue::warp::FragmentIteratorSimt< WarpShape_, Operator_, layout::RowMajor, MmaSimtPolicy_ >::Operator = Operator_ |
| using cutlass::epilogue::warp::FragmentIteratorSimt< WarpShape_, Operator_, layout::RowMajor, MmaSimtPolicy_ >::OutputAccumulatorTile = AccumulatorTile |
| using cutlass::epilogue::warp::FragmentIteratorSimt< WarpShape_, Operator_, layout::RowMajor, MmaSimtPolicy_ >::Policy = SimtPolicy<WarpShape, Operator, Layout, MmaSimtPolicy_> |
| using cutlass::epilogue::warp::FragmentIteratorSimt< WarpShape_, Operator_, layout::RowMajor, MmaSimtPolicy_ >::WarpShape = WarpShape_ |
|
inline |
|
inline |
|
inline |
|
inline |
|
static |
1.8.11