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::RowMajorBlockLinear< BlockRows, BlockColumns > Struct Template Reference

#include <matrix_traits.h>

Public Member Functions

CUTLASS_HOST_DEVICE Coord< kStorageRankoperator() (MatrixCoord const &coord) const
 Maps (row, col) to (row, col, row, col) More...
 

Static Public Member Functions

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

Detailed Description

template<int BlockRows, int BlockColumns>
struct cutlass::MatrixLayout::RowMajorBlockLinear< BlockRows, BlockColumns >

Mapping function for block-linear matrices. Matrix is structured as row-major arrangement of 2D tiles (that are row-major)

Member Function Documentation

◆ operator()()

template<int BlockRows, int BlockColumns>
CUTLASS_HOST_DEVICE Coord<kStorageRank> cutlass::MatrixLayout::RowMajorBlockLinear< BlockRows, BlockColumns >::operator() ( MatrixCoord const &  coord) const
inline

◆ stride()

template<int BlockRows, int BlockColumns>
static CUTLASS_HOST_DEVICE Coord<kStorageRank> cutlass::MatrixLayout::RowMajorBlockLinear< BlockRows, BlockColumns >::stride ( int  ldm)
inlinestatic

Member Data Documentation

◆ kBlockColumns

template<int BlockRows, int BlockColumns>
int const cutlass::MatrixLayout::RowMajorBlockLinear< BlockRows, BlockColumns >::kBlockColumns = BlockColumns
static

◆ kBlockRows

template<int BlockRows, int BlockColumns>
int const cutlass::MatrixLayout::RowMajorBlockLinear< BlockRows, BlockColumns >::kBlockRows = BlockRows
static

◆ kStorageRank

template<int BlockRows, int BlockColumns>
int const cutlass::MatrixLayout::RowMajorBlockLinear< BlockRows, BlockColumns >::kStorageRank = 4
static

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