cutlass::gemm::GemmEpilogue< GemmEpilogueTraits_ > Struct Template Reference

`#include <gemm_epilogue.h>`

Inheritance diagram for cutlass::gemm::GemmEpilogue< GemmEpilogueTraits_ >:

## Public Types

typedef GemmEpilogueTraits_ | Traits
| The traits class. More...
typedef Traits::Params | Params
| The params. More...
typedef Traits::SharedStorage | SharedStorage
| The shared storage. More...
typedef Traits::OutputTile | OutputTile
| The output tile. More...
typedef Traits::Iterations | Iterations
| The number of iterations. More...
typedef Traits::Accumulators | Accumulators
| The accumulators. More...
typedef Traits::Scalar | Scalar
| The scalar. More...
typedef Traits::Functor | Functor
| The functor in charge of the math. More...
typedef Traits::GlobalLoadIteratorC | GlobalLoadIteratorC
| We do not support 3D or 4D shapes. More...
typedef Traits::GlobalTransformerC | GlobalTransformerC
| The transformer for C. More...
typedef Traits::GlobalTransformerD | GlobalTransformerD
| The transformer for D. More...
typedef Traits::GlobalStoreIteratorD | GlobalStoreIteratorD
| The iterator for D in global memory. More...
typedef Traits::SharedStoreIteratorD | SharedStoreIteratorD
| The iterator to store D in shared memory. More...
typedef Traits::SharedStoreTransformerD | SharedStoreTransformerD
| The shared store transformer for D. More...
typedef Traits::SharedLoadIteratorD | SharedLoadIteratorD
| The iterator to load D in shared memory. More...
typedef Copy< typename SharedLoadIteratorD::Fragment > | SharedLoadTransformerD
| The shared load transformer for D. More...
typedef Traits::Index | Index
| The index. More...
typedef GlobalLoadIteratorC::Scalar | ScalarC
| The scalar for C. More...
typedef GlobalStoreIteratorD::Scalar | ScalarD
| The scalar for D. More...

## Public Member Functions

CUTLASS_DEVICE | GemmEpilogue (Params const &params_, SharedStorage &shared_storage_, Index m_, Index n_)
| Ctor. More...
CUTLASS_DEVICE void | epilogue (Coord< 3 > const &block, Accumulators &accumulators)
| Execute the epilogue. More...
template<bool kBetaIsZero_>
CUTLASS_DEVICE void | epilogue_with_or_without_beta (Coord< 3 > const &block, Accumulators &accumulators)
CUTLASS_DEVICE void | shared_load_fence ()
| The memory fence for shared loads. More...
CUTLASS_DEVICE void | shared_store_fence ()
| The memory fence for shared stores. More...

## Public Attributes

Params const & | params
| The params. More...
SharedStorage & | shared_storage
| The shared storage. More...
Index | m
| The dimensions of the GEMM. More...
Index | n

## Member Typedef Documentation

## Accumulators

template<typename GemmEpilogueTraits_ >

typedef Traits::Accumulators cutlass::gemm::GemmEpilogue< GemmEpilogueTraits_ >::Accumulators

## Functor

template<typename GemmEpilogueTraits_ >

typedef Traits::Functor cutlass::gemm::GemmEpilogue< GemmEpilogueTraits_ >::Functor

## GlobalLoadIteratorC

template<typename GemmEpilogueTraits_ >

typedef Traits::GlobalLoadIteratorC cutlass::gemm::GemmEpilogue< GemmEpilogueTraits_ >::GlobalLoadIteratorC

The iterator for C in global memory.

## GlobalStoreIteratorD

template<typename GemmEpilogueTraits_ >

typedef Traits::GlobalStoreIteratorD cutlass::gemm::GemmEpilogue< GemmEpilogueTraits_ >::GlobalStoreIteratorD

## GlobalTransformerC

template<typename GemmEpilogueTraits_ >

typedef Traits::GlobalTransformerC cutlass::gemm::GemmEpilogue< GemmEpilogueTraits_ >::GlobalTransformerC

## GlobalTransformerD

template<typename GemmEpilogueTraits_ >

typedef Traits::GlobalTransformerD cutlass::gemm::GemmEpilogue< GemmEpilogueTraits_ >::GlobalTransformerD

## Index

template<typename GemmEpilogueTraits_ >

typedef Traits::Index cutlass::gemm::GemmEpilogue< GemmEpilogueTraits_ >::Index

## Iterations

template<typename GemmEpilogueTraits_ >

typedef Traits::Iterations cutlass::gemm::GemmEpilogue< GemmEpilogueTraits_ >::Iterations

## OutputTile

template<typename GemmEpilogueTraits_ >

typedef Traits::OutputTile cutlass::gemm::GemmEpilogue< GemmEpilogueTraits_ >::OutputTile

## Params

template<typename GemmEpilogueTraits_ >

typedef Traits::Params cutlass::gemm::GemmEpilogue< GemmEpilogueTraits_ >::Params

## Scalar

template<typename GemmEpilogueTraits_ >

typedef Traits::Scalar cutlass::gemm::GemmEpilogue< GemmEpilogueTraits_ >::Scalar

## ScalarC

template<typename GemmEpilogueTraits_ >

typedef GlobalLoadIteratorC::Scalar cutlass::gemm::GemmEpilogue< GemmEpilogueTraits_ >::ScalarC

## ScalarD

template<typename GemmEpilogueTraits_ >

typedef GlobalStoreIteratorD::Scalar cutlass::gemm::GemmEpilogue< GemmEpilogueTraits_ >::ScalarD

## SharedLoadIteratorD

template<typename GemmEpilogueTraits_ >

typedef Traits::SharedLoadIteratorD cutlass::gemm::GemmEpilogue< GemmEpilogueTraits_ >::SharedLoadIteratorD

## SharedLoadTransformerD

template<typename GemmEpilogueTraits_ >

typedef Copy<typename SharedLoadIteratorD::Fragment> cutlass::gemm::GemmEpilogue< GemmEpilogueTraits_ >::SharedLoadTransformerD

## SharedStorage

template<typename GemmEpilogueTraits_ >

typedef Traits::SharedStorage cutlass::gemm::GemmEpilogue< GemmEpilogueTraits_ >::SharedStorage

## SharedStoreIteratorD

template<typename GemmEpilogueTraits_ >

typedef Traits::SharedStoreIteratorD cutlass::gemm::GemmEpilogue< GemmEpilogueTraits_ >::SharedStoreIteratorD

## SharedStoreTransformerD

template<typename GemmEpilogueTraits_ >

typedef Traits::SharedStoreTransformerD cutlass::gemm::GemmEpilogue< GemmEpilogueTraits_ >::SharedStoreTransformerD

## Traits

template<typename GemmEpilogueTraits_ >

typedef GemmEpilogueTraits_ cutlass::gemm::GemmEpilogue< GemmEpilogueTraits_ >::Traits

## Constructor & Destructor Documentation

## GemmEpilogue()

template<typename GemmEpilogueTraits_ >

CUTLASS_DEVICE cutlass::gemm::GemmEpilogue< GemmEpilogueTraits_ >::GemmEpilogue | ( | Params const & | params_,
|  | SharedStorage & | shared_storage_,
|  | Index | m_,
|  | Index | n_
| ) |  |

inline

## Member Function Documentation

## epilogue()

template<typename GemmEpilogueTraits_ >

CUTLASS_DEVICE void cutlass::gemm::GemmEpilogue< GemmEpilogueTraits_ >::epilogue | ( | Coord< 3 > const & | block,
|  | Accumulators & | accumulators
| ) |  |

inline

## epilogue_with_or_without_beta()

template<typename GemmEpilogueTraits_ >
template<bool kBetaIsZero_>

CUTLASS_DEVICE void cutlass::gemm::GemmEpilogue< GemmEpilogueTraits_ >::epilogue_with_or_without_beta | ( | Coord< 3 > const & | block,
|  | Accumulators & | accumulators
| ) |  |

inline

## shared_load_fence()

template<typename GemmEpilogueTraits_ >

CUTLASS_DEVICE void cutlass::gemm::GemmEpilogue< GemmEpilogueTraits_ >::shared_load_fence | ( |  | ) |

inline

## shared_store_fence()

template<typename GemmEpilogueTraits_ >

CUTLASS_DEVICE void cutlass::gemm::GemmEpilogue< GemmEpilogueTraits_ >::shared_store_fence | ( |  | ) |

inline

## Member Data Documentation

## m

template<typename GemmEpilogueTraits_ >

Index cutlass::gemm::GemmEpilogue< GemmEpilogueTraits_ >::m

## n

template<typename GemmEpilogueTraits_ >

Index cutlass::gemm::GemmEpilogue< GemmEpilogueTraits_ >::n

## params

template<typename GemmEpilogueTraits_ >

Params const& cutlass::gemm::GemmEpilogue< GemmEpilogueTraits_ >::params

## shared_storage

template<typename GemmEpilogueTraits_ >

SharedStorage& cutlass::gemm::GemmEpilogue< GemmEpilogueTraits_ >::shared_storage

The documentation for this struct was generated from the following file:

- gemm_epilogue.h