Back to Cutlass

CUTLASS: default_mma_core_sm75.h File Reference

docs/default__mma__core__sm75_8h.html

4.4.23.7 KB
Original Source

| | CUTLASS

CUDA Templates for Linear Algebra Subroutines and Solvers |

Classes | Namespaces

default_mma_core_sm75.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/platform/platform.h"
#include "cutlass/numeric_types.h"
#include "cutlass/matrix_shape.h"
#include "cutlass/layout/tensor_op_multiplicand_sm75.h"
#include "cutlass/transform/pitch_linear_thread_map.h"
#include "cutlass/transform/threadblock/regular_tile_iterator_tensor_op.h"
#include "cutlass/gemm/warp/default_mma_tensor_op.h"
#include "cutlass/gemm/threadblock/default_mma_core.h"

Include dependency graph for default_mma_core_sm75.h:

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

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

|

Classes

| | struct | [cutlass::gemm::threadblock::DefaultMmaCore< Shape_, WarpShape_, InstructionShape_, ElementA_, layout::ColumnMajor, ElementB_, layout::RowMajor, ElementC_, LayoutC_, arch::OpClassTensorOp, 2, Operator_ >](structcutlass_1_1gemm_1_1threadblock_1_1DefaultMmaCore_3_01Shape 00_01WarpShape 00_01Instrucf60fe02fcdd80d28b7fd419133465dcc.html) | | | | struct | [cutlass::gemm::threadblock::DefaultMmaCore< Shape_, WarpShape_, InstructionShape_, ElementA_, layout::RowMajor, ElementB_, layout::ColumnMajor, ElementC_, LayoutC_, arch::OpClassTensorOp, 2, Operator_ >](structcutlass_1_1gemm_1_1threadblock_1_1DefaultMmaCore_3_01Shape 00_01WarpShape 00_01Instruc24092ddc01fc83dabb7db4c14880fe60.html) | | | | struct | [cutlass::gemm::threadblock::DefaultMmaCore< Shape_, WarpShape_, InstructionShape_, ElementA_, layout::RowMajor, ElementB_, layout::RowMajor, ElementC_, LayoutC_, arch::OpClassTensorOp, 2, Operator_ >](structcutlass_1_1gemm_1_1threadblock_1_1DefaultMmaCore_3_01Shape 00_01WarpShape 00_01Instruc4fee9f2965b8468bfb42b94a74527d22.html) | | | | struct | [cutlass::gemm::threadblock::DefaultMmaCore< Shape_, WarpShape_, InstructionShape_, ElementA_, layout::ColumnMajor, ElementB_, layout::ColumnMajor, ElementC_, LayoutC_, arch::OpClassTensorOp, 2, Operator_ >](structcutlass_1_1gemm_1_1threadblock_1_1DefaultMmaCore_3_01Shape 00_01WarpShape 00_01Instruc803d38bc1e4618c07c47f54c87ae2678.html) | | | | struct | [cutlass::gemm::threadblock::DefaultMmaCore< Shape_, WarpShape_, InstructionShape_, ElementA_, layout::ColumnMajorInterleaved< InterleavedK >, ElementB_, layout::RowMajorInterleaved< InterleavedK >, ElementC_, LayoutC_, arch::OpClassTensorOp, 2, Operator_, AccumulatorsInRowMajor >](structcutlass_1_1gemm_1_1threadblock_1_1DefaultMmaCore_3_01Shape 00_01WarpShape 00_01Instruc2bf00737f4ad0a9da9a8be6d3e66c152.html) | | |

|

Namespaces

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

Detailed Description

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


Generated by 1.8.11