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 |
Static Public Member Functions | |
static CUTLASS_HOST_DEVICE Coord< kStorageRank > | stride (MatrixLayout::Kind layout, int ldm) |
Helper to construct a stride vector based on contiguous matrix layout and leading dimension. More... | |
Static Public Attributes | |
static int const | kStorageRank = 3 |
Arbitrary storage rank. More... | |
static int const | kRow = 0 |
Dimension of rows. More... | |
static int const | kColumn = 1 |
Dimension of columns. More... | |
Mapping function for scenario in which layout is row-major or column-major but this information is only available at runtime.
|
inline |
Mapping function defined by runtime variable. Returns coordinates in n-D storage array as (matrix row, matrix colum, 0)
|
inlinestatic |
|
static |
|
static |
|
static |