Cutlass
CUDA Templates for Linear Algebra Subroutines and Solvers
|
The params.
#include <gemm_global_tile.h>
Public Member Functions | |
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... | |
Public Attributes | |
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... | |
|
inline |
Index cutlass::gemm::GemmGlobalIteratorCd< TileTraits_, Index_ >::Params::inc_advance |
Index cutlass::gemm::GemmGlobalIteratorCd< TileTraits_, Index_ >::Params::inc_h |
Pointer cutlass::gemm::GemmGlobalIteratorCd< TileTraits_, Index_ >::Params::pointer |
Index cutlass::gemm::GemmGlobalIteratorCd< TileTraits_, Index_ >::Params::predicate_inc_advance |
Index cutlass::gemm::GemmGlobalIteratorCd< TileTraits_, Index_ >::Params::predicate_inc_h |
Index cutlass::gemm::GemmGlobalIteratorCd< TileTraits_, Index_ >::Params::predicate_offset |
long long cutlass::gemm::GemmGlobalIteratorCd< TileTraits_, Index_ >::Params::stride_d |
Index cutlass::gemm::GemmGlobalIteratorCd< TileTraits_, Index_ >::Params::stride_h |