default_mma.h File Reference

Template for a pipelined GEMM kernel. Does not compute batching or support split-K. More...

`#include "cutlass/cutlass.h"`
`#include "cutlass/numeric_types.h"`
`#include "cutlass/arch/arch.h"`
`#include "cutlass/arch/wmma.h"`
`#include "cutlass/transform/threadblock/predicated_tile_iterator.h"`
`#include "cutlass/transform/threadblock/predicated_tile_iterator_2dthreadtile.h"`
`#include "cutlass/gemm/threadblock/default_mma_core_sm70.h"`
`#include "cutlass/gemm/threadblock/default_mma_core_sm75.h"`
Include dependency graph for default_mma.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

## Classes

struct | cutlass::gemm::threadblock::DefaultMma< ElementA_, LayoutA_, kAlignmentA, ElementB_, LayoutB_, kAlignmentB, ElementAccumulator_, LayoutC_, OperatorClass_, ArchTag_, ThreadblockShape_, WarpShape_, InstructionShape_, Stages, Operator, AccumulatorsInRowMajor >
struct | cutlass::gemm::threadblock::DefaultMma< ElementA, LayoutA, kAlignmentA, ElementB, LayoutB, kAlignmentB, ElementAccumulator, layout::RowMajor, arch::OpClassSimt, ArchTag, ThreadblockShape, WarpShape, InstructionShape, 2, Operator, false >
| Specialization for row-major output (OperatorClass Simt) More...
struct | cutlass::gemm::threadblock::DefaultMma< ElementA, LayoutA, kAlignmentA, ElementB, LayoutB, kAlignmentB, ElementAccumulator, layout::RowMajor, arch::OpClassTensorOp, ArchTag, ThreadblockShape, WarpShape, InstructionShape, 2, Operator, false >
| Specialization for row-major output (OperatorClass Simt) More...
struct | cutlass::gemm::threadblock::DefaultMma< ElementA, LayoutA, kAlignmentA, ElementB, LayoutB, kAlignmentB, ElementAccumulator, layout::ColumnMajorInterleaved< InterleavedK >, OperatorClass, ArchTag, ThreadblockShape, WarpShape, InstructionShape, 2, Operator, true >
| Specialization for column-major-interleaved output. More...
struct | cutlass::gemm::threadblock::DefaultMma< int8_t, LayoutA, kAlignmentA, int8_t, LayoutB, kAlignmentB, ElementAccumulator, layout::RowMajor, arch::OpClassSimt, ArchTag, ThreadblockShape, WarpShape, GemmShape< 1, 1, 4 >, 2, Operator, false >

## Namespaces

| cutlass
| cutlass::gemm
| cutlass::gemm::threadblock