![]() |
CUTLASS
CUDA Templates for Linear Algebra Subroutines and Solvers
|
Epilogue for threadblock scoped GEMMs using Tensor Ops. More...
#include <assert.h>#include "cutlass/cutlass.h"#include "cutlass/numeric_types.h"#include "cutlass/array.h"#include "cutlass/layout/vector.h"#include "cutlass/layout/tensor.h"#include "cutlass/tensor_coord.h"#include "cutlass/aligned_buffer.h"#include "cutlass/gemm/gemm.h"#include "cutlass/transform/pitch_linear_thread_map.h"#include "cutlass/transform/threadblock/regular_tile_iterator.h"#include "cutlass/epilogue/threadblock/epilogue_base.h"#include "cutlass/epilogue/threadblock/predicated_tile_iterator.h"

Go to the source code of this file.
Classes | |
| class | cutlass::epilogue::threadblock::InterleavedEpilogue< Shape_, WarpMmaOperator_, PartitionsK, OutputTileIterator_, AccumulatorFragmentIterator_, OutputOp_, InterleavedK, IsBetaZero > |
| Epilogue operator without splitk. More... | |
| struct | cutlass::epilogue::threadblock::InterleavedEpilogue< Shape_, WarpMmaOperator_, PartitionsK, OutputTileIterator_, AccumulatorFragmentIterator_, OutputOp_, InterleavedK, IsBetaZero >::SharedStorage |
| Shared storage allocation needed by the epilogue. More... | |
Namespaces | |
| 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.
1.8.11