CUTLASS: cutlass::epilogue::threadblock::InterleavedPredicatedTileIterator< ThreadMap_, Element_, InterleavedK > Class Template Reference

| | CUTLASS

CUDA Templates for Linear Algebra Subroutines and Solvers |

• Main Page

• Modules

• Namespaces

• Classes

• Files

• Class List

• Class Index

• Class Hierarchy

• Class Members

• cutlass

• epilogue

• threadblock

• InterleavedPredicatedTileIterator

Classes | Public Types | Public Member Functions | Static Public Attributes | List of all members

cutlass::epilogue::threadblock::InterleavedPredicatedTileIterator< ThreadMap_, Element_, InterleavedK > Class Template Reference

#include <predicated_tile_iterator.h>

|

Classes

| | struct | Mask | | | Mask object. More...
| | | | struct | Params | | |

|

Public Types

| | using | ThreadMap = ThreadMap_ | | | | using | Element = Element_ | | | | using | Layout = layout::ColumnMajorInterleaved< InterleavedK > | | | | using | TensorRef = TensorRef< Element, Layout > | | | | using | ConstTensorRef = typename TensorRef::ConstTensorRef | | | | using | Index = typename Layout::Index | | | | using | LongIndex = typename Layout::LongIndex | | | | using | TensorCoord = layout::PitchLinearCoord | | | | using | Fragment = Array< Element, ThreadMap::kElementsPerAccess > | | | Fragment object. More...
| | | | using | AccessType = AlignedArray< Element, ThreadMap::kElementsPerAccess > | | | Memory access size. More...
| | |

|

Public Member Functions

| | CUTLASS_DEVICE | InterleavedPredicatedTileIterator (Params const &params, Element *pointer, TensorCoord extent, int thread_idx, TensorCoord threadblock_offset) | | | Constructor. More...
| | | | CUTLASS_HOST_DEVICE void | add_pointer_offset (LongIndex pointer_offset) | | | Adds a pointer offset in units of Element. More...
| | | | CUTLASS_DEVICE void | load (Fragment &frag) | | | Loads a fragment from memory. More...
| | | | CUTLASS_DEVICE void | store (Fragment const &frag) | | | Stores a fragment to memory. More...
| | | | CUTLASS_HOST_DEVICE void | set_iteration_index (int iteration) | | | Overrides the internal iteration index. More...
| | | | CUTLASS_HOST_DEVICE InterleavedPredicatedTileIterator & | operator++ () | | | Advances to the next position to load or store. More...
| | | | CUTLASS_DEVICE void | clear_mask () | | | Efficiently enables all accesses guarded by mask. More...
| | | | CUTLASS_DEVICE void | enable_mask () | | | Sets the mask. More...
| | | | CUTLASS_DEVICE void | get_mask (Mask &mask) | | | Sets the mask. More...
| | | | CUTLASS_DEVICE void | set_mask (Mask const &mask) | | |

|

Static Public Attributes

| | static int const | kElementsPerAccess = ThreadMap::kElementsPerAccess | | | | static int const | kThreads = ThreadMap::kThreads | | | | static int const | kIterations = ThreadMap::Iterations::kCount | | |

Detailed Description

template<typename ThreadMap_, typename Element_, int InterleavedK> class cutlass::epilogue::threadblock::InterleavedPredicatedTileIterator< ThreadMap_, Element_, InterleavedK >

Tile iterator used to load output tile from shared memory in epilogue.

Satisfies: ReadableTileIterator | InterleavedPredicatedTileIterator | ForwardTileIterator

Member Typedef Documentation

template<typename ThreadMap_ , typename Element_ , int InterleavedK>

| using cutlass::epilogue::threadblock::InterleavedPredicatedTileIterator< ThreadMap_, Element_, InterleavedK >::AccessType = AlignedArray<Element, ThreadMap::kElementsPerAccess> |

template<typename ThreadMap_ , typename Element_ , int InterleavedK>

| using cutlass::epilogue::threadblock::InterleavedPredicatedTileIterator< ThreadMap_, Element_, InterleavedK >::ConstTensorRef = typename TensorRef::ConstTensorRef |

template<typename ThreadMap_ , typename Element_ , int InterleavedK>

| using cutlass::epilogue::threadblock::InterleavedPredicatedTileIterator< ThreadMap_, Element_, InterleavedK >::Element = Element_ |

template<typename ThreadMap_ , typename Element_ , int InterleavedK>

| using cutlass::epilogue::threadblock::InterleavedPredicatedTileIterator< ThreadMap_, Element_, InterleavedK >::Fragment = Array<Element, ThreadMap::kElementsPerAccess> |

template<typename ThreadMap_ , typename Element_ , int InterleavedK>

| using cutlass::epilogue::threadblock::InterleavedPredicatedTileIterator< ThreadMap_, Element_, InterleavedK >::Index = typename Layout::Index |

template<typename ThreadMap_ , typename Element_ , int InterleavedK>

| using cutlass::epilogue::threadblock::InterleavedPredicatedTileIterator< ThreadMap_, Element_, InterleavedK >::Layout = layout::ColumnMajorInterleaved<InterleavedK> |

template<typename ThreadMap_ , typename Element_ , int InterleavedK>

| using cutlass::epilogue::threadblock::InterleavedPredicatedTileIterator< ThreadMap_, Element_, InterleavedK >::LongIndex = typename Layout::LongIndex |

template<typename ThreadMap_ , typename Element_ , int InterleavedK>

| using cutlass::epilogue::threadblock::InterleavedPredicatedTileIterator< ThreadMap_, Element_, InterleavedK >::TensorCoord = layout::PitchLinearCoord |

template<typename ThreadMap_ , typename Element_ , int InterleavedK>

| using cutlass::epilogue::threadblock::InterleavedPredicatedTileIterator< ThreadMap_, Element_, InterleavedK >::TensorRef = TensorRef<Element, Layout> |

template<typename ThreadMap_ , typename Element_ , int InterleavedK>

| using cutlass::epilogue::threadblock::InterleavedPredicatedTileIterator< ThreadMap_, Element_, InterleavedK >::ThreadMap = ThreadMap_ |

Constructor & Destructor Documentation

template<typename ThreadMap_ , typename Element_ , int InterleavedK>

|

| CUTLASS_DEVICE cutlass::epilogue::threadblock::InterleavedPredicatedTileIterator< ThreadMap_, Element_, InterleavedK >::InterleavedPredicatedTileIterator | ( | Params const & | params, | | | | Element * | pointer, | | | | TensorCoord | extent, | | | | int | thread_idx, | | | | TensorCoord | threadblock_offset | | | ) | | |

| inline |

Member Function Documentation

template<typename ThreadMap_ , typename Element_ , int InterleavedK>

|

| CUTLASS_HOST_DEVICE void cutlass::epilogue::threadblock::InterleavedPredicatedTileIterator< ThreadMap_, Element_, InterleavedK >::add_pointer_offset | ( | LongIndex | pointer_offset | ) | |

| inline |

template<typename ThreadMap_ , typename Element_ , int InterleavedK>

|

| CUTLASS_DEVICE void cutlass::epilogue::threadblock::InterleavedPredicatedTileIterator< ThreadMap_, Element_, InterleavedK >::clear_mask | ( | | ) | |

| inline |

template<typename ThreadMap_ , typename Element_ , int InterleavedK>

|

| CUTLASS_DEVICE void cutlass::epilogue::threadblock::InterleavedPredicatedTileIterator< ThreadMap_, Element_, InterleavedK >::enable_mask | ( | | ) | |

| inline |

template<typename ThreadMap_ , typename Element_ , int InterleavedK>

|

| CUTLASS_DEVICE void cutlass::epilogue::threadblock::InterleavedPredicatedTileIterator< ThreadMap_, Element_, InterleavedK >::get_mask | ( | Mask & | mask | ) | |

| inline |

template<typename ThreadMap_ , typename Element_ , int InterleavedK>

|

| CUTLASS_DEVICE void cutlass::epilogue::threadblock::InterleavedPredicatedTileIterator< ThreadMap_, Element_, InterleavedK >::load | ( | Fragment & | frag | ) | |

| inline |

template<typename ThreadMap_ , typename Element_ , int InterleavedK>

|

| CUTLASS_HOST_DEVICE InterleavedPredicatedTileIterator& cutlass::epilogue::threadblock::InterleavedPredicatedTileIterator< ThreadMap_, Element_, InterleavedK >::operator++ | ( | | ) | |

| inline |

Efficiently disables all accesses guarded by mask

template<typename ThreadMap_ , typename Element_ , int InterleavedK>

|

| CUTLASS_HOST_DEVICE void cutlass::epilogue::threadblock::InterleavedPredicatedTileIterator< ThreadMap_, Element_, InterleavedK >::set_iteration_index | ( | int | iteration | ) | |

| inline |

template<typename ThreadMap_ , typename Element_ , int InterleavedK>

|

| CUTLASS_DEVICE void cutlass::epilogue::threadblock::InterleavedPredicatedTileIterator< ThreadMap_, Element_, InterleavedK >::set_mask | ( | Mask const & | mask | ) | |

| inline |

template<typename ThreadMap_ , typename Element_ , int InterleavedK>

|

| CUTLASS_DEVICE void cutlass::epilogue::threadblock::InterleavedPredicatedTileIterator< ThreadMap_, Element_, InterleavedK >::store | ( | Fragment const & | frag | ) | |

| inline |

Member Data Documentation

template<typename ThreadMap_ , typename Element_ , int InterleavedK>

|

| int const cutlass::epilogue::threadblock::InterleavedPredicatedTileIterator< ThreadMap_, Element_, InterleavedK >::kElementsPerAccess = ThreadMap::kElementsPerAccess |

| static |

template<typename ThreadMap_ , typename Element_ , int InterleavedK>

|

| int const cutlass::epilogue::threadblock::InterleavedPredicatedTileIterator< ThreadMap_, Element_, InterleavedK >::kIterations = ThreadMap::Iterations::kCount |

| static |

template<typename ThreadMap_ , typename Element_ , int InterleavedK>

|

| int const cutlass::epilogue::threadblock::InterleavedPredicatedTileIterator< ThreadMap_, Element_, InterleavedK >::kThreads = ThreadMap::kThreads |

| static |

---

The documentation for this class was generated from the following file:

• [epilogue/threadblock/predicated_tile_iterator.h](epilogue_2threadblock_2predicated tile iterator_8h_source.html)

---

Generated by 1.8.11