docs/default__epilogue__tensor__op_8h.html
| | CUTLASS
CUDA Templates for Linear Algebra Subroutines and Solvers |
default_epilogue_tensor_op.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/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_tensor_op.h"
#include "cutlass/epilogue/warp/tile_iterator_tensor_op.h"
#include "cutlass/epilogue/threadblock/default_thread_map_tensor_op.h"
#include "cutlass/epilogue/threadblock/predicated_tile_iterator.h"
#include "cutlass/epilogue/threadblock/shared_load_iterator.h"
#include "cutlass/epilogue/threadblock/epilogue.h"
#include "cutlass/epilogue/threadblock/interleaved_epilogue.h"
Include dependency graph for default_epilogue_tensor_op.h:
This graph shows which files directly or indirectly include this file:
[Go to the source code of this file.](default epilogue tensor__op_8h_source.html)
|
|
| struct | cutlass::epilogue::threadblock::DefaultEpilogueTensorOp< Shape_, WarpMmaTensorOp_, PartitionsK, OutputOp_, ElementsPerAccess > |
| | Defines sensible defaults for epilogues for TensorOps. More...
|
| |
| struct | cutlass::epilogue::threadblock::DefaultInterleavedEpilogueTensorOp< Shape_, WarpMmaTensorOp_, PartitionsK, OutputOp_, ElementsPerAccess, InterleavedK, IsBetaZero, isSplitK > |
| |
|
| | | cutlass | | | | | cutlass::epilogue | | | | | cutlass::epilogue::threadblock | | |
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