igemm_traits.h File Reference

Defies structural properties of mixed-precision integer GEMM. Multiplicands are assumed to be packed 8bit integers, accumulators are assumed to be 32b signed integers, and output formats vary. More...

#include <cutlass/convert.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/igemm_epilogue.h>
#include <cutlass/gemm/igemm_global_tile.h>
#include <cutlass/gemm/igemm_multiply_add.h>
#include <cutlass/gemm/igemm_swizzle.h>
#include <cutlass/reshape_tile.h>

Go to the source code of this file.

Classes
struct	cutlass::gemm::IgemmConfig< OutputTile_, ScalarD_, AccumulatorsPerThread_ >
struct	cutlass::gemm::IgemmConfig< OutputTile_, int8_t, AccumulatorsPerThread_ >
struct	cutlass::gemm::IgemmTileTraitsHelperA< kLayout_, GemmConfig_ >
struct	cutlass::gemm::IgemmTileTraitsHelperA< MatrixLayout::kColumnMajor, GemmConfig_ >
struct	cutlass::gemm::IgemmTileTraitsHelperB< kLayout_, GemmConfig_ >
struct	cutlass::gemm::IgemmTileTraitsHelperB< MatrixLayout::kRowMajor, GemmConfig_ >
struct	cutlass::gemm::IgemmTransformerA< kLayout_, Iterator_ >
struct	cutlass::gemm::IgemmTransformerA< MatrixLayout::kRowMajor, Iterator_ >
struct	cutlass::gemm::IgemmTransformerA< MatrixLayout::kColumnMajor, Iterator_ >
struct	cutlass::gemm::IgemmTransformerB< kLayout_, Iterator_ >
struct	cutlass::gemm::IgemmTransformerB< MatrixLayout::kColumnMajor, Iterator_ >
struct	cutlass::gemm::IgemmTransformerB< MatrixLayout::kRowMajor, Iterator_ >
struct	cutlass::gemm::IgemmTraitsHelper< kLayoutA_, kLayoutB_, OutputTile_, ScalarD_, EpilogueFunctor_, AccumulatorsPerThread_, Index_ >
struct	cutlass::gemm::IgemmEpilogueScalar< ScalarD_ >
struct	cutlass::gemm::IgemmEpilogueScalar< int >
struct	cutlass::gemm::IgemmTraits< kLayoutA_, kLayoutB_, OutputTile_, ScalarD_, EpilogueFunctor_, AccumulatorsPerThread_, Index_, Helper_ >

Namespaces
	cutlass
	cutlass::gemm