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

#include <gemm_global_tile.h>

Inheritance diagram for cutlass::gemm::GemmGlobalIteratorCdTraits< Scalar_, Tile_, Threads_, kStrideH_, kAccessSize_ >:
cutlass::gemm::GemmGlobalTileTraits< GemmOperand::kC, MatrixLayout::kColumnMajor, Scalar_, Tile_, Threads_, kAccessSize_ >

Classes

struct  ThreadOffset
 Computes the thread offset in (H, W) based on thread ID. More...
 

Public Types

typedef GemmGlobalTileTraits< GemmOperand::kC, MatrixLayout::kColumnMajor, Scalar_, Tile_, Threads_, kAccessSize_ > Base
 The base class. More...
 
typedef Shape< 0, 0, Base::Delta::kW, Base::Delta::kCDelta
 Override the strides in each dimension between different loads/stores. More...
 
typedef Base::Iterations Iterations
 
typedef Base::Threads Threads
 
typedef Base::ThreadsStrides ThreadsStrides
 
- Public Types inherited from cutlass::gemm::GemmGlobalTileTraits< GemmOperand::kC, MatrixLayout::kColumnMajor, Scalar_, Tile_, Threads_, kAccessSize_ >
typedef nv_std::remove_const< Scalar_ >::type Scalar
 The scalar. More...
 
typedef Scalar_ * Pointer
 The pointer. More...
 
typedef ReshapeTile< Tile_, kAccessSize_ >::Tile Tile
 The tile. More...
 
typedef ReshapeThreads< Tile, Threads_ >::Threads Threads
 The threads. More...
 
typedef Shape< 1, 1, Tile::kC > ThreadsStrides
 The threads strides. More...
 
typedef Shape< 0, Threads::kH, Threads::kW *kAccessSizeDelta
 The strides in each dimension between different loads/stores. More...
 
typedef Shape< 1, Tile::kH/Threads::kH, Tile::kW/Threads::kW, Tile::kC/kAccessSizeIterations
 The number of iterations needed to load/store the tile. More...
 
typedef GemmMultiplicandTraits< Tile, kOperand, kLayoutMultiplicandTraits
 

Static Public Attributes

static int const kStrideH = kStrideH_
 The stride in the H dimension. More...
 
- Static Public Attributes inherited from cutlass::gemm::GemmGlobalTileTraits< GemmOperand::kC, MatrixLayout::kColumnMajor, Scalar_, Tile_, Threads_, kAccessSize_ >
static GemmOperand::Kind const kOperand
 Identity of the operand. More...
 
static MatrixLayout::Kind const kLayout
 The layout. More...
 
static bool const kIsContiguous
 Is it A^N or B^T? More...
 
static int const kAccessSize
 The number of scalars per LDG/STG. More...
 
static MemorySpace::Kind const kMemorySpace
 The memory space. More...
 

Member Typedef Documentation

◆ Base

template<typename Scalar_ , typename Tile_ , typename Threads_ , int kStrideH_, int kAccessSize_>
typedef GemmGlobalTileTraits<GemmOperand::kC, MatrixLayout::kColumnMajor, Scalar_, Tile_, Threads_, kAccessSize_> cutlass::gemm::GemmGlobalIteratorCdTraits< Scalar_, Tile_, Threads_, kStrideH_, kAccessSize_ >::Base

◆ Delta

template<typename Scalar_ , typename Tile_ , typename Threads_ , int kStrideH_, int kAccessSize_>
typedef Shape<0, 0, Base::Delta::kW, Base::Delta::kC> cutlass::gemm::GemmGlobalIteratorCdTraits< Scalar_, Tile_, Threads_, kStrideH_, kAccessSize_ >::Delta

◆ Iterations

template<typename Scalar_ , typename Tile_ , typename Threads_ , int kStrideH_, int kAccessSize_>
typedef Base::Iterations cutlass::gemm::GemmGlobalIteratorCdTraits< Scalar_, Tile_, Threads_, kStrideH_, kAccessSize_ >::Iterations

◆ Threads

template<typename Scalar_ , typename Tile_ , typename Threads_ , int kStrideH_, int kAccessSize_>
typedef Base::Threads cutlass::gemm::GemmGlobalIteratorCdTraits< Scalar_, Tile_, Threads_, kStrideH_, kAccessSize_ >::Threads

◆ ThreadsStrides

template<typename Scalar_ , typename Tile_ , typename Threads_ , int kStrideH_, int kAccessSize_>
typedef Base::ThreadsStrides cutlass::gemm::GemmGlobalIteratorCdTraits< Scalar_, Tile_, Threads_, kStrideH_, kAccessSize_ >::ThreadsStrides

Member Data Documentation

◆ kStrideH

template<typename Scalar_ , typename Tile_ , typename Threads_ , int kStrideH_, int kAccessSize_>
int const cutlass::gemm::GemmGlobalIteratorCdTraits< Scalar_, Tile_, Threads_, kStrideH_, kAccessSize_ >::kStrideH = kStrideH_
static

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