CUTLASS: gemm.h File Reference

| | CUTLASS

CUDA Templates for Linear Algebra Subroutines and Solvers |

• Main Page

• Modules

• Namespaces

• Classes

• Files

• File List

• File Members

• include

• cutlass

• gemm

• device

Classes | Namespaces

include/cutlass/gemm/device/gemm.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/device_kernel.h"
#include "cutlass/gemm/threadblock/threadblock_swizzle.h"
#include "cutlass/gemm/kernel/gemm.h"
#include "cutlass/gemm/kernel/default_gemm.h"
#include "cutlass/gemm/device/default_gemm_configuration.h"

Include dependency graph for include/cutlass/gemm/device/gemm.h:

Go to the source code of this file.

|

Classes

| | class | cutlass::gemm::device::Gemm< ElementA_, LayoutA_, ElementB_, LayoutB_, ElementC_, LayoutC_, ElementAccumulator_, OperatorClass_, ArchTag_, ThreadblockShape_, WarpShape_, InstructionShape_, EpilogueOutputOp_, ThreadblockSwizzle_, Stages, AlignmentA, AlignmentB, SplitKSerial, Operator_, IsBetaZero > | | | | struct | cutlass::gemm::device::Gemm< ElementA_, LayoutA_, ElementB_, LayoutB_, ElementC_, LayoutC_, ElementAccumulator_, OperatorClass_, ArchTag_, ThreadblockShape_, WarpShape_, InstructionShape_, EpilogueOutputOp_, ThreadblockSwizzle_, Stages, AlignmentA, AlignmentB, SplitKSerial, Operator_, IsBetaZero >::Arguments | | | Argument structure. More...
| | | | class | [cutlass::gemm::device::Gemm< ElementA_, LayoutA_, ElementB_, LayoutB_, ElementC_, layout::ColumnMajor, ElementAccumulator_, OperatorClass_, ArchTag_, ThreadblockShape_, WarpShape_, InstructionShape_, EpilogueOutputOp_, ThreadblockSwizzle_, Stages, AlignmentA, AlignmentB, SplitKSerial, Operator_, IsBetaZero >](classcutlass_1_1gemm_1_1device_1_1Gemm_3_01ElementA 00_01LayoutA 00_01ElementB___00_01Layout4d0960ae6b1d1bf19e6239dbd002249c.html) | | | Partial specialization for column-major output exchanges problem size and operand. [More...](classcutlass_1_1gemm_1_1device_1_1Gemm_3_01ElementA 00_01LayoutA 00_01ElementB___00_01Layout4d0960ae6b1d1bf19e6239dbd002249c.html#details)
| | | | struct | [cutlass::gemm::device::Gemm< ElementA_, LayoutA_, ElementB_, LayoutB_, ElementC_, layout::ColumnMajor, ElementAccumulator_, OperatorClass_, ArchTag_, ThreadblockShape_, WarpShape_, InstructionShape_, EpilogueOutputOp_, ThreadblockSwizzle_, Stages, AlignmentA, AlignmentB, SplitKSerial, Operator_, IsBetaZero >::Arguments](structcutlass_1_1gemm_1_1device_1_1Gemm_3_01ElementA 00_01LayoutA 00_01ElementB___00_01Layou1b211cc9c97c022d8fe10f2dd32c8709.html) | | | Argument structure. [More...](structcutlass_1_1gemm_1_1device_1_1Gemm_3_01ElementA 00_01LayoutA 00_01ElementB___00_01Layou1b211cc9c97c022d8fe10f2dd32c8709.html#details)
| | |

|

Namespaces

| | | cutlass | | | | | cutlass::gemm | | | | | cutlass::gemm::device | | |

---

Generated by 1.8.11