Cutlass
CUDA Templates for Linear Algebra Subroutines and Solvers
Classes | Public Types | Static Public Attributes | List of all members
cutlass::gemm::GemmSharedStoreWithSkewIteratorAbTraits< Scalar_, Tile_, Threads_, kScalarsPerSts_, kSkew_ > Struct Template Reference

#include <gemm_shared_tile.h>

Classes

struct  ThreadOffset
 

Public Types

typedef nv_std::remove_const< Scalar_ >::type Scalar
 The scalar. More...
 
typedef Scalar_ * Pointer
 The pointer. More...
 
typedef ReshapeTile< Tile_, kScalarsPerSts_ >::Tile TileWithoutSkew
 The tile without skews. More...
 
typedef ReshapeTile< Shape< Tile_::kD, Tile_::kH, Tile_::kW+kSkew_ >, kScalarsPerSts_ >::Tile Tile
 The tile. More...
 
typedef Threads_ Threads
 The threads. More...
 
typedef Shape< 0, kScalarsPerSts_, ShapeCount< Tile >::kHwc/Threads::kW > ThreadsStrides
 The strides to compute the base position of the thread. More...
 
typedef Shape< 1, TileWithoutSkew::kH/Threads::kW, TileWithoutSkew::kW/Threads::kH > Iterations
 The number of iterations needed to load/store the tile. More...
 
typedef Shape< 0, ShapeCount< Tile >::kWc, Threads::kH *kScalarsPerStsDelta
 The strides in each dimension between different loads/stores. More...
 

Static Public Attributes

static int const kSkew = kSkew_
 The skew. More...
 
static int const kScalarsPerSts = kScalarsPerSts_
 The number of scalars per STS. More...
 
static MemorySpace::Kind const kMemorySpace = MemorySpace::kShared
 The memory space. More...
 

Member Typedef Documentation

◆ Delta

template<typename Scalar_ , typename Tile_ , typename Threads_ , int kScalarsPerSts_, int kSkew_>
typedef Shape<0, ShapeCount<Tile>::kWc, Threads::kH * kScalarsPerSts> cutlass::gemm::GemmSharedStoreWithSkewIteratorAbTraits< Scalar_, Tile_, Threads_, kScalarsPerSts_, kSkew_ >::Delta

◆ Iterations

template<typename Scalar_ , typename Tile_ , typename Threads_ , int kScalarsPerSts_, int kSkew_>
typedef Shape<1, TileWithoutSkew::kH / Threads::kW, TileWithoutSkew::kW / Threads::kH> cutlass::gemm::GemmSharedStoreWithSkewIteratorAbTraits< Scalar_, Tile_, Threads_, kScalarsPerSts_, kSkew_ >::Iterations

◆ Pointer

template<typename Scalar_ , typename Tile_ , typename Threads_ , int kScalarsPerSts_, int kSkew_>
typedef Scalar_* cutlass::gemm::GemmSharedStoreWithSkewIteratorAbTraits< Scalar_, Tile_, Threads_, kScalarsPerSts_, kSkew_ >::Pointer

◆ Scalar

template<typename Scalar_ , typename Tile_ , typename Threads_ , int kScalarsPerSts_, int kSkew_>
typedef nv_std::remove_const<Scalar_>::type cutlass::gemm::GemmSharedStoreWithSkewIteratorAbTraits< Scalar_, Tile_, Threads_, kScalarsPerSts_, kSkew_ >::Scalar

◆ Threads

template<typename Scalar_ , typename Tile_ , typename Threads_ , int kScalarsPerSts_, int kSkew_>
typedef Threads_ cutlass::gemm::GemmSharedStoreWithSkewIteratorAbTraits< Scalar_, Tile_, Threads_, kScalarsPerSts_, kSkew_ >::Threads

◆ ThreadsStrides

template<typename Scalar_ , typename Tile_ , typename Threads_ , int kScalarsPerSts_, int kSkew_>
typedef Shape<0, kScalarsPerSts_, ShapeCount<Tile>::kHwc / Threads::kW> cutlass::gemm::GemmSharedStoreWithSkewIteratorAbTraits< Scalar_, Tile_, Threads_, kScalarsPerSts_, kSkew_ >::ThreadsStrides

◆ Tile

template<typename Scalar_ , typename Tile_ , typename Threads_ , int kScalarsPerSts_, int kSkew_>
typedef ReshapeTile<Shape<Tile_::kD, Tile_::kH, Tile_::kW + kSkew_>, kScalarsPerSts_>::Tile cutlass::gemm::GemmSharedStoreWithSkewIteratorAbTraits< Scalar_, Tile_, Threads_, kScalarsPerSts_, kSkew_ >::Tile

◆ TileWithoutSkew

template<typename Scalar_ , typename Tile_ , typename Threads_ , int kScalarsPerSts_, int kSkew_>
typedef ReshapeTile<Tile_, kScalarsPerSts_>::Tile cutlass::gemm::GemmSharedStoreWithSkewIteratorAbTraits< Scalar_, Tile_, Threads_, kScalarsPerSts_, kSkew_ >::TileWithoutSkew

Member Data Documentation

◆ kMemorySpace

template<typename Scalar_ , typename Tile_ , typename Threads_ , int kScalarsPerSts_, int kSkew_>
MemorySpace::Kind const cutlass::gemm::GemmSharedStoreWithSkewIteratorAbTraits< Scalar_, Tile_, Threads_, kScalarsPerSts_, kSkew_ >::kMemorySpace = MemorySpace::kShared
static

◆ kScalarsPerSts

template<typename Scalar_ , typename Tile_ , typename Threads_ , int kScalarsPerSts_, int kSkew_>
int const cutlass::gemm::GemmSharedStoreWithSkewIteratorAbTraits< Scalar_, Tile_, Threads_, kScalarsPerSts_, kSkew_ >::kScalarsPerSts = kScalarsPerSts_
static

◆ kSkew

template<typename Scalar_ , typename Tile_ , typename Threads_ , int kScalarsPerSts_, int kSkew_>
int const cutlass::gemm::GemmSharedStoreWithSkewIteratorAbTraits< Scalar_, Tile_, Threads_, kScalarsPerSts_, kSkew_ >::kSkew = kSkew_
static

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