#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 ¶ms_, 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_ >
|
inline |
Member Function Documentation
◆ epilogue()
template<typename GemmEpilogueTraits_ >
|
inline |
◆ epilogue_with_or_without_beta()
template<typename GemmEpilogueTraits_ >
template<bool kBetaIsZero_>
|
inline |
◆ shared_load_fence()
template<typename GemmEpilogueTraits_ >
|
inline |
◆ shared_store_fence()
template<typename GemmEpilogueTraits_ >
|
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:
