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

Templates implementing warp-level matrix multiply-accumulate operations targeting Tensor Cores. More...

#include "cutlass/cutlass.h"
#include "cutlass/array.h"
#include "cutlass/numeric_types.h"
#include "cutlass/matrix_shape.h"
#include "cutlass/arch/memory_sm75.h"
#include "cutlass/arch/mma_sm75.h"
#include "cutlass/gemm/gemm.h"
#include "cutlass/gemm/warp/mma.h"
#include "cutlass/gemm/warp/mma_tensor_op_policy.h"
#include "cutlass/gemm/warp/mma_tensor_op_tile_iterator.h"
Include dependency graph for mma_tensor_op.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

class  cutlass::gemm::warp::MmaTensorOp< Shape_, ElementA_, LayoutA_, ElementB_, LayoutB_, ElementC_, LayoutC_, Policy_, PartitionsK_, AccumulatorsInRowMajor, PartitionsN_, Enable >
 Structure to compute the matrix product targeting CUDA cores and SIMT math instructions. More...
 

Namespaces

 cutlass
 
 cutlass::gemm
 
 cutlass::gemm::warp