CUTLASS
CUDA Templates for Linear Algebra Subroutines and Solvers
Classes | Namespaces
tensor_op_policy.h File Reference

Defines basic structures needed for implementing the warp-scoped phase of the epilogue. These quantities assume a 'column-major' arrangement of TensorOp instructions, of which a row-oriented slice is visible per iteration. More...

#include "cutlass/matrix_shape.h"
#include "cutlass/layout/matrix.h"
Include dependency graph for tensor_op_policy.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

struct  cutlass::epilogue::warp::TensorOpPolicy< WarpShape, OperatorShape, Layout >
 Policy details related to the epilogue. More...
 
struct  cutlass::epilogue::warp::TensorOpPolicy< WarpShape, OperatorShape, layout::RowMajor >
 Partial specialization for row-major. More...
 
struct  cutlass::epilogue::warp::TensorOpPolicy< WarpShape, OperatorShape, layout::ColumnMajorInterleaved< InterleavedK > >
 Partial specialization for column-major-interleaved. More...
 

Namespaces

 cutlass
 
 cutlass::epilogue
 
 cutlass::epilogue::warp