![]() |
CUTLASS
CUDA Templates for Linear Algebra Subroutines and Solvers
|
Default kernel-level GEMM definitions combine threadblock-scoped matrix multiply-add with the appropriate threadblock-scoped epilogue. More...
#include "cutlass/cutlass.h"#include "cutlass/gemm/kernel/default_gemm.h"#include "cutlass/gemm/kernel/gemm_splitk_parallel.h"

Go to the source code of this file.
Namespaces | |
| cutlass | |
| cutlass::gemm | |
| cutlass::gemm::kernel | |
Note, CUTLASS epilogues universally target row-major outputs. Column-major outputs are accommodated by exchanging A and B operands and assuming transposed layouts. Partial specializations here choose 'device::GemmTransposed' to implement this functionality.
1.8.11