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

• include

• cutlass

• gemm

• warp

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.](mma complex tensor__op_8h_source.html)

|

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 | | |

---

Generated by 1.8.11