743 lines
46 KiB
HTML
743 lines
46 KiB
HTML
<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
|
|
<html xmlns="http://www.w3.org/1999/xhtml">
|
|
<head>
|
|
<meta http-equiv="Content-Type" content="text/xhtml;charset=UTF-8"/>
|
|
<meta http-equiv="X-UA-Compatible" content="IE=9"/>
|
|
<meta name="generator" content="Doxygen 1.8.14"/>
|
|
<meta name="viewport" content="width=device-width, initial-scale=1"/>
|
|
<title>Cutlass: cutlass::gemm::GemmEpilogue< GemmEpilogueTraits_ > Struct Template Reference</title>
|
|
<link href="tabs.css" rel="stylesheet" type="text/css"/>
|
|
<script type="text/javascript" src="jquery.js"></script>
|
|
<script type="text/javascript" src="dynsections.js"></script>
|
|
<link href="search/search.css" rel="stylesheet" type="text/css"/>
|
|
<script type="text/javascript" src="search/searchdata.js"></script>
|
|
<script type="text/javascript" src="search/search.js"></script>
|
|
<script type="text/x-mathjax-config">
|
|
MathJax.Hub.Config({
|
|
extensions: ["tex2jax.js"],
|
|
jax: ["input/TeX","output/HTML-CSS"],
|
|
});
|
|
</script><script type="text/javascript" async src="http://cdn.mathjax.org/mathjax/latest/MathJax.js"></script>
|
|
<link href="doxygen.css" rel="stylesheet" type="text/css" />
|
|
</head>
|
|
<body>
|
|
<div id="top"><!-- do not remove this div, it is closed by doxygen! -->
|
|
<div id="titlearea">
|
|
<table cellspacing="0" cellpadding="0">
|
|
<tbody>
|
|
<tr style="height: 56px;">
|
|
<td id="projectalign" style="padding-left: 0.5em;">
|
|
<div id="projectname">Cutlass
|
|
</div>
|
|
<div id="projectbrief">CUDA Templates for Linear Algebra Subroutines and Solvers</div>
|
|
</td>
|
|
</tr>
|
|
</tbody>
|
|
</table>
|
|
</div>
|
|
<!-- end header part -->
|
|
<!-- Generated by Doxygen 1.8.14 -->
|
|
<script type="text/javascript">
|
|
/* @license magnet:?xt=urn:btih:cf05388f2679ee054f2beb29a391d25f4e673ac3&dn=gpl-2.0.txt GPL-v2 */
|
|
var searchBox = new SearchBox("searchBox", "search",false,'Search');
|
|
/* @license-end */
|
|
</script>
|
|
<script type="text/javascript" src="menudata.js"></script>
|
|
<script type="text/javascript" src="menu.js"></script>
|
|
<script type="text/javascript">
|
|
/* @license magnet:?xt=urn:btih:cf05388f2679ee054f2beb29a391d25f4e673ac3&dn=gpl-2.0.txt GPL-v2 */
|
|
$(function() {
|
|
initMenu('',true,false,'search.php','Search');
|
|
$(document).ready(function() { init_search(); });
|
|
});
|
|
/* @license-end */</script>
|
|
<div id="main-nav"></div>
|
|
<!-- window showing the filter options -->
|
|
<div id="MSearchSelectWindow"
|
|
onmouseover="return searchBox.OnSearchSelectShow()"
|
|
onmouseout="return searchBox.OnSearchSelectHide()"
|
|
onkeydown="return searchBox.OnSearchSelectKey(event)">
|
|
</div>
|
|
|
|
<!-- iframe showing the search results (closed by default) -->
|
|
<div id="MSearchResultsWindow">
|
|
<iframe src="javascript:void(0)" frameborder="0"
|
|
name="MSearchResults" id="MSearchResults">
|
|
</iframe>
|
|
</div>
|
|
|
|
<div id="nav-path" class="navpath">
|
|
<ul>
|
|
<li class="navelem"><a class="el" href="namespacecutlass.html">cutlass</a></li><li class="navelem"><a class="el" href="namespacecutlass_1_1gemm.html">gemm</a></li><li class="navelem"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html">GemmEpilogue</a></li> </ul>
|
|
</div>
|
|
</div><!-- top -->
|
|
<div class="header">
|
|
<div class="summary">
|
|
<a href="#pub-types">Public Types</a> |
|
|
<a href="#pub-methods">Public Member Functions</a> |
|
|
<a href="#pub-attribs">Public Attributes</a> |
|
|
<a href="structcutlass_1_1gemm_1_1GemmEpilogue-members.html">List of all members</a> </div>
|
|
<div class="headertitle">
|
|
<div class="title">cutlass::gemm::GemmEpilogue< GemmEpilogueTraits_ > Struct Template Reference</div> </div>
|
|
</div><!--header-->
|
|
<div class="contents">
|
|
|
|
<p><code>#include <<a class="el" href="gemm__epilogue_8h_source.html">gemm_epilogue.h</a>></code></p>
|
|
<div class="dynheader">
|
|
Inheritance diagram for cutlass::gemm::GemmEpilogue< GemmEpilogueTraits_ >:</div>
|
|
<div class="dyncontent">
|
|
<div class="center">
|
|
<img src="structcutlass_1_1gemm_1_1GemmEpilogue.png" usemap="#cutlass::gemm::GemmEpilogue_3C_20GemmEpilogueTraits_5F_20_3E_map" alt=""/>
|
|
<map id="cutlass::gemm::GemmEpilogue_3C_20GemmEpilogueTraits_5F_20_3E_map" name="cutlass::gemm::GemmEpilogue_3C_20GemmEpilogueTraits_5F_20_3E_map">
|
|
<area href="structcutlass_1_1gemm_1_1IgemmEpilogue.html" alt="cutlass::gemm::IgemmEpilogue< GemmEpilogueTraits_, bool >" shape="rect" coords="0,56,357,80"/>
|
|
<area href="structcutlass_1_1gemm_1_1IgemmEpilogue_3_01GemmEpilogueTraits___00_01true_01_4.html" alt="cutlass::gemm::IgemmEpilogue< GemmEpilogueTraits_, true >" shape="rect" coords="367,56,724,80"/>
|
|
</map>
|
|
</div></div>
|
|
<table class="memberdecls">
|
|
<tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="pub-types"></a>
|
|
Public Types</h2></td></tr>
|
|
<tr class="memitem:a645ab6e9e63163ee6bf536717a30fb1b"><td class="memItemLeft" align="right" valign="top">typedef GemmEpilogueTraits_ </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#a645ab6e9e63163ee6bf536717a30fb1b">Traits</a></td></tr>
|
|
<tr class="memdesc:a645ab6e9e63163ee6bf536717a30fb1b"><td class="mdescLeft"> </td><td class="mdescRight">The traits class. <a href="#a645ab6e9e63163ee6bf536717a30fb1b">More...</a><br /></td></tr>
|
|
<tr class="separator:a645ab6e9e63163ee6bf536717a30fb1b"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:ae5209fa80705442693833c63d535161e"><td class="memItemLeft" align="right" valign="top">typedef Traits::Params </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#ae5209fa80705442693833c63d535161e">Params</a></td></tr>
|
|
<tr class="memdesc:ae5209fa80705442693833c63d535161e"><td class="mdescLeft"> </td><td class="mdescRight">The params. <a href="#ae5209fa80705442693833c63d535161e">More...</a><br /></td></tr>
|
|
<tr class="separator:ae5209fa80705442693833c63d535161e"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:ac36dad8a7b6bc7fc6ef88e44068468dc"><td class="memItemLeft" align="right" valign="top">typedef Traits::SharedStorage </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#ac36dad8a7b6bc7fc6ef88e44068468dc">SharedStorage</a></td></tr>
|
|
<tr class="memdesc:ac36dad8a7b6bc7fc6ef88e44068468dc"><td class="mdescLeft"> </td><td class="mdescRight">The shared storage. <a href="#ac36dad8a7b6bc7fc6ef88e44068468dc">More...</a><br /></td></tr>
|
|
<tr class="separator:ac36dad8a7b6bc7fc6ef88e44068468dc"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a92a135fac401d43a8d2f14982d90274b"><td class="memItemLeft" align="right" valign="top">typedef Traits::OutputTile </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#a92a135fac401d43a8d2f14982d90274b">OutputTile</a></td></tr>
|
|
<tr class="memdesc:a92a135fac401d43a8d2f14982d90274b"><td class="mdescLeft"> </td><td class="mdescRight">The output tile. <a href="#a92a135fac401d43a8d2f14982d90274b">More...</a><br /></td></tr>
|
|
<tr class="separator:a92a135fac401d43a8d2f14982d90274b"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a8e3c978da6ed56239783bf4db0a936ae"><td class="memItemLeft" align="right" valign="top">typedef Traits::Iterations </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#a8e3c978da6ed56239783bf4db0a936ae">Iterations</a></td></tr>
|
|
<tr class="memdesc:a8e3c978da6ed56239783bf4db0a936ae"><td class="mdescLeft"> </td><td class="mdescRight">The number of iterations. <a href="#a8e3c978da6ed56239783bf4db0a936ae">More...</a><br /></td></tr>
|
|
<tr class="separator:a8e3c978da6ed56239783bf4db0a936ae"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:afe6bebd94e3379c94054d04c5196edce"><td class="memItemLeft" align="right" valign="top">typedef Traits::Accumulators </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#afe6bebd94e3379c94054d04c5196edce">Accumulators</a></td></tr>
|
|
<tr class="memdesc:afe6bebd94e3379c94054d04c5196edce"><td class="mdescLeft"> </td><td class="mdescRight">The accumulators. <a href="#afe6bebd94e3379c94054d04c5196edce">More...</a><br /></td></tr>
|
|
<tr class="separator:afe6bebd94e3379c94054d04c5196edce"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a0d38914bf97084e04102e7897aee4295"><td class="memItemLeft" align="right" valign="top">typedef Traits::Scalar </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#a0d38914bf97084e04102e7897aee4295">Scalar</a></td></tr>
|
|
<tr class="memdesc:a0d38914bf97084e04102e7897aee4295"><td class="mdescLeft"> </td><td class="mdescRight">The scalar. <a href="#a0d38914bf97084e04102e7897aee4295">More...</a><br /></td></tr>
|
|
<tr class="separator:a0d38914bf97084e04102e7897aee4295"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a6c30bea1b2a1bd2e981025851d5b12d1"><td class="memItemLeft" align="right" valign="top">typedef Traits::Functor </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#a6c30bea1b2a1bd2e981025851d5b12d1">Functor</a></td></tr>
|
|
<tr class="memdesc:a6c30bea1b2a1bd2e981025851d5b12d1"><td class="mdescLeft"> </td><td class="mdescRight">The functor in charge of the math. <a href="#a6c30bea1b2a1bd2e981025851d5b12d1">More...</a><br /></td></tr>
|
|
<tr class="separator:a6c30bea1b2a1bd2e981025851d5b12d1"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:aecb5429363c7156ee3ad596fe250120a"><td class="memItemLeft" align="right" valign="top">typedef Traits::GlobalLoadIteratorC </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#aecb5429363c7156ee3ad596fe250120a">GlobalLoadIteratorC</a></td></tr>
|
|
<tr class="memdesc:aecb5429363c7156ee3ad596fe250120a"><td class="mdescLeft"> </td><td class="mdescRight">We do not support 3D or 4D shapes. <a href="#aecb5429363c7156ee3ad596fe250120a">More...</a><br /></td></tr>
|
|
<tr class="separator:aecb5429363c7156ee3ad596fe250120a"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a41edfd24b7dd2759f8b72ae8534182a9"><td class="memItemLeft" align="right" valign="top">typedef Traits::GlobalTransformerC </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#a41edfd24b7dd2759f8b72ae8534182a9">GlobalTransformerC</a></td></tr>
|
|
<tr class="memdesc:a41edfd24b7dd2759f8b72ae8534182a9"><td class="mdescLeft"> </td><td class="mdescRight">The transformer for C. <a href="#a41edfd24b7dd2759f8b72ae8534182a9">More...</a><br /></td></tr>
|
|
<tr class="separator:a41edfd24b7dd2759f8b72ae8534182a9"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a32f618ff19d984447fba7355d46a69a7"><td class="memItemLeft" align="right" valign="top">typedef Traits::GlobalTransformerD </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#a32f618ff19d984447fba7355d46a69a7">GlobalTransformerD</a></td></tr>
|
|
<tr class="memdesc:a32f618ff19d984447fba7355d46a69a7"><td class="mdescLeft"> </td><td class="mdescRight">The transformer for D. <a href="#a32f618ff19d984447fba7355d46a69a7">More...</a><br /></td></tr>
|
|
<tr class="separator:a32f618ff19d984447fba7355d46a69a7"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a1c766374d900535c944cf2a2de6925f4"><td class="memItemLeft" align="right" valign="top">typedef Traits::GlobalStoreIteratorD </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#a1c766374d900535c944cf2a2de6925f4">GlobalStoreIteratorD</a></td></tr>
|
|
<tr class="memdesc:a1c766374d900535c944cf2a2de6925f4"><td class="mdescLeft"> </td><td class="mdescRight">The iterator for D in global memory. <a href="#a1c766374d900535c944cf2a2de6925f4">More...</a><br /></td></tr>
|
|
<tr class="separator:a1c766374d900535c944cf2a2de6925f4"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:aab0a964efe223c5c29bc816c393b5a9a"><td class="memItemLeft" align="right" valign="top">typedef Traits::SharedStoreIteratorD </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#aab0a964efe223c5c29bc816c393b5a9a">SharedStoreIteratorD</a></td></tr>
|
|
<tr class="memdesc:aab0a964efe223c5c29bc816c393b5a9a"><td class="mdescLeft"> </td><td class="mdescRight">The iterator to store D in shared memory. <a href="#aab0a964efe223c5c29bc816c393b5a9a">More...</a><br /></td></tr>
|
|
<tr class="separator:aab0a964efe223c5c29bc816c393b5a9a"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a9063e7fc044a679652d5a3a31aa77e7c"><td class="memItemLeft" align="right" valign="top">typedef Traits::SharedStoreTransformerD </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#a9063e7fc044a679652d5a3a31aa77e7c">SharedStoreTransformerD</a></td></tr>
|
|
<tr class="memdesc:a9063e7fc044a679652d5a3a31aa77e7c"><td class="mdescLeft"> </td><td class="mdescRight">The shared store transformer for D. <a href="#a9063e7fc044a679652d5a3a31aa77e7c">More...</a><br /></td></tr>
|
|
<tr class="separator:a9063e7fc044a679652d5a3a31aa77e7c"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a8e5d38a7795b50a021fc43dc7424288b"><td class="memItemLeft" align="right" valign="top">typedef Traits::SharedLoadStreamD </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#a8e5d38a7795b50a021fc43dc7424288b">SharedLoadStreamD</a></td></tr>
|
|
<tr class="memdesc:a8e5d38a7795b50a021fc43dc7424288b"><td class="mdescLeft"> </td><td class="mdescRight">The iterator to load D in shared memory. <a href="#a8e5d38a7795b50a021fc43dc7424288b">More...</a><br /></td></tr>
|
|
<tr class="separator:a8e5d38a7795b50a021fc43dc7424288b"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a07c93d583bfddd8f916fba6ef809832e"><td class="memItemLeft" align="right" valign="top">typedef Traits::Index </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#a07c93d583bfddd8f916fba6ef809832e">Index</a></td></tr>
|
|
<tr class="memdesc:a07c93d583bfddd8f916fba6ef809832e"><td class="mdescLeft"> </td><td class="mdescRight">The index. <a href="#a07c93d583bfddd8f916fba6ef809832e">More...</a><br /></td></tr>
|
|
<tr class="separator:a07c93d583bfddd8f916fba6ef809832e"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:abb0741601652df8fdf927d49c2c0e4d0"><td class="memItemLeft" align="right" valign="top">typedef GlobalLoadIteratorC::Scalar </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#abb0741601652df8fdf927d49c2c0e4d0">ScalarC</a></td></tr>
|
|
<tr class="memdesc:abb0741601652df8fdf927d49c2c0e4d0"><td class="mdescLeft"> </td><td class="mdescRight">The scalar for C. <a href="#abb0741601652df8fdf927d49c2c0e4d0">More...</a><br /></td></tr>
|
|
<tr class="separator:abb0741601652df8fdf927d49c2c0e4d0"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a4887b56a96694ce6350db77f78bb505f"><td class="memItemLeft" align="right" valign="top">typedef GlobalStoreIteratorD::Scalar </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#a4887b56a96694ce6350db77f78bb505f">ScalarD</a></td></tr>
|
|
<tr class="memdesc:a4887b56a96694ce6350db77f78bb505f"><td class="mdescLeft"> </td><td class="mdescRight">The scalar for D. <a href="#a4887b56a96694ce6350db77f78bb505f">More...</a><br /></td></tr>
|
|
<tr class="separator:a4887b56a96694ce6350db77f78bb505f"><td class="memSeparator" colspan="2"> </td></tr>
|
|
</table><table class="memberdecls">
|
|
<tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="pub-methods"></a>
|
|
Public Member Functions</h2></td></tr>
|
|
<tr class="memitem:af91ebe8d2ed73808b9ea8846c140d3f8"><td class="memItemLeft" align="right" valign="top">CUTLASS_DEVICE </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#af91ebe8d2ed73808b9ea8846c140d3f8">GemmEpilogue</a> (<a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#ae5209fa80705442693833c63d535161e">Params</a> const &params_, <a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#ac36dad8a7b6bc7fc6ef88e44068468dc">SharedStorage</a> &shared_storage_, <a class="el" href="structcutlass_1_1Coord.html">Coord</a>< 3 > const &_problem_size)</td></tr>
|
|
<tr class="memdesc:af91ebe8d2ed73808b9ea8846c140d3f8"><td class="mdescLeft"> </td><td class="mdescRight">Ctor. <a href="#af91ebe8d2ed73808b9ea8846c140d3f8">More...</a><br /></td></tr>
|
|
<tr class="separator:af91ebe8d2ed73808b9ea8846c140d3f8"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a6f1f96715425b7196e960907676a7db3"><td class="memItemLeft" align="right" valign="top">CUTLASS_DEVICE void </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#a6f1f96715425b7196e960907676a7db3">epilogue</a> (<a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#afe6bebd94e3379c94054d04c5196edce">Accumulators</a> &accumulators, <a class="el" href="structcutlass_1_1Coord.html">Coord</a>< 3 > const &block=<a class="el" href="namespacecutlass.html#a7419519fa453a121dfa5f26bf87318d9">make_Coord</a>(0, 0, 0), int batch_id=0)</td></tr>
|
|
<tr class="memdesc:a6f1f96715425b7196e960907676a7db3"><td class="mdescLeft"> </td><td class="mdescRight">Execute the epilogue. <a href="#a6f1f96715425b7196e960907676a7db3">More...</a><br /></td></tr>
|
|
<tr class="separator:a6f1f96715425b7196e960907676a7db3"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a7af2eb421840e037263e6a144cca5c32"><td class="memTemplParams" colspan="2">template<bool kSourceRequired> </td></tr>
|
|
<tr class="memitem:a7af2eb421840e037263e6a144cca5c32"><td class="memTemplItemLeft" align="right" valign="top">CUTLASS_DEVICE void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#a7af2eb421840e037263e6a144cca5c32">epilogue_with_or_without_beta</a> (<a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#afe6bebd94e3379c94054d04c5196edce">Accumulators</a> &accumulators, <a class="el" href="structcutlass_1_1Coord.html">Coord</a>< 3 > const &block, int batch_id)</td></tr>
|
|
<tr class="separator:a7af2eb421840e037263e6a144cca5c32"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a9b5e42f222fec98ff479bc1650221b84"><td class="memItemLeft" align="right" valign="top">CUTLASS_DEVICE void </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#a9b5e42f222fec98ff479bc1650221b84">shared_load_fence</a> ()</td></tr>
|
|
<tr class="memdesc:a9b5e42f222fec98ff479bc1650221b84"><td class="mdescLeft"> </td><td class="mdescRight">The memory fence for shared loads. <a href="#a9b5e42f222fec98ff479bc1650221b84">More...</a><br /></td></tr>
|
|
<tr class="separator:a9b5e42f222fec98ff479bc1650221b84"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:ac1b2a16b4ccf3e9617faf4d8a2c43691"><td class="memItemLeft" align="right" valign="top">CUTLASS_DEVICE void </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#ac1b2a16b4ccf3e9617faf4d8a2c43691">shared_store_fence</a> ()</td></tr>
|
|
<tr class="memdesc:ac1b2a16b4ccf3e9617faf4d8a2c43691"><td class="mdescLeft"> </td><td class="mdescRight">The memory fence for shared stores. <a href="#ac1b2a16b4ccf3e9617faf4d8a2c43691">More...</a><br /></td></tr>
|
|
<tr class="separator:ac1b2a16b4ccf3e9617faf4d8a2c43691"><td class="memSeparator" colspan="2"> </td></tr>
|
|
</table><table class="memberdecls">
|
|
<tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="pub-attribs"></a>
|
|
Public Attributes</h2></td></tr>
|
|
<tr class="memitem:a81b028a18df51d3caa1b0ba0c990e362"><td class="memItemLeft" align="right" valign="top"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#ae5209fa80705442693833c63d535161e">Params</a> const & </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#a81b028a18df51d3caa1b0ba0c990e362">params</a></td></tr>
|
|
<tr class="memdesc:a81b028a18df51d3caa1b0ba0c990e362"><td class="mdescLeft"> </td><td class="mdescRight">The params. <a href="#a81b028a18df51d3caa1b0ba0c990e362">More...</a><br /></td></tr>
|
|
<tr class="separator:a81b028a18df51d3caa1b0ba0c990e362"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a442b5b5688cd658c3b3476650c00281e"><td class="memItemLeft" align="right" valign="top"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#ac36dad8a7b6bc7fc6ef88e44068468dc">SharedStorage</a> & </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#a442b5b5688cd658c3b3476650c00281e">shared_storage</a></td></tr>
|
|
<tr class="memdesc:a442b5b5688cd658c3b3476650c00281e"><td class="mdescLeft"> </td><td class="mdescRight">The shared storage. <a href="#a442b5b5688cd658c3b3476650c00281e">More...</a><br /></td></tr>
|
|
<tr class="separator:a442b5b5688cd658c3b3476650c00281e"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a6bea57249e3645a4b08a11619cb1687e"><td class="memItemLeft" align="right" valign="top"><a class="el" href="structcutlass_1_1Coord.html">Coord</a>< 3 > </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#a6bea57249e3645a4b08a11619cb1687e">problem_size</a></td></tr>
|
|
<tr class="memdesc:a6bea57249e3645a4b08a11619cb1687e"><td class="mdescLeft"> </td><td class="mdescRight">The dimensions of the GEMM. <a href="#a6bea57249e3645a4b08a11619cb1687e">More...</a><br /></td></tr>
|
|
<tr class="separator:a6bea57249e3645a4b08a11619cb1687e"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a5aa254702fcf24630e04ceb174aa3d57"><td class="memItemLeft" align="right" valign="top"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#a6c30bea1b2a1bd2e981025851d5b12d1">Functor</a> </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#a5aa254702fcf24630e04ceb174aa3d57">functor</a></td></tr>
|
|
<tr class="separator:a5aa254702fcf24630e04ceb174aa3d57"><td class="memSeparator" colspan="2"> </td></tr>
|
|
</table>
|
|
<h2 class="groupheader">Member Typedef Documentation</h2>
|
|
<a id="afe6bebd94e3379c94054d04c5196edce"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#afe6bebd94e3379c94054d04c5196edce">◆ </a></span>Accumulators</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmEpilogueTraits_ > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">typedef Traits::Accumulators <a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html">cutlass::gemm::GemmEpilogue</a>< GemmEpilogueTraits_ >::<a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#afe6bebd94e3379c94054d04c5196edce">Accumulators</a></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="a6c30bea1b2a1bd2e981025851d5b12d1"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a6c30bea1b2a1bd2e981025851d5b12d1">◆ </a></span>Functor</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmEpilogueTraits_ > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">typedef Traits::Functor <a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html">cutlass::gemm::GemmEpilogue</a>< GemmEpilogueTraits_ >::<a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#a6c30bea1b2a1bd2e981025851d5b12d1">Functor</a></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="aecb5429363c7156ee3ad596fe250120a"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#aecb5429363c7156ee3ad596fe250120a">◆ </a></span>GlobalLoadIteratorC</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmEpilogueTraits_ > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">typedef Traits::GlobalLoadIteratorC <a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html">cutlass::gemm::GemmEpilogue</a>< GemmEpilogueTraits_ >::<a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#aecb5429363c7156ee3ad596fe250120a">GlobalLoadIteratorC</a></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<p>The iterator for C in global memory. </p>
|
|
|
|
</div>
|
|
</div>
|
|
<a id="a1c766374d900535c944cf2a2de6925f4"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a1c766374d900535c944cf2a2de6925f4">◆ </a></span>GlobalStoreIteratorD</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmEpilogueTraits_ > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">typedef Traits::GlobalStoreIteratorD <a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html">cutlass::gemm::GemmEpilogue</a>< GemmEpilogueTraits_ >::<a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#a1c766374d900535c944cf2a2de6925f4">GlobalStoreIteratorD</a></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="a41edfd24b7dd2759f8b72ae8534182a9"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a41edfd24b7dd2759f8b72ae8534182a9">◆ </a></span>GlobalTransformerC</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmEpilogueTraits_ > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">typedef Traits::GlobalTransformerC <a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html">cutlass::gemm::GemmEpilogue</a>< GemmEpilogueTraits_ >::<a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#a41edfd24b7dd2759f8b72ae8534182a9">GlobalTransformerC</a></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="a32f618ff19d984447fba7355d46a69a7"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a32f618ff19d984447fba7355d46a69a7">◆ </a></span>GlobalTransformerD</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmEpilogueTraits_ > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">typedef Traits::GlobalTransformerD <a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html">cutlass::gemm::GemmEpilogue</a>< GemmEpilogueTraits_ >::<a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#a32f618ff19d984447fba7355d46a69a7">GlobalTransformerD</a></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="a07c93d583bfddd8f916fba6ef809832e"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a07c93d583bfddd8f916fba6ef809832e">◆ </a></span>Index</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmEpilogueTraits_ > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">typedef Traits::Index <a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html">cutlass::gemm::GemmEpilogue</a>< GemmEpilogueTraits_ >::<a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#a07c93d583bfddd8f916fba6ef809832e">Index</a></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="a8e3c978da6ed56239783bf4db0a936ae"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a8e3c978da6ed56239783bf4db0a936ae">◆ </a></span>Iterations</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmEpilogueTraits_ > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">typedef Traits::Iterations <a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html">cutlass::gemm::GemmEpilogue</a>< GemmEpilogueTraits_ >::<a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#a8e3c978da6ed56239783bf4db0a936ae">Iterations</a></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="a92a135fac401d43a8d2f14982d90274b"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a92a135fac401d43a8d2f14982d90274b">◆ </a></span>OutputTile</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmEpilogueTraits_ > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">typedef Traits::OutputTile <a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html">cutlass::gemm::GemmEpilogue</a>< GemmEpilogueTraits_ >::<a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#a92a135fac401d43a8d2f14982d90274b">OutputTile</a></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="ae5209fa80705442693833c63d535161e"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#ae5209fa80705442693833c63d535161e">◆ </a></span>Params</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmEpilogueTraits_ > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">typedef Traits::Params <a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html">cutlass::gemm::GemmEpilogue</a>< GemmEpilogueTraits_ >::<a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#ae5209fa80705442693833c63d535161e">Params</a></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="a0d38914bf97084e04102e7897aee4295"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a0d38914bf97084e04102e7897aee4295">◆ </a></span>Scalar</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmEpilogueTraits_ > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">typedef Traits::Scalar <a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html">cutlass::gemm::GemmEpilogue</a>< GemmEpilogueTraits_ >::<a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#a0d38914bf97084e04102e7897aee4295">Scalar</a></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="abb0741601652df8fdf927d49c2c0e4d0"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#abb0741601652df8fdf927d49c2c0e4d0">◆ </a></span>ScalarC</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmEpilogueTraits_ > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">typedef GlobalLoadIteratorC::Scalar <a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html">cutlass::gemm::GemmEpilogue</a>< GemmEpilogueTraits_ >::<a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#abb0741601652df8fdf927d49c2c0e4d0">ScalarC</a></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="a4887b56a96694ce6350db77f78bb505f"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a4887b56a96694ce6350db77f78bb505f">◆ </a></span>ScalarD</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmEpilogueTraits_ > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">typedef GlobalStoreIteratorD::Scalar <a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html">cutlass::gemm::GemmEpilogue</a>< GemmEpilogueTraits_ >::<a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#a4887b56a96694ce6350db77f78bb505f">ScalarD</a></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="a8e5d38a7795b50a021fc43dc7424288b"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a8e5d38a7795b50a021fc43dc7424288b">◆ </a></span>SharedLoadStreamD</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmEpilogueTraits_ > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">typedef Traits::SharedLoadStreamD <a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html">cutlass::gemm::GemmEpilogue</a>< GemmEpilogueTraits_ >::<a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#a8e5d38a7795b50a021fc43dc7424288b">SharedLoadStreamD</a></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="ac36dad8a7b6bc7fc6ef88e44068468dc"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#ac36dad8a7b6bc7fc6ef88e44068468dc">◆ </a></span>SharedStorage</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmEpilogueTraits_ > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">typedef Traits::SharedStorage <a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html">cutlass::gemm::GemmEpilogue</a>< GemmEpilogueTraits_ >::<a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#ac36dad8a7b6bc7fc6ef88e44068468dc">SharedStorage</a></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="aab0a964efe223c5c29bc816c393b5a9a"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#aab0a964efe223c5c29bc816c393b5a9a">◆ </a></span>SharedStoreIteratorD</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmEpilogueTraits_ > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">typedef Traits::SharedStoreIteratorD <a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html">cutlass::gemm::GemmEpilogue</a>< GemmEpilogueTraits_ >::<a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#aab0a964efe223c5c29bc816c393b5a9a">SharedStoreIteratorD</a></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="a9063e7fc044a679652d5a3a31aa77e7c"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a9063e7fc044a679652d5a3a31aa77e7c">◆ </a></span>SharedStoreTransformerD</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmEpilogueTraits_ > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">typedef Traits::SharedStoreTransformerD <a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html">cutlass::gemm::GemmEpilogue</a>< GemmEpilogueTraits_ >::<a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#a9063e7fc044a679652d5a3a31aa77e7c">SharedStoreTransformerD</a></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="a645ab6e9e63163ee6bf536717a30fb1b"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a645ab6e9e63163ee6bf536717a30fb1b">◆ </a></span>Traits</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmEpilogueTraits_ > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">typedef GemmEpilogueTraits_ <a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html">cutlass::gemm::GemmEpilogue</a>< GemmEpilogueTraits_ >::<a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#a645ab6e9e63163ee6bf536717a30fb1b">Traits</a></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<h2 class="groupheader">Constructor & Destructor Documentation</h2>
|
|
<a id="af91ebe8d2ed73808b9ea8846c140d3f8"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#af91ebe8d2ed73808b9ea8846c140d3f8">◆ </a></span>GemmEpilogue()</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmEpilogueTraits_ > </div>
|
|
<table class="mlabels">
|
|
<tr>
|
|
<td class="mlabels-left">
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">CUTLASS_DEVICE <a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html">cutlass::gemm::GemmEpilogue</a>< GemmEpilogueTraits_ >::<a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html">GemmEpilogue</a> </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#ae5209fa80705442693833c63d535161e">Params</a> const & </td>
|
|
<td class="paramname"><em>params_</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#ac36dad8a7b6bc7fc6ef88e44068468dc">SharedStorage</a> & </td>
|
|
<td class="paramname"><em>shared_storage_</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1Coord.html">Coord</a>< 3 > const & </td>
|
|
<td class="paramname"><em>_problem_size</em> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</td>
|
|
<td class="mlabels-right">
|
|
<span class="mlabels"><span class="mlabel">inline</span></span> </td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<h2 class="groupheader">Member Function Documentation</h2>
|
|
<a id="a6f1f96715425b7196e960907676a7db3"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a6f1f96715425b7196e960907676a7db3">◆ </a></span>epilogue()</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmEpilogueTraits_ > </div>
|
|
<table class="mlabels">
|
|
<tr>
|
|
<td class="mlabels-left">
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">CUTLASS_DEVICE void <a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html">cutlass::gemm::GemmEpilogue</a>< GemmEpilogueTraits_ >::epilogue </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#afe6bebd94e3379c94054d04c5196edce">Accumulators</a> & </td>
|
|
<td class="paramname"><em>accumulators</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1Coord.html">Coord</a>< 3 > const & </td>
|
|
<td class="paramname"><em>block</em> = <code><a class="el" href="namespacecutlass.html#a7419519fa453a121dfa5f26bf87318d9">make_Coord</a>(0, 0, 0)</code>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">int </td>
|
|
<td class="paramname"><em>batch_id</em> = <code>0</code> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</td>
|
|
<td class="mlabels-right">
|
|
<span class="mlabels"><span class="mlabel">inline</span></span> </td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="a7af2eb421840e037263e6a144cca5c32"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a7af2eb421840e037263e6a144cca5c32">◆ </a></span>epilogue_with_or_without_beta()</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmEpilogueTraits_ > </div>
|
|
<div class="memtemplate">
|
|
template<bool kSourceRequired> </div>
|
|
<table class="mlabels">
|
|
<tr>
|
|
<td class="mlabels-left">
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">CUTLASS_DEVICE void <a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html">cutlass::gemm::GemmEpilogue</a>< GemmEpilogueTraits_ >::epilogue_with_or_without_beta </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#afe6bebd94e3379c94054d04c5196edce">Accumulators</a> & </td>
|
|
<td class="paramname"><em>accumulators</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1Coord.html">Coord</a>< 3 > const & </td>
|
|
<td class="paramname"><em>block</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">int </td>
|
|
<td class="paramname"><em>batch_id</em> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</td>
|
|
<td class="mlabels-right">
|
|
<span class="mlabels"><span class="mlabel">inline</span></span> </td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="a9b5e42f222fec98ff479bc1650221b84"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a9b5e42f222fec98ff479bc1650221b84">◆ </a></span>shared_load_fence()</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmEpilogueTraits_ > </div>
|
|
<table class="mlabels">
|
|
<tr>
|
|
<td class="mlabels-left">
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">CUTLASS_DEVICE void <a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html">cutlass::gemm::GemmEpilogue</a>< GemmEpilogueTraits_ >::shared_load_fence </td>
|
|
<td>(</td>
|
|
<td class="paramname"></td><td>)</td>
|
|
<td></td>
|
|
</tr>
|
|
</table>
|
|
</td>
|
|
<td class="mlabels-right">
|
|
<span class="mlabels"><span class="mlabel">inline</span></span> </td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="ac1b2a16b4ccf3e9617faf4d8a2c43691"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#ac1b2a16b4ccf3e9617faf4d8a2c43691">◆ </a></span>shared_store_fence()</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmEpilogueTraits_ > </div>
|
|
<table class="mlabels">
|
|
<tr>
|
|
<td class="mlabels-left">
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">CUTLASS_DEVICE void <a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html">cutlass::gemm::GemmEpilogue</a>< GemmEpilogueTraits_ >::shared_store_fence </td>
|
|
<td>(</td>
|
|
<td class="paramname"></td><td>)</td>
|
|
<td></td>
|
|
</tr>
|
|
</table>
|
|
</td>
|
|
<td class="mlabels-right">
|
|
<span class="mlabels"><span class="mlabel">inline</span></span> </td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<h2 class="groupheader">Member Data Documentation</h2>
|
|
<a id="a5aa254702fcf24630e04ceb174aa3d57"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a5aa254702fcf24630e04ceb174aa3d57">◆ </a></span>functor</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmEpilogueTraits_ > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#a6c30bea1b2a1bd2e981025851d5b12d1">Functor</a> <a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html">cutlass::gemm::GemmEpilogue</a>< GemmEpilogueTraits_ >::functor</td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="a81b028a18df51d3caa1b0ba0c990e362"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a81b028a18df51d3caa1b0ba0c990e362">◆ </a></span>params</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmEpilogueTraits_ > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#ae5209fa80705442693833c63d535161e">Params</a> const& <a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html">cutlass::gemm::GemmEpilogue</a>< GemmEpilogueTraits_ >::params</td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="a6bea57249e3645a4b08a11619cb1687e"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a6bea57249e3645a4b08a11619cb1687e">◆ </a></span>problem_size</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmEpilogueTraits_ > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname"><a class="el" href="structcutlass_1_1Coord.html">Coord</a><3> <a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html">cutlass::gemm::GemmEpilogue</a>< GemmEpilogueTraits_ >::problem_size</td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="a442b5b5688cd658c3b3476650c00281e"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a442b5b5688cd658c3b3476650c00281e">◆ </a></span>shared_storage</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmEpilogueTraits_ > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname"><a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html#ac36dad8a7b6bc7fc6ef88e44068468dc">SharedStorage</a>& <a class="el" href="structcutlass_1_1gemm_1_1GemmEpilogue.html">cutlass::gemm::GemmEpilogue</a>< GemmEpilogueTraits_ >::shared_storage</td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<hr/>The documentation for this struct was generated from the following file:<ul>
|
|
<li><a class="el" href="gemm__epilogue_8h_source.html">gemm_epilogue.h</a></li>
|
|
</ul>
|
|
</div><!-- contents -->
|
|
<!-- start footer part -->
|
|
<hr class="footer"/><address class="footer"><small>
|
|
Generated on Fri Oct 26 2018 14:53:39 for Cutlass by  <a href="http://www.doxygen.org/index.html">
|
|
<img class="footer" src="doxygen.png" alt="doxygen"/>
|
|
</a> 1.8.14
|
|
</small></address>
|
|
</body>
|
|
</html>
|