Back to Cutlass

CUTLASS: tensor_op_policy.h File Reference

docs/tensor__op__policy_8h.html

4.4.22.5 KB
Original Source

| | 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.](tensor op policy_8h_source.html)

|

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 | | |


Generated by 1.8.11