Cutlass CUDA Templates for Linear Algebra Subroutines and Solvers

Classes | Namespaces
gemm_traits.h File Reference

Defines structural properties of complete GEMM computation. More...

`#include "cutlass/convert.h"`
`#include "cutlass/matrix_traits.h"`
`#include "cutlass/reshape_tile.h"`
`#include "cutlass/tile_allocation.h"`
`#include "cutlass/tile_iterator.h"`
`#include "cutlass/kernel_launch.h"`
`#include "cutlass/gemm/clear_accumulators.h"`
`#include "cutlass/gemm/gemm_config.h"`
`#include "cutlass/gemm/gemm_desc.h"`
`#include "cutlass/gemm/gemm_stream_pair.h"`
`#include "cutlass/gemm/gemm_global_stream.h"`
`#include "cutlass/gemm/gemm_operand.h"`
`#include "cutlass/gemm/gemm_shared_stream.h"`
`#include "cutlass/gemm/threadblock_swizzle.h"`
`#include "cutlass/gemm/gemm.h"`

Go to the source code of this file.

## Classes
struct | cutlass::gemm::GemmTileTraitsHelperA< Kind, GemmConfig_ >
struct | cutlass::gemm::GemmTileTraitsHelperA< MatrixLayout::kColumnMajor, GemmConfig_ >
struct | cutlass::gemm::GemmTileTraitsHelperA< MatrixLayout::kRowMajor, GemmConfig_ >
struct | cutlass::gemm::GemmTileTraitsHelperB< Kind, GemmConfig_ >
struct | cutlass::gemm::GemmTileTraitsHelperB< MatrixLayout::kColumnMajor, GemmConfig_ >
struct | cutlass::gemm::GemmTileTraitsHelperB< MatrixLayout::kRowMajor, GemmConfig_ >
struct | cutlass::gemm::GemmTraits< GemmConfig_, GlobalLoadStreamA_, GlobalLoadStreamB_, SharedLoadStreamA_, SharedLoadStreamB_, Epilogue_, BlockSwizzle_, Index_, ClearAccumulators_ >
struct | cutlass::gemm::GemmTraits< GemmConfig_, GlobalLoadStreamA_, GlobalLoadStreamB_, SharedLoadStreamA_, SharedLoadStreamB_, Epilogue_, BlockSwizzle_, Index_, ClearAccumulators_ >::Params
| Parameters object constructable on the host. More...
struct | cutlass::gemm::GemmTraits< GemmConfig_, GlobalLoadStreamA_, GlobalLoadStreamB_, SharedLoadStreamA_, SharedLoadStreamB_, Epilogue_, BlockSwizzle_, Index_, ClearAccumulators_ >::MainLoopSharedStorage
union | cutlass::gemm::GemmTraits< GemmConfig_, GlobalLoadStreamA_, GlobalLoadStreamB_, SharedLoadStreamA_, SharedLoadStreamB_, Epilogue_, BlockSwizzle_, Index_, ClearAccumulators_ >::SharedStorage
| The storage in shared memory. More...
struct | cutlass::gemm::SimplifiedGemmTraitsHelper< GemmTileTraitsHelperA_, GemmTileTraitsHelperB_, Index_ >
struct | cutlass::gemm::SimplifiedGemmTraits< kLayoutA_, kLayoutB_, GemmConfig_, Epilogue_, Index_, GemmTileTraitsHelperA_, GemmTileTraitsHelperB_, Helper_ >

## Namespaces
| cutlass
| cutlass::gemm