CUTLASS
CUDA Templates for Linear Algebra Subroutines and Solvers
Classes | Namespaces
arch/mma_sm60.h File Reference

Matrix multiply. More...

#include <cuda_fp16.h>
#include "cutlass/arch/mma.h"
#include "cutlass/layout/matrix.h"
Include dependency graph for arch/mma_sm60.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

struct  cutlass::arch::Mma< gemm::GemmShape< 2, 1, 1 >, 1, half_t, LayoutA, half_t, LayoutB, half_t, LayoutC, OpMultiplyAdd >
 Matrix multiply-add operation. More...
 
struct  cutlass::arch::Mma< gemm::GemmShape< 1, 2, 1 >, 1, half_t, LayoutA, half_t, LayoutB, half_t, layout::RowMajor, OpMultiplyAdd >
 Matrix multiply-add operation. More...
 
struct  cutlass::arch::Mma< gemm::GemmShape< 2, 2, 1 >, 1, half_t, layout::ColumnMajor, half_t, layout::RowMajor, half_t, layout::ColumnMajor, OpMultiplyAdd >
 Matrix multiply-add operation. More...
 
struct  cutlass::arch::Mma< gemm::GemmShape< 2, 2, 1 >, 1, half_t, layout::ColumnMajor, half_t, layout::RowMajor, half_t, layout::RowMajor, OpMultiplyAdd >
 Matrix multiply-add operation. More...
 

Namespaces

 cutlass
 
 cutlass::arch