Cutlass
CUDA Templates for Linear Algebra Subroutines and Solvers
Public Member Functions | Static Public Member Functions | Static Public Attributes | List of all members
cutlass::MatrixLayout::ContiguousLayout Struct Reference

#include <matrix_traits.h>

Public Member Functions

CUTLASS_HOST_DEVICE Coord< kStorageRankoperator() (MatrixCoord const &coord) const
 

Static Public Member Functions

static CUTLASS_HOST_DEVICE Coord< kStorageRankstride (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...
 

Detailed Description

Mapping function for scenario in which layout is row-major or column-major but this information is only available at runtime.

Member Function Documentation

◆ operator()()

CUTLASS_HOST_DEVICE Coord<kStorageRank> cutlass::MatrixLayout::ContiguousLayout::operator() ( MatrixCoord const &  coord) const
inline

Mapping function defined by runtime variable. Returns coordinates in n-D storage array as (matrix row, matrix colum, 0)

◆ stride()

static CUTLASS_HOST_DEVICE Coord<kStorageRank> cutlass::MatrixLayout::ContiguousLayout::stride ( MatrixLayout::Kind  layout,
int  ldm 
)
inlinestatic

Member Data Documentation

◆ kColumn

int const cutlass::MatrixLayout::ContiguousLayout::kColumn = 1
static

◆ kRow

int const cutlass::MatrixLayout::ContiguousLayout::kRow = 0
static

◆ kStorageRank

int const cutlass::MatrixLayout::ContiguousLayout::kStorageRank = 3
static

The documentation for this struct was generated from the following file: