The params.
#include <wmma_gemm_global_tile.h>
Inheritance diagram for cutlass::gemm::WmmaGemmGlobalIteratorCd< TileTraits_, Index_ >::Params:
Public Member Functions | |
| CUTLASS_HOST_DEVICE int | initialize (Pointer pointer, long long batch_stride, Index ldm, Index n, Index epilogue_stride_w, Index epilogue_delta_w) |
| Setup the params. More... | |
 Public Member Functions inherited from cutlass::gemm::GemmGlobalIteratorCd< TileTraits_, Index_ >::Params | |
| CUTLASS_HOST_DEVICE int | initialize (Pointer pointer, long long batch_stride, Index ldm, Index bound, Index epilogue_stride_w, Index epilogue_delta_w) |
| Setup the params. More... | |
Additional Inherited Members | |
| Public Attributes inherited from cutlass::gemm::GemmGlobalIteratorCd< TileTraits_, Index_ >::Params | |
| Pointer | pointer |
| The pointer. More... | |
| long long | stride_d |
| The stride in the D dimension. More... | |
| Index | stride_h |
| The stride in the H dimension to setup the thread in the block. More... | |
| Index | inc_advance |
| The strides to increment the pointer. More... | |
| Index | inc_h |
| Index | predicate_inc_advance |
| The strides to increment the predicate offset. More... | |
| Index | predicate_inc_h |
| Index | predicate_offset |
| The column offset to compute the predicate for the columns. More... | |
Member Function Documentation
◆ initialize()
template<typename TileTraits_ , typename Index_ = int>
|
inline |
The documentation for this struct was generated from the following file:
