CUTLASS: gemm_splitk_parallel.h File Reference

| CUTLASS CUDA Templates for Linear Algebra Subroutines and Solvers

- Main Page
- Modules
- Namespaces
- Classes
- Files

- File List
- File Members

Classes | Namespaces
device/gemm_splitk_parallel.h File Reference

Template for GEMM performing a reduction over K partitions in parallel. 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_splitk_parallel.h"`
`#include "cutlass/gemm/device/default_gemm_configuration.h"`
`#include "cutlass/epilogue/thread/conversion_op.h"`
`#include "cutlass/reduction/kernel/reduce_split_k.h"`
`#include "cutlass/reduction/thread/reduction_operators.h"`
Include dependency graph for device/gemm_splitk_parallel.h:

Go to the source code of this file.

## Classes
class | cutlass::gemm::device::GemmSplitKParallel< ElementA_, LayoutA_, ElementB_, LayoutB_, ElementC_, LayoutC_, ElementAccumulator_, OperatorClass_, ArchTag_, ThreadblockShape_, WarpShape_, InstructionShape_, EpilogueOutputOp_, ConvertScaledOp_, ReductionOp_, ThreadblockSwizzle_, Stages, kAlignmentA, kAlignmentB, Operator_ >
struct | cutlass::gemm::device::GemmSplitKParallel< ElementA_, LayoutA_, ElementB_, LayoutB_, ElementC_, LayoutC_, ElementAccumulator_, OperatorClass_, ArchTag_, ThreadblockShape_, WarpShape_, InstructionShape_, EpilogueOutputOp_, ConvertScaledOp_, ReductionOp_, ThreadblockSwizzle_, Stages, kAlignmentA, kAlignmentB, Operator_ >::Arguments
| Argument structure. More...
class | cutlass::gemm::device::GemmSplitKParallel< ElementA_, LayoutA_, ElementB_, LayoutB_, ElementC_, layout::ColumnMajor, ElementAccumulator_, OperatorClass_, ArchTag_, ThreadblockShape_, WarpShape_, InstructionShape_, EpilogueOutputOp_, ConvertScaledOp_, ReductionOp_, ThreadblockSwizzle_, Stages, kAlignmentA, kAlignmentB, Operator_ >
| Partial specialization for column-major output. More...
struct | cutlass::gemm::device::GemmSplitKParallel< ElementA_, LayoutA_, ElementB_, LayoutB_, ElementC_, layout::ColumnMajor, ElementAccumulator_, OperatorClass_, ArchTag_, ThreadblockShape_, WarpShape_, InstructionShape_, EpilogueOutputOp_, ConvertScaledOp_, ReductionOp_, ThreadblockSwizzle_, Stages, kAlignmentA, kAlignmentB, Operator_ >::Arguments
| Argument structure. More...

## Namespaces
| cutlass
| cutlass::gemm
| cutlass::gemm::device