<ahref="gemm__desc_8h.html">Go to the documentation of this file.</a><divclass="fragment"><divclass="line"><aname="l00001"></a><spanclass="lineno"> 1</span> <spanclass="comment">/***************************************************************************************************</span></div><divclass="line"><aname="l00002"></a><spanclass="lineno"> 2</span> <spanclass="comment"> * Copyright (c) 2017-2018, NVIDIA CORPORATION. All rights reserved.</span></div><divclass="line"><aname="l00003"></a><spanclass="lineno"> 3</span> <spanclass="comment"> *</span></div><divclass="line"><aname="l00004"></a><spanclass="lineno"> 4</span> <spanclass="comment"> * Redistribution and use in source and binary forms, with or without modification, are permitted</span></div><divclass="line"><aname="l00005"></a><spanclass="lineno"> 5</span> <spanclass="comment"> * provided that the following conditions are met:</span></div><divclass="line"><aname="l00006"></a><spanclass="lineno"> 6</span> <spanclass="comment"> * * Redistributions of source code must retain the above copyright notice, this list of</span></div><divclass="line"><aname="l00007"></a><spanclass="lineno"> 7</span> <spanclass="comment"> * conditions and the following disclaimer.</span></div><divclass="line"><aname="l00008"></a><spanclass="lineno"> 8</span> <spanclass="comment"> * * Redistributions in binary form must reproduce the above copyright notice, this list of</span></div><divclass="line"><aname="l00009"></a><spanclass="lineno"> 9</span> <spanclass="comment"> * conditions and the following disclaimer in the documentation and/or other materials</span></div><divclass="line"><aname="l00010"></a><spanclass="lineno"> 10</span> <spanclass="comment"> * provided with the distribution.</span></div><divclass="line"><aname="l00011"></a><spanclass="lineno"> 11</span> <spanclass="comment"> * * Neither the name of the NVIDIA CORPORATION nor the names of its contributors may be used</span></div><divclass="line"><aname="l00012"></a><spanclass="lineno"> 12</span> <spanclass="comment"> * to endorse or promote products derived from this software without specific prior written</span></div><divclass="line"><aname="l00013"></a><spanclass="lineno"> 13</span> <spanclass="comment"> * permission.</span></div><divclass="line"><aname="l00014"></a><spanclass="lineno"> 14</span> <spanclass="comment"> *</span></div><divclass="line"><aname="l00015"></a><spanclass="lineno"> 15</span> <spanclass="comment"> * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR</span></div><divclass="line"><aname="l00016"></a><spanclass="lineno"> 16</span> <spanclass="comment"> * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND</span></div><divclass="line"><aname="l00017"></a><spanclass="lineno"> 17</span> <spanclass="comment"> * FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL NVIDIA CORPORATION BE LIABLE</span></div><divclass="line"><aname="l00018"></a><spanclass="lineno"> 18</span> <spanclass="comment"> * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,</span></div><divclass="line"><aname="l00019"></a><spanclass="lineno"> 19</span> <spanclass="comment"> * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;</span></div><divclass="line"><aname="l00020"></a><spanclass="lineno"> 20</span> <spanclass="comment"> * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,</span></div><divclass="line"><aname="l00021"></a><spanclass="lineno"> 21</span> <spanclass="comment"> * STRICT LIABILITY, OR TOR (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE</span></div><divclass="line"><aname="l00022"><
<divclass="ttc"id="structcutlass_1_1gemm_1_1GemmDesc_html_a51f77118d545308068b0dc287abf4e8b"><divclass="ttname"><ahref="structcutlass_1_1gemm_1_1GemmDesc.html#a51f77118d545308068b0dc287abf4e8b">cutlass::gemm::GemmDesc::TensorRefC</a></div><divclass="ttdeci">TensorRef< CType const, 2 > TensorRefC</div><divclass="ttdoc">Tensor reference to C operand. </div><divclass="ttdef"><b>Definition:</b> gemm_desc.h:74</div></div>
<divclass="ttc"id="tensor__ref_8h_html"><divclass="ttname"><ahref="tensor__ref_8h.html">tensor_ref.h</a></div><divclass="ttdoc">Defines a structure containing strides, bounds, and a pointer to tensor data. </div></div>
<divclass="ttc"id="structcutlass_1_1gemm_1_1GemmDesc_html_af96cd9c8f2454ea118dfa7358dd24824"><divclass="ttname"><ahref="structcutlass_1_1gemm_1_1GemmDesc.html#af96cd9c8f2454ea118dfa7358dd24824">cutlass::gemm::GemmDesc::GemmDesc</a></div><divclass="ttdeci">CUTLASS_HOST_DEVICE GemmDesc(GemmCoord _problem_size, SType _alpha, TensorRefA const &_A, long long _batch_stride_A, TensorRefB const &_B, long long _batch_stride_B, SType _beta, TensorRefC const &_C, long long _batch_stride_C, TensorRefD const &_D, long long _batch_stride_D)</div><divclass="ttdoc">Constructor for strided batch GEMM GEMM. </div><divclass="ttdef"><b>Definition:</b> gemm_desc.h:179</div></div>
<divclass="ttc"id="structcutlass_1_1gemm_1_1GemmDesc_html_ab384a226ab370fa5e25468fb99c63e30"><divclass="ttname"><ahref="structcutlass_1_1gemm_1_1GemmDesc.html#ab384a226ab370fa5e25468fb99c63e30">cutlass::gemm::GemmDesc::C</a></div><divclass="ttdeci">TensorRefC C</div><divclass="ttdoc">The source matrix C. </div><divclass="ttdef"><b>Definition:</b> gemm_desc.h:111</div></div>
<divclass="ttc"id="structcutlass_1_1gemm_1_1GemmDesc_html_a80b0aae6e67b733ae5bf289d979a7c9b"><divclass="ttname"><ahref="structcutlass_1_1gemm_1_1GemmDesc.html#a80b0aae6e67b733ae5bf289d979a7c9b">cutlass::gemm::GemmDesc::A</a></div><divclass="ttdeci">TensorRefA A</div><divclass="ttdoc">The source matrix A. </div><divclass="ttdef"><b>Definition:</b> gemm_desc.h:96</div></div>
<divclass="ttc"id="structcutlass_1_1gemm_1_1GemmDesc_html_a52a5538b1b5208090417cf98d8735e3f"><divclass="ttname"><ahref="structcutlass_1_1gemm_1_1GemmDesc.html#a52a5538b1b5208090417cf98d8735e3f">cutlass::gemm::GemmDesc::problem_size</a></div><divclass="ttdeci">GemmCoord problem_size</div><divclass="ttdoc">The dimensions of the GEMM. </div><divclass="ttdef"><b>Definition:</b> gemm_desc.h:90</div></div>
<divclass="ttc"id="structcutlass_1_1gemm_1_1GemmDesc_html_a7b9afcc7e3105da1d002b1baa68d83de"><divclass="ttname"><ahref="structcutlass_1_1gemm_1_1GemmDesc.html#a7b9afcc7e3105da1d002b1baa68d83de">cutlass::gemm::GemmDesc::batch_stride_D</a></div><divclass="ttdeci">long long batch_stride_D</div><divclass="ttdoc">batch stride for D operand </div><divclass="ttdef"><b>Definition:</b> gemm_desc.h:120</div></div>
<divclass="ttc"id="structcutlass_1_1gemm_1_1GemmDesc_html_af0ac89b161f9cad96307f1ff3c80a774"><divclass="ttname"><ahref="structcutlass_1_1gemm_1_1GemmDesc.html#af0ac89b161f9cad96307f1ff3c80a774">cutlass::gemm::GemmDesc::B</a></div><divclass="ttdeci">TensorRefB B</div><divclass="ttdoc">The source matrix B. </div><divclass="ttdef"><b>Definition:</b> gemm_desc.h:102</div></div>
<divclass="ttc"id="structcutlass_1_1gemm_1_1GemmDesc_html_aea3a523662dc801bf9abf205a8e400fe"><divclass="ttname"><ahref="structcutlass_1_1gemm_1_1GemmDesc.html#aea3a523662dc801bf9abf205a8e400fe">cutlass::gemm::GemmDesc::TensorRefA</a></div><divclass="ttdeci">TensorRef< AType const, 2 > TensorRefA</div><divclass="ttdoc">Tensor reference to A operand. </div><divclass="ttdef"><b>Definition:</b> gemm_desc.h:62</div></div>
<divclass="ttc"id="structcutlass_1_1gemm_1_1GemmDesc_html_a8b5a86d14eba0d3c5173753212d62599"><divclass="ttname"><ahref="structcutlass_1_1gemm_1_1GemmDesc.html#a8b5a86d14eba0d3c5173753212d62599">cutlass::gemm::GemmDesc::batch_stride_A</a></div><divclass="ttdeci">long long batch_stride_A</div><divclass="ttdoc">batch stride for A operand </div><divclass="ttdef"><b>Definition:</b> gemm_desc.h:99</div></div>
<divclass="ttc"id="structcutlass_1_1gemm_1_1GemmDesc_html_a258e8c008db365b88654f46df3017ec2"><divclass="ttname"><ahref="structcutlass_1_1gemm_1_1GemmDesc.html#a258e8c008db365b88654f46df3017ec2">cutlass::gemm::GemmDesc::SType</a></div><divclass="ttdeci">SType_ SType</div><divclass="ttdoc">Scalar type for alpha and beta. </div><divclass="ttdef"><b>Definition:</b> gemm_desc.h:83</div></div>
<divclass="ttc"id="structcutlass_1_1gemm_1_1GemmDesc_html_a95e3fe05e4ca0d4019cbef2b1a54419a"><divclass="ttname"><ahref="structcutlass_1_1gemm_1_1GemmDesc.html#a95e3fe05e4ca0d4019cbef2b1a54419a">cutlass::gemm::GemmDesc::batch_stride_B</a></div><divclass="ttdeci">long long batch_stride_B</div><divclass="ttdoc">batch stride for B operand </div><divclass="ttdef"><b>Definition:</b> gemm_desc.h:105</div></div>
<divclass="ttc"id="structcutlass_1_1gemm_1_1GemmDesc_html_adf95451422c529587d55aac0fecf0d9f"><divclass="ttname"><ahref="structcutlass_1_1gemm_1_1GemmDesc.html#adf95451422c529587d55aac0fecf0d9f">cutlass::gemm::GemmDesc::D</a></div><divclass="ttdeci">TensorRefD D</div><divclass="ttdoc">The destination matrix D. </div><divclass="ttdef"><b>Definition:</b> gemm_desc.h:117</div></div>
<divclass="ttc"id="structcutlass_1_1gemm_1_1GemmDesc_html_a1f59ad60a529aa1db6e66ade3cb65052"><divclass="ttname"><ahref="structcutlass_1_1gemm_1_1GemmDesc.html#a1f59ad60a529aa1db6e66ade3cb65052">cutlass::gemm::GemmDesc::TensorRefD</a></div><divclass="ttdeci">TensorRef< DType, 2 > TensorRefD</div><divclass="ttdoc">Tensor reference to D operand. </div><divclass="ttdef"><b>Definition:</b> gemm_desc.h:80</div></div>
<divclass="ttc"id="structcutlass_1_1gemm_1_1GemmCoord_html_ac441e93411ed6954a93ab36f0c747d26"><divclass="ttname"><ahref="structcutlass_1_1gemm_1_1GemmCoord.html#ac441e93411ed6954a93ab36f0c747d26">cutlass::gemm::GemmCoord::batch</a></div><divclass="ttdeci">CUTLASS_HOST_DEVICE Index const & batch() const</div><divclass="ttdoc">Returns the GEMM batch coordinate. </div><divclass="ttdef"><b>Definition:</b> gemm_coord.h:113</div></div>
<divclass="ttc"id="structcutlass_1_1gemm_1_1GemmDesc_html_aad3590dffa2e1ba82c834efae6b35ad2"><divclass="ttname"><ahref="structcutlass_1_1gemm_1_1GemmDesc.html#aad3590dffa2e1ba82c834efae6b35ad2">cutlass::gemm::GemmDesc::batch_stride_C</a></div><divclass="ttdeci">long long batch_stride_C</div><divclass="ttdoc">batch stride for C operand </div><divclass="ttdef"><b>Definition:</b> gemm_desc.h:114</div></div>
<divclass="ttc"id="structcutlass_1_1gemm_1_1GemmDesc_html_abd12fea9779ada02c0f2266909602171"><divclass="ttname"><ahref="structcutlass_1_1gemm_1_1GemmDesc.html#abd12fea9779ada02c0f2266909602171">cutlass::gemm::GemmDesc::Index</a></div><divclass="ttdeci">Index_ Index</div><divclass="ttdoc">Index type for dimensions and strides. </div><divclass="ttdef"><b>Definition:</b> gemm_desc.h:56</div></div>
<divclass="ttc"id="gemm__coord_8h_html"><divclass="ttname"><ahref="gemm__coord_8h.html">gemm_coord.h</a></div><divclass="ttdoc">GemmCoord is a structure derived from Coord<4> that specifies a location within the coordinate system...</div></div>