52 static int const kK = 0;
55 static int const kN = 1;
58 static int const kM = 2;
89 Index const &
m()
const {
return this->
at(kM); }
97 Index const &
n()
const {
return this->
at(kN); }
CUTLASS_HOST_DEVICE GemmCoord & operator/=(Base const &b)
In-place division.
Definition: gemm_coord.h:200
int Index
Integer-valued index.
Definition: gemm_coord.h:46
CUTLASS_HOST_DEVICE Coord< 3 > knm() const
Obtains a Coord<3> from GemmCoord.
Definition: gemm_coord.h:121
CUTLASS_HOST_DEVICE Coord< 2 > km() const
Obtains a Coord<2> from GemmCoord.
Definition: gemm_coord.h:139
Coord< 4, Index > Base
Base type is a Coord of rank=4.
Definition: gemm_coord.h:49
A Coord is a coordinate of arbitrary rank into a tensor or matrix.
CUTLASS_HOST_DEVICE Index const & n() const
Returns the GEMM N coordinate.
Definition: gemm_coord.h:97
CUTLASS_HOST_DEVICE GemmCoord operator/(Base const &b) const
Element-wise division.
Definition: gemm_coord.h:173
CUTLASS_HOST_DEVICE Coord< 1 > make_Coord(int _0)
Helper to make a 2-element coordinate.
Definition: coord.h:368
CUTLASS_HOST_DEVICE Index & m()
Returns reference to the GEMM M coordinate.
Definition: gemm_coord.h:93
Definition: gemm_coord.h:43
static int const kBatch
Batch dimension - for generalizing to larger problems.
Definition: gemm_coord.h:61
CUTLASS_HOST_DEVICE GemmCoord operator-(Base const &b) const
Element-wise subtraction.
Definition: gemm_coord.h:161
CUTLASS_HOST_DEVICE Coord & operator*=(Coord const &b)
In-place multiplication.
Definition: coord.h:197
CUTLASS_HOST_DEVICE GemmCoord(Coord< 4, Index > const &coord)
Constructs from Coord<4>
Definition: gemm_coord.h:77
CUTLASS_HOST_DEVICE Index & k()
Returns reference to the GEMM K coordinate.
Definition: gemm_coord.h:109
CUTLASS_HOST_DEVICE GemmCoord(Coord< 3, Index > const &coord, Index _batch=0)
Constructs from Coord<3> and a batch.
Definition: gemm_coord.h:73
CUTLASS_HOST_DEVICE GemmCoord operator+(Base const &b) const
Element-wise addition.
Definition: gemm_coord.h:155
CUTLASS_HOST_DEVICE Coord & operator-=(Coord const &b)
In-place subtraction.
Definition: coord.h:188
static int const kK
GEMM K dimension - inner dimension of the GEMM problem.
Definition: gemm_coord.h:52
static int const kN
GEMM N dimension - columns of the output C matrix.
Definition: gemm_coord.h:55
CUTLASS_HOST_DEVICE Coord & operator+=(Coord const &b)
In-place addition.
Definition: coord.h:179
#define CUTLASS_HOST_DEVICE
Definition: cutlass.h:46
CUTLASS_HOST_DEVICE Coord< 2 > kn() const
Obtains a Coord<2> from GemmCoord.
Definition: gemm_coord.h:145
CUTLASS_HOST_DEVICE Index & at()
Gets the index of a given Coord element.
Definition: coord.h:240
CUTLASS_HOST_DEVICE Coord< 2 > mn() const
Obtains a Coord<2> from GemmCoord.
Definition: gemm_coord.h:133
CUTLASS_HOST_DEVICE Coord & operator/=(Coord const &b)
In-place division.
Definition: coord.h:206
CUTLASS_HOST_DEVICE GemmCoord & operator+=(Base const &b)
In-place addition.
Definition: gemm_coord.h:179
static int const kM
GEMM M dimension - rows of the output C matrix.
Definition: gemm_coord.h:58
Statically-sized array specifying Coords within a tensor.
Definition: coord.h:49
CUTLASS_HOST_DEVICE GemmCoord()
Default ctor.
Definition: gemm_coord.h:69
CUTLASS_HOST_DEVICE GemmCoord & operator-=(Base const &b)
In-place subtraction.
Definition: gemm_coord.h:186
CUTLASS_HOST_DEVICE Index const & m() const
Returns the GEMM M coordinate.
Definition: gemm_coord.h:89
CUTLASS_HOST_DEVICE GemmCoord operator*(Base const &b) const
Element-wise multiplication.
Definition: gemm_coord.h:167
CUTLASS_HOST_DEVICE Index const & k() const
Returns the GEMM K coordinate.
Definition: gemm_coord.h:105
CUTLASS_HOST_DEVICE GemmCoord & operator*=(Base const &b)
In-place multiplication.
Definition: gemm_coord.h:193
CUTLASS_HOST_DEVICE Index const & batch() const
Returns the GEMM batch coordinate.
Definition: gemm_coord.h:113
Basic include for CUTLASS macros.
CUTLASS_HOST_DEVICE Index & n()
Returns reference to the GEMM N coordinate.
Definition: gemm_coord.h:101
CUTLASS_HOST_DEVICE Coord< 2 > nm() const
Obtains a Coord<2> from GemmCoord.
Definition: gemm_coord.h:127
CUTLASS_HOST_DEVICE GemmCoord(Index k, Index n, Index m, Index batch=0)
Helper to construct from a K, N, M, batch variables.
Definition: gemm_coord.h:85
CUTLASS_HOST_DEVICE Index & batch()
Returns reference to the GEMM batch coordinate.
Definition: gemm_coord.h:117
CUTLASS_HOST_DEVICE GemmCoord(Index coord[4])
Constructs from an array of coordinate elements.
Definition: gemm_coord.h:81