Back to Cutlass

CUTLASS: mma_sm60.h File Reference

docs/gemm_2thread_2mma__sm60_8h.html

4.4.25.8 KB
Original Source

| | 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 >](structcutlass_1_1gemm_1_1thread_1_1Mma_3_01Shape _00_01half t_00_01LayoutA_00_01half__t_00_01L066c9d2371712cdf0cac099ca9bcc578.html) | | | Structure to compute the matrix product. [More...](structcutlass_1_1gemm_1_1thread_1_1Mma_3_01Shape _00_01half t_00_01LayoutA_00_01half__t_00_01L066c9d2371712cdf0cac099ca9bcc578.html#details)
| | | | struct | [cutlass::gemm::thread::detail::EnableMma_Crow_SM60< LayoutA, LayoutB >](structcutlass_1_1gemm_1_1thread_1_1detail_1_1EnableMma Crow SM60.html) | | | Determines whether to enable thread::Gemm<> specializations compatible with SM50. [More...](structcutlass_1_1gemm_1_1thread_1_1detail_1_1EnableMma Crow SM60.html#details)
| | | | 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 >](structcutlass_1_1gemm_1_1thread_1_1Mma_3_01Shape _00_01half t_00_01LayoutA _00_01half t_00_088f0e99e501b6012297eb30b4e89bcea.html) | | | Computes matrix product when C is row-major. [More...](structcutlass_1_1gemm_1_1thread_1_1Mma_3_01Shape _00_01half t_00_01LayoutA _00_01half t_00_088f0e99e501b6012297eb30b4e89bcea.html#details)
| | |

|

Namespaces

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


Generated by 1.8.11