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_, AccumulatorsPerThread_, 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_, AccumulatorsPerThread_, kScalarsPerLdgA_, kScalarsPerLdgB_, Index_ >
struct	cutlass::gemm::HgemmTraits< kLayoutA_, kLayoutB_, OutputTile_, EpilogueFunctor_, AccumulatorsPerThread_, kScalarsPerLdgA_, kScalarsPerLdgB_, Index_, Helper_ >

Namespaces
	cutlass
	cutlass::gemm