Cutlass
CUDA Templates for Linear Algebra Subroutines and Solvers
Public Types | Static Public Attributes | List of all members
cutlass::gemm::HgemmTileTraitsHelperB< MatrixLayout::kColumnMajor, GemmConfig_ > Struct Template Reference

#include <hgemm_traits.h>

Inheritance diagram for cutlass::gemm::HgemmTileTraitsHelperB< MatrixLayout::kColumnMajor, GemmConfig_ >:
cutlass::gemm::GemmTileTraitsHelperB< MatrixLayout::kColumnMajor, GemmConfig_ >

Public Types

typedef GemmTileTraitsHelperB< MatrixLayout::kColumnMajor, GemmConfig_ > Base
 The base config. More...
 
typedef HgemmCrosswiseGlobalTileTraits< GemmOperand::kB, MatrixLayout::kColumnMajor, half const, Shape< 1, GemmConfig_::OutputTile::kH, GemmConfig_::OutputTile::kD >, Shape< 1, GemmConfig_::kThreads/GemmConfig_::OutputTile::kD, GemmConfig_::OutputTile::kD >, GemmConfig_::kScalarsPerLdgB > GlobalTileTraits
 The traits class to build the iterator to load data from global memory for B^N. More...
 
- Public Types inherited from cutlass::gemm::GemmTileTraitsHelperB< MatrixLayout::kColumnMajor, GemmConfig_ >
typedef GemmConfig_::ScalarB Scalar
 The input scalar. More...
 
typedef GemmConfig_::MultiplyAdd::ScalarB MultiplyAddScalar
 The scalar stored in shared memory. More...
 
typedef GemmGlobalTileTraits< GemmOperand::kB, MatrixLayout::kColumnMajor, Scalar const, Shape< 1, GemmConfig_::OutputTile::kH, GemmConfig_::OutputTile::kD >, Shape< 1, GemmConfig_::kThreads/GemmConfig_::OutputTile::kD, GemmConfig_::OutputTile::kD >, GemmConfig_::kScalarsPerLdgB > GlobalTileTraits
 The traits class to build the iterator to load data from global memory for B^N. More...
 

Static Public Attributes

static int const kSkewB = 128 / sizeof(half) / GlobalTileTraits::Threads::kW / 2
 
- Static Public Attributes inherited from cutlass::gemm::GemmTileTraitsHelperB< MatrixLayout::kColumnMajor, GemmConfig_ >
static MatrixLayout::Kind const kLayout = MatrixLayout::kColumnMajor
 The layout. More...
 
static int const kScalarsIn4B = sizeof(MultiplyAddScalar) > 4 ? 1 : 4 / sizeof(MultiplyAddScalar)
 The number of scalars in 4B. More...
 
static int const kSkewB
 The skew for B. More...
 

Member Typedef Documentation

◆ Base

template<typename GemmConfig_ >
typedef GemmTileTraitsHelperB<MatrixLayout::kColumnMajor, GemmConfig_> cutlass::gemm::HgemmTileTraitsHelperB< MatrixLayout::kColumnMajor, GemmConfig_ >::Base

◆ GlobalTileTraits

template<typename GemmConfig_ >
typedef HgemmCrosswiseGlobalTileTraits< GemmOperand::kB, MatrixLayout::kColumnMajor, half const, Shape<1, GemmConfig_::OutputTile::kH, GemmConfig_::OutputTile::kD>, Shape<1, GemmConfig_::kThreads / GemmConfig_::OutputTile::kD, GemmConfig_::OutputTile::kD>, GemmConfig_::kScalarsPerLdgB> cutlass::gemm::HgemmTileTraitsHelperB< MatrixLayout::kColumnMajor, GemmConfig_ >::GlobalTileTraits

Member Data Documentation

◆ kSkewB

template<typename GemmConfig_ >
int const cutlass::gemm::HgemmTileTraitsHelperB< MatrixLayout::kColumnMajor, GemmConfig_ >::kSkewB = 128 / sizeof(half) / GlobalTileTraits::Threads::kW / 2
static

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