Back to Cutlass

CUTLASS: default_epilogue_simt.h File Reference

docs/default__epilogue__simt_8h.html

4.4.22.4 KB
Original Source

| | CUTLASS

CUDA Templates for Linear Algebra Subroutines and Solvers |

Classes | Namespaces

default_epilogue_simt.h File Reference

Epilogue for threadblock scoped GEMMs using SIMT. More...

#include "cutlass/cutlass.h"
#include "cutlass/numeric_types.h"
#include "cutlass/array.h"
#include "cutlass/gemm/gemm.h"
#include "cutlass/epilogue/thread/linear_combination.h"
#include "cutlass/epilogue/thread/conversion_op.h"
#include "cutlass/epilogue/thread/reduction_op.h"
#include "cutlass/transform/threadblock/regular_tile_iterator_pitch_linear.h"
#include "cutlass/epilogue/warp/fragment_iterator_simt.h"
#include "cutlass/epilogue/warp/tile_iterator_simt.h"
#include "cutlass/epilogue/threadblock/default_thread_map_simt.h"
#include "cutlass/epilogue/threadblock/predicated_tile_iterator.h"
#include "cutlass/epilogue/threadblock/shared_load_iterator.h"
#include "cutlass/epilogue/threadblock/epilogue.h"

Include dependency graph for default_epilogue_simt.h:

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

[Go to the source code of this file.](default epilogue simt_8h_source.html)

|

Classes

| | struct | cutlass::epilogue::threadblock::DefaultEpilogueSimt< Shape_, WarpMmaSimt_, OutputOp_, ElementsPerAccess > | | | Defines sensible defaults for epilogues for SimtOps. 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