![]() |
CUTLASS
CUDA Templates for Linear Algebra Subroutines and Solvers
|
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"

Go to the source code of this file.
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 | |
This is a work in progress.
1.8.11