Matrix multiply-add operation.
#include <mma_sm60.h>
template<typename LayoutA , typename LayoutB >
using cutlass::arch::Mma< gemm::GemmShape< 1, 2, 1 >, 1, half_t, LayoutA, half_t, LayoutB, half_t, layout::RowMajor, OpMultiplyAdd >::Shape = gemm::GemmShape<1, 2, 1> |
template<typename LayoutA , typename LayoutB >
CUTLASS_HOST_DEVICE void cutlass::arch::Mma< gemm::GemmShape< 1, 2, 1 >, 1, half_t, LayoutA, half_t, LayoutB, half_t, layout::RowMajor, OpMultiplyAdd >::operator() |
( |
Array< half_t, 2 > & |
d, |
|
|
Array< half_t, 1 > const & |
a, |
|
|
Array< half_t, 2 > const & |
b, |
|
|
Array< half_t, 2 > const & |
c |
|
) |
| |
|
inline |
The documentation for this struct was generated from the following file: