41 template <GemmOperand::Kind kOperand_, MatrixLayout::Kind kLayout_>
49 template <
typename GemmOperand::Kind kOperand_,
typename Tile_>
52 template <
typename Tile_>
54 static const int kExtent = Tile_::kW;
57 template <
typename Tile_>
59 static const int kExtent = Tile_::kH;
66 template <
typename ThreadBlockTile_, GemmOperand::Kind Usage, MatrixLayout::Kind Layout>
70 "MultiplicandTileShape defined only for A or B operands.");
95 template <GemmOperand::Kind operand,
bool Kstr
ided = true>
99 template <
bool Kstr
ided>
112 template <
bool Kstr
ided>
static bool const kKstrided
Definition: gemm_operand.h:82
static CUTLASS_HOST_DEVICE Coord< 3 > project(Coord< 3 > const &coord)
Definition: gemm_operand.h:115
CUTLASS_HOST_DEVICE Coord< 1 > make_Coord(int _0)
Helper to make a 2-element coordinate.
Definition: coord.h:368
Kind
Enumeration defining fundamental contiguous layouts.
Definition: matrix_traits.h:159
static CUTLASS_HOST_DEVICE Coord< 3 > project(Coord< 3 > const &coord)
Definition: gemm_operand.h:128
ThreadBlockTile_ ThreadBlockTile
Shape of GEMM thread block tile (K, N, M)
Definition: gemm_operand.h:70
platform::conditional< kKstrided, Shape< 1, ThreadBlockTile::kD, GetExtent< Usage, ThreadBlockTile >::kExtent >, Shape< 1, GetExtent< Usage, ThreadBlockTile >::kExtent, ThreadBlockTile::kD > >::type Shape
Map the ThreadBlockShape onto (kH, kW) dimensions for A and B operand.
Definition: gemm_operand.h:88
Defines a type for restructuring a tile.
Definition: gemm_operand.h:67
static const bool Congruous
Definition: gemm_operand.h:43
Definition: matrix_traits.h:357
#define CUTLASS_HOST_DEVICE
Definition: cutlass.h:46
static MatrixLayout::Kind const kLayout
Layout of tile.
Definition: gemm_operand.h:79
static CUTLASS_HOST_DEVICE Coord< 3 > project(Coord< 3 > const &coord)
Definition: gemm_operand.h:102
A Shape implementing Layout Concept describing the dimensions of a cube.
Definition: shape.h:64
Definition: gemm_operand.h:96
Definition: gemm_operand.h:50
Definition: matrix_traits.h:159
Gemm operand - D = A * B + C.
Definition: matrix_traits.h:356
static CUTLASS_HOST_DEVICE Coord< 3 > project(Coord< 3 > const &coord)
Definition: gemm_operand.h:135
Kind
Definition: matrix_traits.h:357
Definition: matrix_traits.h:357
static GemmOperand::Kind const kUsage
Identifies multiplicand.
Definition: gemm_operand.h:76
Defines properties of matrices used to denote layout and operands to GEMM kernels.
Helper to describe attributes of GEMM matrix operands.
Definition: gemm_operand.h:42