Back to Cutlass

CUTLASS: mma_tensor_op_sm70.h File Reference

docs/mma__tensor__op__sm70_8h.html

4.4.21.9 KB
Original Source

| | CUTLASS

CUDA Templates for Linear Algebra Subroutines and Solvers |

Classes | Namespaces

mma_tensor_op_sm70.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/mma.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_sm70.h"

Include dependency graph for mma_tensor_op_sm70.h:

This graph shows which files directly or indirectly include this file:

[Go to the source code of this file.](mma tensor op__sm70_8h_source.html)

|

Classes

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

|

Namespaces

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

Detailed Description

This is a work in progress.


Generated by 1.8.11