Cutlass
CUDA Templates for Linear Algebra Subroutines and Solvers
|
#include <matrix_traits.h>
Public Member Functions | |
CUTLASS_HOST_DEVICE Coord< kStorageRank > | operator() (MatrixCoord const &coord) const |
Maps (row, col) to (row, col, row, col) More... | |
Static Public Member Functions | |
static CUTLASS_HOST_DEVICE Coord< kStorageRank > | stride (int ldm) |
Helper to compute stride vector from leading dimension. More... | |
Static Public Attributes | |
static int const | kStorageRank = 4 |
Rank of storage n-D array. More... | |
static int const | kBlockRows = BlockRows |
Interleaving size in rows dimension. More... | |
static int const | kBlockColumns = BlockColumns |
Interleaving size in columns dimension. More... | |
Mapping function for block-linear matrices. Matrix is structured as row-major arrangement of 2D tiles (that are row-major)
|
inline |
|
inlinestatic |
|
static |
|
static |
|
static |