CUTLASS
CUDA Templates for Linear Algebra Subroutines and Solvers
Public Types | Public Member Functions | List of all members
cutlass::arch::Mma< gemm::GemmShape< 1, 1, 1 >, 1, float, LayoutA, complex< float >, LayoutB, complex< float >, LayoutC, OpMultiplyAdd > Struct Template Reference

Matrix multiply-add operation.

#include <mma_sm50.h>

Public Types

using Shape = gemm::GemmShape< 1, 1, 1 >
 

Public Member Functions

CUTLASS_HOST_DEVICE void operator() (Array< complex< float >, 1 > &d, Array< float, 1 > const &a, Array< complex< float >, 1 > const &b, Array< complex< float >, 1 > const &c)
 

Member Typedef Documentation

template<typename LayoutA , typename LayoutB , typename LayoutC >
using cutlass::arch::Mma< gemm::GemmShape< 1, 1, 1 >, 1, float, LayoutA, complex< float >, LayoutB, complex< float >, LayoutC, OpMultiplyAdd >::Shape = gemm::GemmShape<1, 1, 1>

Member Function Documentation

template<typename LayoutA , typename LayoutB , typename LayoutC >
CUTLASS_HOST_DEVICE void cutlass::arch::Mma< gemm::GemmShape< 1, 1, 1 >, 1, float, LayoutA, complex< float >, LayoutB, complex< float >, LayoutC, OpMultiplyAdd >::operator() ( Array< complex< float >, 1 > &  d,
Array< float, 1 > const &  a,
Array< complex< float >, 1 > const &  b,
Array< complex< float >, 1 > const &  c 
)
inline

The documentation for this struct was generated from the following file: