device/gemm_batched.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_batched.h"`
`#include "cutlass/gemm/kernel/default_gemm.h"`
`#include "cutlass/gemm/device/default_gemm_configuration.h"`
Include dependency graph for device/gemm_batched.h:

Go to the source code of this file.

## Classes

class | cutlass::gemm::device::GemmBatched< ElementA_, LayoutA_, ElementB_, LayoutB_, ElementC_, LayoutC_, ElementAccumulator_, OperatorClass_, ArchTag_, ThreadblockShape_, WarpShape_, InstructionShape_, EpilogueOutputOp_, ThreadblockSwizzle_, Stages, AlignmentA, AlignmentB, Operator_ >
struct | cutlass::gemm::device::GemmBatched< ElementA_, LayoutA_, ElementB_, LayoutB_, ElementC_, LayoutC_, ElementAccumulator_, OperatorClass_, ArchTag_, ThreadblockShape_, WarpShape_, InstructionShape_, EpilogueOutputOp_, ThreadblockSwizzle_, Stages, AlignmentA, AlignmentB, Operator_ >::Arguments
| Argument structure. More...
class | cutlass::gemm::device::GemmBatched< ElementA_, LayoutA_, ElementB_, LayoutB_, ElementC_, layout::ColumnMajor, ElementAccumulator_, OperatorClass_, ArchTag_, ThreadblockShape_, WarpShape_, InstructionShape_, EpilogueOutputOp_, ThreadblockSwizzle_, Stages, AlignmentA, AlignmentB, Operator_ >
| Parital specialization for column-major output exchanges problem size and operand. More...
struct | cutlass::gemm::device::GemmBatched< ElementA_, LayoutA_, ElementB_, LayoutB_, ElementC_, layout::ColumnMajor, ElementAccumulator_, OperatorClass_, ArchTag_, ThreadblockShape_, WarpShape_, InstructionShape_, EpilogueOutputOp_, ThreadblockSwizzle_, Stages, AlignmentA, AlignmentB, Operator_ >::Arguments
| Argument structure. More...

## Namespaces

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