Back to Cutlass

CUTLASS: default_mma_core.h File Reference

docs/default__mma__core_8h.html

4.4.21.9 KB
Original Source

| | CUTLASS

CUDA Templates for Linear Algebra Subroutines and Solvers |

Classes | Namespaces

default_mma_core.h File Reference

Defines basic properties needed by CTA-level GEMMs assuming expectations about data layout of the global memory fragments, data types, and internal tile sizes. More...

#include "cutlass/cutlass.h"
#include "cutlass/array.h"
#include "cutlass/numeric_types.h"
#include "cutlass/matrix_shape.h"
#include "cutlass/gemm/warp/mma.h"
#include "cutlass/gemm/threadblock/mma_pipelined.h"
#include "cutlass/gemm/threadblock/mma_singlestage.h"

Include dependency graph for default_mma_core.h:

This graph shows which files directly or indirectly include this file:

[Go to the source code of this file.](default mma core_8h_source.html)

|

Classes

| | struct | cutlass::gemm::threadblock::DefaultMmaCore< Shape, WarpShape, InstructionShape, ElementA, LayoutA, ElementB, LayoutB, ElementC, LayoutC, OperatorClass, Stages, Operator, AccumulatorsInRowMajor > | | |

|

Namespaces

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

Detailed Description

Partial specializations for threadblock::Mma operations targeting TensorOp instructions.


Generated by 1.8.11