Cutlass
CUDA Templates for Linear Algebra Subroutines and Solvers
Classes | Namespaces
hgemm_traits.h File Reference

Defies structural properties of half-precision GEMM computation. More...

#include "cutlass/convert.h"
#include "cutlass/reshape_tile.h"
#include "cutlass/gemm/gemm.h"
#include "cutlass/gemm/gemm_epilogue.h"
#include "cutlass/gemm/gemm_epilogue_traits.h"
#include "cutlass/gemm/gemm_global_tile.h"
#include "cutlass/gemm/gemm_shared_tile.h"
#include "cutlass/gemm/gemm_traits.h"
#include "cutlass/gemm/hgemm_global_tile.h"
#include "cutlass/gemm/hgemm_multiply_add.h"
#include "cutlass/gemm/hgemm_swizzle.h"

Go to the source code of this file.

Classes

struct  cutlass::gemm::HgemmConfig< OutputTile_, ThreadGemmShape_, kScalarsPerLdgA_, kScalarsPerLdgB_ >
 
struct  cutlass::gemm::HgemmTransformerA< kLayout_, Iterator_ >
 
struct  cutlass::gemm::HgemmTransformerA< MatrixLayout::kColumnMajor, Iterator_ >
 
struct  cutlass::gemm::HgemmTransformerA< MatrixLayout::kRowMajor, Iterator_ >
 
struct  cutlass::gemm::HgemmTransformerB< kLayout_, Iterator_ >
 
struct  cutlass::gemm::HgemmTransformerB< MatrixLayout::kRowMajor, Iterator_ >
 
struct  cutlass::gemm::HgemmTransformerB< MatrixLayout::kColumnMajor, Iterator_ >
 
struct  cutlass::gemm::HgemmTileTraitsHelperA< kLayout_, GemmConfig_ >
 
struct  cutlass::gemm::HgemmTileTraitsHelperA< MatrixLayout::kRowMajor, GemmConfig_ >
 
struct  cutlass::gemm::HgemmTileTraitsHelperB< kLayout_, GemmConfig_ >
 
struct  cutlass::gemm::HgemmTileTraitsHelperB< MatrixLayout::kColumnMajor, GemmConfig_ >
 
struct  cutlass::gemm::HgemmTraitsHelper< kLayoutA_, kLayoutB_, OutputTile_, EpilogueFunctor_, ThreadGemmShape_, kScalarsPerLdgA_, kScalarsPerLdgB_, Index_ >
 
struct  cutlass::gemm::HgemmTraits< kLayoutA_, kLayoutB_, OutputTile_, EpilogueFunctor_, ThreadGemmShape_, kScalarsPerLdgA_, kScalarsPerLdgB_, Index_, Helper_ >
 

Namespaces

 cutlass
 
 cutlass::gemm