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

#include <igemm_traits.h>

Inheritance diagram for cutlass::gemm::IgemmTileTraitsHelperA< MatrixLayout::kColumnMajor, GemmConfig_, Index_ >:
cutlass::gemm::GemmTileTraitsHelperA< MatrixLayout::kColumnMajor, GemmConfig_ >

Public Types

typedef GemmTileTraitsHelperA< MatrixLayout::kColumnMajor, GemmConfig_ > Base
 The base config. More...
 
typedef IgemmGlobalTileTraits< GemmOperand::kA, MatrixLayout::kColumnMajor, int8_t const, Shape< 1, GemmConfig_::OutputTile::kD, GemmConfig_::OutputTile::kW >, Shape< 1, ShapeCount< typename GemmConfig_::Warps >::kCount, GemmConfig_::kWarpSize >, GemmConfig_::kScalarsPerLdgA > GlobalTileTraits
 The traits class to build the iterator to load data from global memory for A^N. More...
 
typedef GemmGlobalIteratorAb< GlobalTileTraits, Index_ > GlobalLoadIterator
 The global load iterator. More...
 
typedef GemmSharedStoreTileAbTraits< int8_t, Shape< GemmConfig_::kStages, GemmConfig_::OutputTile::kD/4, GemmConfig_::OutputTile::kW *4 >, typename GlobalTileTraits::Threads, kScalarsPerStsASharedStoreTileTraits
 The traits class to build the iterator to store data to shared memory for A^N. More...
 
- Public Types inherited from cutlass::gemm::GemmTileTraitsHelperA< MatrixLayout::kColumnMajor, GemmConfig_ >
typedef GemmConfig_::ScalarA Scalar
 The input scalar. More...
 
typedef GemmConfig_::MultiplyAdd::ScalarA MultiplyAddScalar
 The scalar stored in shared memory. More...
 
typedef GemmGlobalTileTraits< GemmOperand::kA, MatrixLayout::kColumnMajor, Scalar const, Shape< 1, GemmConfig_::OutputTile::kD, GemmConfig_::OutputTile::kW >, Shape< 1, ShapeCount< typename GemmConfig_::Warps >::kCount, GemmConfig_::kWarpSize >, GemmConfig_::kScalarsPerLdgA > GlobalTileTraits
 The traits class to build the iterator to load data from global memory for A^N. More...
 
typedef GemmSharedStoreTileAbTraits< MultiplyAddScalar, Shape< GemmConfig_::kStages, GemmConfig_::OutputTile::kD/GemmConfig_::InstructionShape::kD, GemmConfig_::OutputTile::kW *GemmConfig_::InstructionShape::kD >, typename GlobalTileTraits::Threads, GemmConfig_::kScalarsPerStsA > SharedStoreTileTraits
 The traits class to build the iterator to store data to shared memory for A^N. More...
 
typedef GemmSharedLoadTileATraits< MultiplyAddScalar const, typename GemmConfig_::OutputTile, typename GemmConfig_::Warps, typename GemmConfig_::MultiplyAdd::ThreadsPerWarp, typename GemmConfig_::InstructionShape, GemmConfig_::kStages, GemmConfig_::kScalarsPerLdsA, 0 > SharedLoadTileTraits
 The traits class to build the iterator to load from shared memory for A^N. More...
 

Static Public Attributes

static int const kScalarsPerStsA = 16
 The number of scalars per LDG/STS/LDS for A. More...
 
- Static Public Attributes inherited from cutlass::gemm::GemmTileTraitsHelperA< MatrixLayout::kColumnMajor, GemmConfig_ >
static MatrixLayout::Kind const kLayout = MatrixLayout::kColumnMajor
 The layout. More...
 

Member Typedef Documentation

◆ Base

template<typename GemmConfig_ , typename Index_ >
typedef GemmTileTraitsHelperA<MatrixLayout::kColumnMajor, GemmConfig_> cutlass::gemm::IgemmTileTraitsHelperA< MatrixLayout::kColumnMajor, GemmConfig_, Index_ >::Base

◆ GlobalLoadIterator

template<typename GemmConfig_ , typename Index_ >
typedef GemmGlobalIteratorAb<GlobalTileTraits, Index_> cutlass::gemm::IgemmTileTraitsHelperA< MatrixLayout::kColumnMajor, GemmConfig_, Index_ >::GlobalLoadIterator

◆ GlobalTileTraits

template<typename GemmConfig_ , typename Index_ >
typedef IgemmGlobalTileTraits< GemmOperand::kA, MatrixLayout::kColumnMajor, int8_t const, Shape<1, GemmConfig_::OutputTile::kD, GemmConfig_::OutputTile::kW>, Shape<1, ShapeCount<typename GemmConfig_::Warps>::kCount, GemmConfig_::kWarpSize>, GemmConfig_::kScalarsPerLdgA> cutlass::gemm::IgemmTileTraitsHelperA< MatrixLayout::kColumnMajor, GemmConfig_, Index_ >::GlobalTileTraits

◆ SharedStoreTileTraits

template<typename GemmConfig_ , typename Index_ >
typedef GemmSharedStoreTileAbTraits< int8_t, Shape<GemmConfig_::kStages, GemmConfig_::OutputTile::kD / 4, GemmConfig_::OutputTile::kW * 4>, typename GlobalTileTraits::Threads, kScalarsPerStsA> cutlass::gemm::IgemmTileTraitsHelperA< MatrixLayout::kColumnMajor, GemmConfig_, Index_ >::SharedStoreTileTraits

Member Data Documentation

◆ kScalarsPerStsA

template<typename GemmConfig_ , typename Index_ >
int const cutlass::gemm::IgemmTileTraitsHelperA< MatrixLayout::kColumnMajor, GemmConfig_, Index_ >::kScalarsPerStsA = 16
static

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