CUTLASS: gemm_complex.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_complex.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_complex.h"
#include "cutlass/gemm/device/default_gemm_configuration.h"

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

Go to the source code of this file.

|

Classes

| | class | cutlass::gemm::device::GemmComplex< ElementA_, LayoutA_, ElementB_, LayoutB_, ElementC_, LayoutC_, ElementAccumulator_, OperatorClass_, ArchTag_, ThreadblockShape_, WarpShape_, InstructionShape_, EpilogueOutputOp_, ThreadblockSwizzle_, Stages, TransformA, TransformB, SplitKSerial > | | | | struct | cutlass::gemm::device::GemmComplex< ElementA_, LayoutA_, ElementB_, LayoutB_, ElementC_, LayoutC_, ElementAccumulator_, OperatorClass_, ArchTag_, ThreadblockShape_, WarpShape_, InstructionShape_, EpilogueOutputOp_, ThreadblockSwizzle_, Stages, TransformA, TransformB, SplitKSerial >::Arguments | | | Argument structure. More...
| | | | class | [cutlass::gemm::device::GemmComplex< ElementA_, LayoutA_, ElementB_, LayoutB_, ElementC_, layout::ColumnMajor, ElementAccumulator_, OperatorClass_, ArchTag_, ThreadblockShape_, WarpShape_, InstructionShape_, EpilogueOutputOp_, ThreadblockSwizzle_, Stages, TransformA, TransformB, SplitKSerial >](classcutlass_1_1gemm_1_1device_1_1GemmComplex_3_01ElementA 00_01LayoutA 00_01ElementB___00_07c56401b4df75709ae636675d9980a9a.html) | | | Partial specialization for column-major output exchanges problem size and operand. [More...](classcutlass_1_1gemm_1_1device_1_1GemmComplex_3_01ElementA 00_01LayoutA 00_01ElementB___00_07c56401b4df75709ae636675d9980a9a.html#details)
| | | | struct | [cutlass::gemm::device::GemmComplex< ElementA_, LayoutA_, ElementB_, LayoutB_, ElementC_, layout::ColumnMajor, ElementAccumulator_, OperatorClass_, ArchTag_, ThreadblockShape_, WarpShape_, InstructionShape_, EpilogueOutputOp_, ThreadblockSwizzle_, Stages, TransformA, TransformB, SplitKSerial >::Arguments](structcutlass_1_1gemm_1_1device_1_1GemmComplex_3_01ElementA 00_01LayoutA 00_01ElementB___00_a3923967cafb5cb9774c320dc24baa77.html) | | | Argument structure. [More...](structcutlass_1_1gemm_1_1device_1_1GemmComplex_3_01ElementA 00_01LayoutA 00_01ElementB___00_a3923967cafb5cb9774c320dc24baa77.html#details)
| | |

|

Namespaces

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

---

Generated by 1.8.11