Back to Cutlass

CUTLASS: predicated_tile_iterator.h File Reference

docs/epilogue_2threadblock_2predicated__tile__iterator_8h.html

4.4.23.4 KB
Original Source

| | CUTLASS

CUDA Templates for Linear Algebra Subroutines and Solvers |

Classes | Namespaces

epilogue/threadblock/predicated_tile_iterator.h File Reference

Epilogue for threadblock scoped GEMMs using Tensor Ops. More...

#include "cutlass/cutlass.h"
#include "cutlass/numeric_types.h"
#include "cutlass/array.h"
#include "cutlass/layout/matrix.h"
#include "cutlass/matrix_shape.h"
#include "cutlass/tensor_ref.h"
#include "cutlass/transform/pitch_linear_thread_map.h"
#include "cutlass/epilogue/threadblock/output_tile_thread_map.h"

Include dependency graph for epilogue/threadblock/predicated_tile_iterator.h:

This graph shows which files directly or indirectly include this file:

[Go to the source code of this file.](epilogue_2threadblock_2predicated tile iterator_8h_source.html)

|

Classes

| | class | cutlass::epilogue::threadblock::PredicatedTileIterator< ThreadMap_, Element_ > | | | | struct | cutlass::epilogue::threadblock::PredicatedTileIterator< ThreadMap_, Element_ >::Params | | | | struct | cutlass::epilogue::threadblock::PredicatedTileIterator< ThreadMap_, Element_ >::Mask | | | Mask object. More...
| | | | class | cutlass::epilogue::threadblock::InterleavedPredicatedTileIterator< ThreadMap_, Element_, InterleavedK > | | | | struct | cutlass::epilogue::threadblock::InterleavedPredicatedTileIterator< ThreadMap_, Element_, InterleavedK >::Params | | | | struct | cutlass::epilogue::threadblock::InterleavedPredicatedTileIterator< ThreadMap_, Element_, InterleavedK >::Mask | | | Mask object. More...
| | |

|

Namespaces

| | | cutlass | | | | | cutlass::epilogue | | | | | cutlass::epilogue::threadblock | | |

Detailed Description

The epilogue rearranges the result of a matrix product through shared memory to match canonical tensor layouts in global memory. Epilogues support conversion and reduction operations.


Generated by 1.8.11