CUTLASS: mma_complex_tensor_op.h File Reference

| CUTLASS CUDA Templates for Linear Algebra Subroutines and Solvers

- Main Page
- Modules
- Namespaces
- Classes
- Files

- File List
- File Members

Classes | Namespaces
mma_complex_tensor_op.h File Reference

Templates implementing warp-level matrix multiply-accumulate operations targeting Tensor Cores. More...

`#include "cutlass/cutlass.h"`
`#include "cutlass/array.h"`
`#include "cutlass/complex.h"`
`#include "cutlass/numeric_types.h"`
`#include "cutlass/matrix_shape.h"`
`#include "cutlass/arch/memory_sm75.h"`
`#include "cutlass/arch/mma_sm75.h"`
`#include "cutlass/gemm/gemm.h"`
`#include "cutlass/gemm/warp/mma.h"`
`#include "cutlass/gemm/warp/mma_tensor_op_policy.h"`
`#include "cutlass/gemm/warp/mma_tensor_op.h"`
`#include "cutlass/gemm/warp/mma_tensor_op_tile_iterator.h"`
Include dependency graph for mma_complex_tensor_op.h:

Go to the source code of this file.

## Classes
class | cutlass::gemm::warp::MmaComplexTensorOp< Shape_, RealElementA, LayoutA_, RealElementB, LayoutB_, RealElementC, LayoutC_, Policy_, TransformA, TransformB, Enable >
class | cutlass::gemm::warp::MmaComplexTensorOp< Shape_, complex< RealElementA >, LayoutA_, complex< RealElementB >, LayoutB_, complex< RealElementC >, LayoutC_, Policy_, TransformA, TransformB, Enable >
| Partial specialization for complex*complex+complex => complex using real-valued TensorOps. More...

## Namespaces
| cutlass
| cutlass::gemm
| cutlass::gemm::warp