CUTLASS
CUDA Templates for Linear Algebra Subroutines and Solvers
|
Configuration for basic GEMM operations.
#include <library.h>
Public Attributes | |
gemm::GemmCoord | problem_size |
GEMM problem size. More... | |
int64_t | lda |
Leading dimension of A matrix. More... | |
int64_t | ldb |
Leading dimension of B matrix. More... | |
int64_t | ldc |
Leading dimension of C matrix. More... | |
int64_t | ldd |
Leading dimension of D matrix. More... | |
int | split_k_slices |
Number of partitions of K dimension. More... | |
int64_t cutlass::library::GemmConfiguration::lda |
int64_t cutlass::library::GemmConfiguration::ldb |
int64_t cutlass::library::GemmConfiguration::ldc |
int64_t cutlass::library::GemmConfiguration::ldd |
gemm::GemmCoord cutlass::library::GemmConfiguration::problem_size |
int cutlass::library::GemmConfiguration::split_k_slices |