<ahref="gemm__epilogue_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="l0002
<divclass="ttc"id="structcutlass_1_1gemm_1_1GemmEpilogue_html_a6bea57249e3645a4b08a11619cb1687e"><divclass="ttname"><ahref="structcutlass_1_1gemm_1_1GemmEpilogue.html#a6bea57249e3645a4b08a11619cb1687e">cutlass::gemm::GemmEpilogue::problem_size</a></div><divclass="ttdeci">Coord< 3 > problem_size</div><divclass="ttdoc">The dimensions of the GEMM. </div><divclass="ttdef"><b>Definition:</b> gemm_epilogue.h:215</div></div>
<divclass="ttc"id="structcutlass_1_1gemm_1_1GemmEpilogue_html_aab0a964efe223c5c29bc816c393b5a9a"><divclass="ttname"><ahref="structcutlass_1_1gemm_1_1GemmEpilogue.html#aab0a964efe223c5c29bc816c393b5a9a">cutlass::gemm::GemmEpilogue::SharedStoreIteratorD</a></div><divclass="ttdeci">Traits::SharedStoreIteratorD SharedStoreIteratorD</div><divclass="ttdoc">The iterator to store D in shared memory. </div><divclass="ttdef"><b>Definition:</b> gemm_epilogue.h:73</div></div>
<divclass="ttc"id="coord_8h_html"><divclass="ttname"><ahref="coord_8h.html">coord.h</a></div><divclass="ttdoc">A Coord is a coordinate of arbitrary rank into a tensor or matrix. </div></div>
<divclass="ttc"id="namespacecutlass_html_a7419519fa453a121dfa5f26bf87318d9"><divclass="ttname"><ahref="namespacecutlass.html#a7419519fa453a121dfa5f26bf87318d9">cutlass::make_Coord</a></div><divclass="ttdeci">CUTLASS_HOST_DEVICE Coord< 1 > make_Coord(int _0)</div><divclass="ttdoc">Helper to make a 2-element coordinate. </div><divclass="ttdef"><b>Definition:</b> coord.h:368</div></div>
<divclass="ttc"id="structcutlass_1_1gemm_1_1GemmEpilogue_html_a32f618ff19d984447fba7355d46a69a7"><divclass="ttname"><ahref="structcutlass_1_1gemm_1_1GemmEpilogue.html#a32f618ff19d984447fba7355d46a69a7">cutlass::gemm::GemmEpilogue::GlobalTransformerD</a></div><divclass="ttdeci">Traits::GlobalTransformerD GlobalTransformerD</div><divclass="ttdoc">The transformer for D. </div><divclass="ttdef"><b>Definition:</b> gemm_epilogue.h:69</div></div>
<divclass="ttc"id="structcutlass_1_1gemm_1_1GemmEpilogue_html_a9063e7fc044a679652d5a3a31aa77e7c"><divclass="ttname"><ahref="structcutlass_1_1gemm_1_1GemmEpilogue.html#a9063e7fc044a679652d5a3a31aa77e7c">cutlass::gemm::GemmEpilogue::SharedStoreTransformerD</a></div><divclass="ttdeci">Traits::SharedStoreTransformerD SharedStoreTransformerD</div><divclass="ttdoc">The shared store transformer for D. </div><divclass="ttdef"><b>Definition:</b> gemm_epilogue.h:75</div></div>
<divclass="ttc"id="structcutlass_1_1gemm_1_1GemmEpilogue_html_a1c766374d900535c944cf2a2de6925f4"><divclass="ttname"><ahref="structcutlass_1_1gemm_1_1GemmEpilogue.html#a1c766374d900535c944cf2a2de6925f4">cutlass::gemm::GemmEpilogue::GlobalStoreIteratorD</a></div><divclass="ttdeci">Traits::GlobalStoreIteratorD GlobalStoreIteratorD</div><divclass="ttdoc">The iterator for D in global memory. </div><divclass="ttdef"><b>Definition:</b> gemm_epilogue.h:71</div></div>
<divclass="ttc"id="structcutlass_1_1gemm_1_1GemmEpilogue_html_abb0741601652df8fdf927d49c2c0e4d0"><divclass="ttname"><ahref="structcutlass_1_1gemm_1_1GemmEpilogue.html#abb0741601652df8fdf927d49c2c0e4d0">cutlass::gemm::GemmEpilogue::ScalarC</a></div><divclass="ttdeci">GlobalLoadIteratorC::Scalar ScalarC</div><divclass="ttdoc">The scalar for C. </div><divclass="ttdef"><b>Definition:</b> gemm_epilogue.h:83</div></div>
<divclass="ttc"id="structcutlass_1_1gemm_1_1GemmEpilogue_html_a8e5d38a7795b50a021fc43dc7424288b"><divclass="ttname"><ahref="structcutlass_1_1gemm_1_1GemmEpilogue.html#a8e5d38a7795b50a021fc43dc7424288b">cutlass::gemm::GemmEpilogue::SharedLoadStreamD</a></div><divclass="ttdeci">Traits::SharedLoadStreamD SharedLoadStreamD</div><divclass="ttdoc">The iterator to load D in shared memory. </div><divclass="ttdef"><b>Definition:</b> gemm_epilogue.h:77</div></div>
<divclass="ttc"id="structcutlass_1_1gemm_1_1GemmEpilogue_html_a6c30bea1b2a1bd2e981025851d5b12d1"><divclass="ttname"><ahref="structcutlass_1_1gemm_1_1GemmEpilogue.html#a6c30bea1b2a1bd2e981025851d5b12d1">cutlass::gemm::GemmEpilogue::Functor</a></div><divclass="ttdeci">Traits::Functor Functor</div><divclass="ttdoc">The functor in charge of the math. </div><divclass="ttdef"><b>Definition:</b> gemm_epilogue.h:59</div></div>
<divclass="ttc"id="structcutlass_1_1gemm_1_1GemmEpilogue_html_a6f1f96715425b7196e960907676a7db3"><divclass="ttname"><ahref="structcutlass_1_1gemm_1_1GemmEpilogue.html#a6f1f96715425b7196e960907676a7db3">cutlass::gemm::GemmEpilogue::epilogue</a></div><divclass="ttdeci">CUTLASS_DEVICE void epilogue(Accumulators &accumulators, Coord< 3 > const &block=make_Coord(0, 0, 0), int batch_id=0)</div><divclass="ttdoc">Execute the epilogue. </div><divclass="ttdef"><b>Definition:</b> gemm_epilogue.h:94</div></div>
<divclass="ttc"id="structcutlass_1_1gemm_1_1GemmEpilogue_html_a8e3c978da6ed56239783bf4db0a936ae"><divclass="ttname"><ahref="structcutlass_1_1gemm_1_1GemmEpilogue.html#a8e3c978da6ed56239783bf4db0a936ae">cutlass::gemm::GemmEpilogue::Iterations</a></div><divclass="ttdeci">Traits::Iterations Iterations</div><divclass="ttdoc">The number of iterations. </div><divclass="ttdef"><b>Definition:</b> gemm_epilogue.h:53</div></div>
<divclass="ttc"id="fragment_8h_html"><divclass="ttname"><ahref="fragment_8h.html">fragment.h</a></div><divclass="ttdoc">Defines Fragment, a statically-sized array for storing parts of matrices within a thread&#39;s registers...</div></div>
<divclass="ttc"id="convert_8h_html"><divclass="ttname"><ahref="convert_8h.html">convert.h</a></div><divclass="ttdoc">Defines conversion operations among Fragments of different base type. </div></div>
<divclass="ttc"id="structcutlass_1_1gemm_1_1GemmEpilogue_html_a41edfd24b7dd2759f8b72ae8534182a9"><divclass="ttname"><ahref="structcutlass_1_1gemm_1_1GemmEpilogue.html#a41edfd24b7dd2759f8b72ae8534182a9">cutlass::gemm::GemmEpilogue::GlobalTransformerC</a></div><divclass="ttdeci">Traits::GlobalTransformerC GlobalTransformerC</div><divclass="ttdoc">The transformer for C. </div><divclass="ttdef"><b>Definition:</b> gemm_epilogue.h:67</div></div>
<divclass="ttc"id="structcutlass_1_1gemm_1_1GemmEpilogue_html_aecb5429363c7156ee3ad596fe250120a"><divclass="ttname"><ahref="structcutlass_1_1gemm_1_1GemmEpilogue.html#aecb5429363c7156ee3ad596fe250120a">cutlass::gemm::GemmEpilogue::GlobalLoadIteratorC</a></div><divclass="ttdeci">Traits::GlobalLoadIteratorC GlobalLoadIteratorC</div><divclass="ttdoc">We do not support 3D or 4D shapes. </div><divclass="ttdef"><b>Definition:</b> gemm_epilogue.h:62</div></div>
