Cutlass
CUDA Templates for Linear Algebra Subroutines and Solvers
Public Member Functions | Public Attributes | List of all members
cutlass::gemm::GemmGlobalIteratorCd< TileTraits_, Index_ >::Params Struct Reference

The params.

#include <gemm_global_tile.h>

Inheritance diagram for cutlass::gemm::GemmGlobalIteratorCd< TileTraits_, Index_ >::Params:
cutlass::gemm::WmmaGemmGlobalIteratorCd< TileTraits_, Index_ >::Params

Public Member Functions

CUTLASS_HOST_DEVICE int initialize (Pointer pointer, int stride_d_, Index ldm, Index bound, Index epilogue_stride_w, Index epilogue_delta_w)
 Setup the params. More...
 
CUTLASS_HOST_DEVICE int initialize (Pointer pointer, long long _stride_d, Index _stride_h, Index _inc_advance, Index _inc_h, Index _predicate_inc_advance, Index _predicate_inc_h, Index _predicate_offset)
 

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

Member Function Documentation

◆ initialize() [1/2]

template<typename TileTraits_ , typename Index_ = int>
CUTLASS_HOST_DEVICE int cutlass::gemm::GemmGlobalIteratorCd< TileTraits_, Index_ >::Params::initialize ( Pointer  pointer,
int  stride_d_,
Index  ldm,
Index  bound,
Index  epilogue_stride_w,
Index  epilogue_delta_w 
)
inline

◆ initialize() [2/2]

template<typename TileTraits_ , typename Index_ = int>
CUTLASS_HOST_DEVICE int cutlass::gemm::GemmGlobalIteratorCd< TileTraits_, Index_ >::Params::initialize ( Pointer  pointer,
long long  _stride_d,
Index  _stride_h,
Index  _inc_advance,
Index  _inc_h,
Index  _predicate_inc_advance,
Index  _predicate_inc_h,
Index  _predicate_offset 
)
inline

Member Data Documentation

◆ inc_advance

template<typename TileTraits_ , typename Index_ = int>
Index cutlass::gemm::GemmGlobalIteratorCd< TileTraits_, Index_ >::Params::inc_advance

◆ inc_h

template<typename TileTraits_ , typename Index_ = int>
Index cutlass::gemm::GemmGlobalIteratorCd< TileTraits_, Index_ >::Params::inc_h

◆ pointer

template<typename TileTraits_ , typename Index_ = int>
Pointer cutlass::gemm::GemmGlobalIteratorCd< TileTraits_, Index_ >::Params::pointer

◆ predicate_inc_advance

template<typename TileTraits_ , typename Index_ = int>
Index cutlass::gemm::GemmGlobalIteratorCd< TileTraits_, Index_ >::Params::predicate_inc_advance

◆ predicate_inc_h

template<typename TileTraits_ , typename Index_ = int>
Index cutlass::gemm::GemmGlobalIteratorCd< TileTraits_, Index_ >::Params::predicate_inc_h

◆ predicate_offset

template<typename TileTraits_ , typename Index_ = int>
Index cutlass::gemm::GemmGlobalIteratorCd< TileTraits_, Index_ >::Params::predicate_offset

◆ stride_d

template<typename TileTraits_ , typename Index_ = int>
long long cutlass::gemm::GemmGlobalIteratorCd< TileTraits_, Index_ >::Params::stride_d

◆ stride_h

template<typename TileTraits_ , typename Index_ = int>
Index cutlass::gemm::GemmGlobalIteratorCd< TileTraits_, Index_ >::Params::stride_h

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