CUTLASS
CUDA Templates for Linear Algebra Subroutines and Solvers
|
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/epilogue/threadblock/output_tile_thread_map.h"
Go to the source code of this file.
Classes | |
class | cutlass::epilogue::threadblock::SharedLoadIterator< ThreadMap_, Element_, MaxAlignment > |
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.