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

Templates exposing architecture support for multiply-add operations. More...

#include "cutlass/cutlass.h"
#include "cutlass/tensor_ref.h"
#include "cutlass/layout/matrix.h"
#include "cutlass/gemm/gemm.h"
#include "cutlass/gemm/thread/mma.h"
#include "cutlass/functional.h"
#include "cutlass/reduction/thread/reduce.h"
Include dependency graph for gemm/thread/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::gemm::thread::detail::Mma_HFMA2< Shape, LayoutA, LayoutB, LayoutC, bool >
 Structure to compute the matrix product for HFMA. More...
 
struct  cutlass::gemm::thread::detail::Mma_HFMA2< Shape, layout::ColumnMajor, layout::ColumnMajor, layout::ColumnMajor, true >
 
struct  cutlass::gemm::thread::detail::Mma_HFMA2< Shape, layout::ColumnMajor, layout::ColumnMajor, layout::RowMajor, true >
 
struct  cutlass::gemm::thread::detail::Mma_HFMA2< Shape, layout::ColumnMajor, layout::RowMajor, layout::ColumnMajor, true >
 
struct  cutlass::gemm::thread::detail::Mma_HFMA2< Shape, layout::ColumnMajor, layout::RowMajor, layout::RowMajor, true >
 
struct  cutlass::gemm::thread::detail::Mma_HFMA2< Shape, layout::RowMajor, layout::ColumnMajor, layout::ColumnMajor, true >
 
struct  cutlass::gemm::thread::detail::Mma_HFMA2< Shape, layout::RowMajor, layout::ColumnMajor, layout::RowMajor, true >
 
struct  cutlass::gemm::thread::detail::Mma_HFMA2< Shape, layout::RowMajor, layout::RowMajor, layout::ColumnMajor, true >
 
struct  cutlass::gemm::thread::detail::Mma_HFMA2< Shape, layout::RowMajor, layout::RowMajor, layout::RowMajor, true >
 
struct  cutlass::gemm::thread::detail::Mma_HFMA2< Shape, LayoutA, LayoutB, layout::RowMajor, false >
 
struct  cutlass::gemm::thread::detail::Mma_HFMA2< Shape, LayoutA, LayoutB, layout::ColumnMajor, false >
 
struct  cutlass::gemm::thread::Mma< Shape_, half_t, LayoutA, half_t, LayoutB, half_t, LayoutC, arch::OpMultiplyAdd >
 Structure to compute the matrix product. More...
 
struct  cutlass::gemm::thread::detail::EnableMma_Crow_SM60< LayoutA, LayoutB >
 Determines whether to enable thread::Gemm<> specializations compatible with SM50. More...
 
struct  cutlass::gemm::thread::Mma< Shape_, half_t, LayoutA_, half_t, LayoutB_, half_t, layout::RowMajor, arch::OpMultiplyAdd, typename platform::enable_if< detail::EnableMma_Crow_SM60< LayoutA_, LayoutB_ >::value >::type >
 Computes matrix product when C is row-major. More...
 

Namespaces

 cutlass
 
 cutlass::gemm
 
 cutlass::gemm::thread
 
 cutlass::gemm::thread::detail