58   static int const kM = M;
    59   static int const kN = N;
    60   static int const kK = K;
    62   static int const kMN = M * N;
    63   static int const kMK = M * K;
    64   static int const kKN = N * K;
    65   static int const kMNK = M * N * K;
    67   static int const kCount = kMNK;
   103   static int const kM = 0;
   106   static int const kN = 1;
   109   static int const kK = 2;
   129   Index 
const & 
m()
 const { 
return this->at(kM); }
   133   Index & 
m() { 
return this->at(kM); }
   137   Index 
const & 
n()
 const { 
return this->at(kN); }
   141   Index & 
n() { 
return this->at(kN); }
   145   Index 
const & 
k()
 const { 
return this->at(kK); }
   149   Index & 
k() { 
return this->at(kK); }
   269   static int const kM = 0;
   272   static int const kN = 1;
   275   static int const kK = 2;
   278   static int const kBatch = 3;
   298   Index 
const & 
m()
 const { 
return this->at(kM); }
   302   Index & 
m() { 
return this->at(kM); }
   306   Index 
const & 
n()
 const { 
return this->at(kN); }
   310   Index & 
n() { 
return this->at(kN); }
   314   Index 
const & 
k()
 const { 
return this->at(kK); }
   318   Index & 
k() { 
return this->at(kK); }
   322   Index 
const & 
batch()
 const { 
return this->at(kBatch); }
   326   Index & 
batch() { 
return this->at(kBatch); }
 CUTLASS_HOST_DEVICE Coord< 4 > mnkb() const 
Obtains a Coord<4> from BatchedGemmCoord. 
Definition: include/cutlass/gemm/gemm.h:336
Coord< 4, Index > Base
Base type is a Coord of rank=4. 
Definition: include/cutlass/gemm/gemm.h:266
CUTLASS_HOST_DEVICE Index & m()
Returns reference to the GEMM M coordinate. 
Definition: include/cutlass/gemm/gemm.h:302
Definition: aligned_buffer.h:35
CUTLASS_HOST_DEVICE BatchedGemmCoord operator/(Base const &b) const 
Element-wise division. 
Definition: include/cutlass/gemm/gemm.h:364
CUTLASS_HOST_DEVICE GemmCoord & operator/=(Base const &b)
In-place division. 
Definition: include/cutlass/gemm/gemm.h:250
int Index
Integer-valued index. 
Definition: include/cutlass/gemm/gemm.h:97
CUTLASS_HOST_DEVICE GemmCoord(Coord< 3, Index > const &coord)
Constructs from Coord<3> and a batch. 
Definition: include/cutlass/gemm/gemm.h:121
CUTLASS_HOST_DEVICE GemmCoord mnk() const 
Obtains a GemmCoord from BatchedGemmCoord. 
Definition: include/cutlass/gemm/gemm.h:330
CUTLASS_HOST_DEVICE GemmCoord operator+(Base const &b) const 
Element-wise addition. 
Definition: include/cutlass/gemm/gemm.h:205
A Coord is a coordinate of arbitrary rank into a tensor or matrix. 
CUTLASS_HOST_DEVICE Coord< 1 > make_Coord(int _0)
Helper to make a 2-element coordinate. 
Definition: coord.h:387
CUTLASS_HOST_DEVICE Index & m()
Returns reference to the GEMM M coordinate. 
Definition: include/cutlass/gemm/gemm.h:133
Operand
GEMM operand enumeration: D = A * B + C. 
Definition: include/cutlass/gemm/gemm.h:39
Definition: include/cutlass/gemm/gemm.h:94
CUTLASS_HOST_DEVICE Coord< 2 > mn() const 
Obtains a Coord<2> from GemmCoord. 
Definition: include/cutlass/gemm/gemm.h:171
CUTLASS_HOST_DEVICE half_t & operator/=(half_t &lhs, half_t const &rhs)
Definition: half.h:684
CUTLASS_HOST_DEVICE BatchedGemmCoord operator-(Base const &b) const 
Element-wise subtraction. 
Definition: include/cutlass/gemm/gemm.h:352
CUTLASS_HOST_DEVICE Index & batch()
Returns reference to the GEMM batch coordinate. 
Definition: include/cutlass/gemm/gemm.h:326
CUTLASS_HOST_DEVICE Index const & n() const 
Returns the GEMM N coordinate. 
Definition: include/cutlass/gemm/gemm.h:137
CUTLASS_HOST_DEVICE Coord< 2 > nm() const 
Obtains a Coord<2> from GemmCoord. 
Definition: include/cutlass/gemm/gemm.h:165
CUTLASS_HOST_DEVICE BatchedGemmCoord operator+(Base const &b) const 
Element-wise addition. 
Definition: include/cutlass/gemm/gemm.h:346
CUTLASS_HOST_DEVICE Index & k()
Returns reference to the GEMM K coordinate. 
Definition: include/cutlass/gemm/gemm.h:149
CUTLASS_HOST_DEVICE Index & n()
Returns reference to the GEMM N coordinate. 
Definition: include/cutlass/gemm/gemm.h:310
CUTLASS_HOST_DEVICE GemmCoord operator/(Base const &b) const 
Element-wise division. 
Definition: include/cutlass/gemm/gemm.h:223
CUTLASS_HOST_DEVICE GemmCoord(Index m, Index n, Index k)
Helper to construct from a K, N, M, batch variables. 
Definition: include/cutlass/gemm/gemm.h:125
int Index
Integer-valued index. 
Definition: include/cutlass/gemm/gemm.h:263
CUTLASS_HOST_DEVICE half_t & operator+=(half_t &lhs, half_t const &rhs)
Definition: half.h:654
CUTLASS_HOST_DEVICE Coord< 2 > nk() const 
Obtains a Coord<2> from GemmCoord. 
Definition: include/cutlass/gemm/gemm.h:189
CUTLASS_HOST_DEVICE BatchedGemmCoord(Base const &coord)
Constructs from Coord<4> 
Definition: include/cutlass/gemm/gemm.h:290
CUTLASS_HOST_DEVICE Index const & k() const 
Returns the GEMM K coordinate. 
Definition: include/cutlass/gemm/gemm.h:145
CUTLASS_HOST_DEVICE half_t & operator-=(half_t &lhs, half_t const &rhs)
Definition: half.h:664
Coord< 3, Index > Base
Base type is a Coord of rank=4. 
Definition: include/cutlass/gemm/gemm.h:100
static CUTLASS_HOST_DEVICE Coord< 3 > toCoord()
Returns a Coord object. 
Definition: include/cutlass/gemm/gemm.h:76
CUTLASS_HOST_DEVICE Coord< 2 > km() const 
Obtains a Coord<2> from GemmCoord. 
Definition: include/cutlass/gemm/gemm.h:183
Definition: include/cutlass/gemm/gemm.h:260
CUTLASS_HOST_DEVICE Coord< 3 > mnk() const 
Obtains a Coord<3> from GemmCoord. 
Definition: include/cutlass/gemm/gemm.h:153
CUTLASS_HOST_DEVICE GemmCoord operator-(Base const &b) const 
Element-wise subtraction. 
Definition: include/cutlass/gemm/gemm.h:211
CUTLASS_HOST_DEVICE Index const & batch() const 
Returns the GEMM batch coordinate. 
Definition: include/cutlass/gemm/gemm.h:322
CUTLASS_HOST_DEVICE BatchedGemmCoord & operator*=(Base const &b)
In-place multiplication. 
Definition: include/cutlass/gemm/gemm.h:384
CUTLASS_HOST_DEVICE BatchedGemmCoord operator*(Base const &b) const 
Element-wise multiplication. 
Definition: include/cutlass/gemm/gemm.h:358
#define CUTLASS_HOST_DEVICE
Definition: cutlass.h:89
CUTLASS_HOST_DEVICE Index const & k() const 
Returns the GEMM K coordinate. 
Definition: include/cutlass/gemm/gemm.h:314
Shape of a matrix multiply-add operation. 
Definition: include/cutlass/gemm/gemm.h:57
CUTLASS_HOST_DEVICE Coord< 2 > mk() const 
Obtains a Coord<2> from GemmCoord. 
Definition: include/cutlass/gemm/gemm.h:177
CUTLASS_HOST_DEVICE BatchedGemmCoord()
Default ctor. 
Definition: include/cutlass/gemm/gemm.h:286
CUTLASS_HOST_DEVICE Index & k()
Returns reference to the GEMM K coordinate. 
Definition: include/cutlass/gemm/gemm.h:318
CUTLASS_HOST_DEVICE Coord< 3 > knm() const 
Obtains a Coord<3> from GemmCoord. 
Definition: include/cutlass/gemm/gemm.h:159
CUTLASS_HOST_DEVICE GemmCoord & operator+=(Base const &b)
In-place addition. 
Definition: include/cutlass/gemm/gemm.h:229
CUTLASS_HOST_DEVICE half_t & operator*=(half_t &lhs, half_t const &rhs)
Definition: half.h:674
CUTLASS_HOST_DEVICE GemmCoord()
Default ctor. 
Definition: include/cutlass/gemm/gemm.h:117
CUTLASS_HOST_DEVICE BatchedGemmCoord & operator/=(Base const &b)
In-place division. 
Definition: include/cutlass/gemm/gemm.h:391
CUTLASS_HOST_DEVICE GemmCoord & operator-=(Base const &b)
In-place subtraction. 
Definition: include/cutlass/gemm/gemm.h:236
CUTLASS_HOST_DEVICE BatchedGemmCoord & operator+=(Base const &b)
In-place addition. 
Definition: include/cutlass/gemm/gemm.h:370
CUTLASS_HOST_DEVICE Coord< 2 > kn() const 
Obtains a Coord<2> from GemmCoord. 
Definition: include/cutlass/gemm/gemm.h:195
CUTLASS_HOST_DEVICE Index const & n() const 
Returns the GEMM N coordinate. 
Definition: include/cutlass/gemm/gemm.h:306
CUTLASS_HOST_DEVICE BatchedGemmCoord(Index m, Index n, Index k, Index b)
Helper to construct from a K, N, M, and batch variables. 
Definition: include/cutlass/gemm/gemm.h:294
CUTLASS_HOST_DEVICE Index const & m() const 
Returns the GEMM M coordinate. 
Definition: include/cutlass/gemm/gemm.h:129
CUTLASS_HOST_DEVICE GemmCoord & operator*=(Base const &b)
In-place multiplication. 
Definition: include/cutlass/gemm/gemm.h:243
CUTLASS_HOST_DEVICE BatchedGemmCoord & operator-=(Base const &b)
In-place subtraction. 
Definition: include/cutlass/gemm/gemm.h:377
Basic include for CUTLASS. 
CUTLASS_HOST_DEVICE Index & n()
Returns reference to the GEMM N coordinate. 
Definition: include/cutlass/gemm/gemm.h:141
CUTLASS_HOST_DEVICE GemmCoord operator*(Base const &b) const 
Element-wise multiplication. 
Definition: include/cutlass/gemm/gemm.h:217
CUTLASS_HOST_DEVICE Index const & m() const 
Returns the GEMM M coordinate. 
Definition: include/cutlass/gemm/gemm.h:298