662 lines
47 KiB
HTML
662 lines
47 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::GemmTraits< GemmConfig_, GlobalLoadStreamA_, GlobalLoadStreamB_, SharedLoadStreamA_, SharedLoadStreamB_, Epilogue_, BlockSwizzle_, Index_, ClearAccumulators_ >::Params Struct 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_1GemmTraits.html">GemmTraits</a></li><li class="navelem"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits_1_1Params.html">Params</a></li> </ul>
|
|
</div>
|
|
</div><!-- top -->
|
|
<div class="header">
|
|
<div class="summary">
|
|
<a href="#pub-methods">Public Member Functions</a> |
|
|
<a href="#pub-attribs">Public Attributes</a> |
|
|
<a href="structcutlass_1_1gemm_1_1GemmTraits_1_1Params-members.html">List of all members</a> </div>
|
|
<div class="headertitle">
|
|
<div class="title">cutlass::gemm::GemmTraits< GemmConfig_, GlobalLoadStreamA_, GlobalLoadStreamB_, SharedLoadStreamA_, SharedLoadStreamB_, Epilogue_, BlockSwizzle_, Index_, ClearAccumulators_ >::Params Struct Reference</div> </div>
|
|
</div><!--header-->
|
|
<div class="contents">
|
|
|
|
<p>Parameters object constructable on the host.
|
|
</p>
|
|
|
|
<p><code>#include <<a class="el" href="gemm__traits_8h_source.html">gemm_traits.h</a>></code></p>
|
|
<div class="dynheader">
|
|
Inheritance diagram for cutlass::gemm::GemmTraits< GemmConfig_, GlobalLoadStreamA_, GlobalLoadStreamB_, SharedLoadStreamA_, SharedLoadStreamB_, Epilogue_, BlockSwizzle_, Index_, ClearAccumulators_ >::Params:</div>
|
|
<div class="dyncontent">
|
|
<div class="center">
|
|
<img src="structcutlass_1_1gemm_1_1GemmTraits_1_1Params.png" usemap="#cutlass::gemm::GemmTraits_3C_20GemmConfig_5F_2C_20GlobalLoadStreamA_5F_2C_20GlobalLoadStreamB_5F_2C_20SharedLoadStreamA_5F_2C_20SharedLoadStreamB_5F_2C_20Epilogue_5F_2C_20BlockSwizzle_5F_2C_20Index_5F_2C_20ClearAccumulators_5F_20_3E::Params_map" alt=""/>
|
|
<map id="cutlass::gemm::GemmTraits_3C_20GemmConfig_5F_2C_20GlobalLoadStreamA_5F_2C_20GlobalLoadStreamB_5F_2C_20SharedLoadStreamA_5F_2C_20SharedLoadStreamB_5F_2C_20Epilogue_5F_2C_20BlockSwizzle_5F_2C_20Index_5F_2C_20ClearAccumulators_5F_20_3E::Params_map" name="cutlass::gemm::GemmTraits_3C_20GemmConfig_5F_2C_20GlobalLoadStreamA_5F_2C_20GlobalLoadStreamB_5F_2C_20SharedLoadStreamA_5F_2C_20SharedLoadStreamB_5F_2C_20Epilogue_5F_2C_20BlockSwizzle_5F_2C_20Index_5F_2C_20ClearAccumulators_5F_20_3E::Params_map">
|
|
<area href="structcutlass_1_1KernelLaunchConfiguration.html" title="Structure containing the basic launch configuration of a CUDA kernel. " alt="cutlass::KernelLaunchConfiguration" shape="rect" coords="0,0,1154,24"/>
|
|
</map>
|
|
</div></div>
|
|
<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:a40023f0ffdd8bee4ccbcaac28222e983"><td class="memTemplParams" colspan="2">template<typename GemmDesc_ > </td></tr>
|
|
<tr class="memitem:a40023f0ffdd8bee4ccbcaac28222e983"><td class="memTemplItemLeft" align="right" valign="top"><a class="el" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a> int </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits_1_1Params.html#a40023f0ffdd8bee4ccbcaac28222e983">initialize</a> (GemmDesc_ const &desc)</td></tr>
|
|
<tr class="memdesc:a40023f0ffdd8bee4ccbcaac28222e983"><td class="mdescLeft"> </td><td class="mdescRight">Initialize the parameters. <a href="#a40023f0ffdd8bee4ccbcaac28222e983">More...</a><br /></td></tr>
|
|
<tr class="separator:a40023f0ffdd8bee4ccbcaac28222e983"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:ac4cd0b74130ffc09e4ccb7b0acba87f8"><td class="memItemLeft" align="right" valign="top"><a class="el" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a> int </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits_1_1Params.html#ac4cd0b74130ffc09e4ccb7b0acba87f8">initialize</a> (<a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> m, <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> n, <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> k, typename Epilogue::Scalar alpha, <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#a96d64bdc48db4971798b620d6b49b3f6">ScalarA</a> const *d_a, <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> lda, <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#aa0e8fd28f5247764dfb7843f7670c698">ScalarB</a> const *d_b, <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> ldb, typename Epilogue::Scalar beta, <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#a8f78d4a68817760099081523aa7fd443">ScalarC</a> const *d_c, <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> ldc, <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#a3129be75ee087603170f8367e10e070e">ScalarD</a> *d_d, <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> ldd)</td></tr>
|
|
<tr class="memdesc:ac4cd0b74130ffc09e4ccb7b0acba87f8"><td class="mdescLeft"> </td><td class="mdescRight">Helper to construct a GEMM params using a BLAS-like API. <a href="#ac4cd0b74130ffc09e4ccb7b0acba87f8">More...</a><br /></td></tr>
|
|
<tr class="separator:ac4cd0b74130ffc09e4ccb7b0acba87f8"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a95314e9a9279c4870f37f68a2357e064"><td class="memItemLeft" align="right" valign="top"><a class="el" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a> int </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits_1_1Params.html#a95314e9a9279c4870f37f68a2357e064">initialize</a> (<a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> m, <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> n, <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> k, typename Epilogue::Scalar alpha, <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#a96d64bdc48db4971798b620d6b49b3f6">ScalarA</a> const *d_a, <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> lda, long long int batch_stride_A, <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#aa0e8fd28f5247764dfb7843f7670c698">ScalarB</a> const *d_b, <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> ldb, long long int batch_stride_B, typename Epilogue::Scalar beta, <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#a8f78d4a68817760099081523aa7fd443">ScalarC</a> const *d_c, <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> ldc, long long int batch_stride_C, <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#a3129be75ee087603170f8367e10e070e">ScalarD</a> *d_d, <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> ldd, long long int batch_stride_D, <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> batch_count)</td></tr>
|
|
<tr class="memdesc:a95314e9a9279c4870f37f68a2357e064"><td class="mdescLeft"> </td><td class="mdescRight">Helper to construct a batched GEMM params. <a href="#a95314e9a9279c4870f37f68a2357e064">More...</a><br /></td></tr>
|
|
<tr class="separator:a95314e9a9279c4870f37f68a2357e064"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a9f0b980cba9d6c5b3df70dd73f455985"><td class="memTemplParams" colspan="2">template<typename GemmDesc_ > </td></tr>
|
|
<tr class="memitem:a9f0b980cba9d6c5b3df70dd73f455985"><td class="memTemplItemLeft" align="right" valign="top"><a class="el" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a> int </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits_1_1Params.html#a9f0b980cba9d6c5b3df70dd73f455985">initialize</a> (GemmDesc_ const &partitonK_desc, <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> partitionK_count_)</td></tr>
|
|
<tr class="memdesc:a9f0b980cba9d6c5b3df70dd73f455985"><td class="mdescLeft"> </td><td class="mdescRight">Helper to construct a partitionedK GEMM params. <a href="#a9f0b980cba9d6c5b3df70dd73f455985">More...</a><br /></td></tr>
|
|
<tr class="separator:a9f0b980cba9d6c5b3df70dd73f455985"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:ae2ac243a087c64c1c5b024a0392bfc22"><td class="memItemLeft" align="right" valign="top"><a class="el" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a> int </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits_1_1Params.html#ae2ac243a087c64c1c5b024a0392bfc22">initialize</a> (<a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> m, <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> n, <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> k, typename Epilogue::Scalar alpha, <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#a96d64bdc48db4971798b620d6b49b3f6">ScalarA</a> const *d_a, <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> lda, <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#aa0e8fd28f5247764dfb7843f7670c698">ScalarB</a> const *d_b, <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> ldb, typename Epilogue::Scalar beta, <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#a8f78d4a68817760099081523aa7fd443">ScalarC</a> const *d_c, <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> ldc, <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#a3129be75ee087603170f8367e10e070e">ScalarD</a> *d_d, <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> ldd, <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> partitionK_count_)</td></tr>
|
|
<tr class="memdesc:ae2ac243a087c64c1c5b024a0392bfc22"><td class="mdescLeft"> </td><td class="mdescRight">Helper to construct a partitionedK GEMM params. <a href="#ae2ac243a087c64c1c5b024a0392bfc22">More...</a><br /></td></tr>
|
|
<tr class="separator:ae2ac243a087c64c1c5b024a0392bfc22"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="inherit_header pub_methods_structcutlass_1_1KernelLaunchConfiguration"><td colspan="2" onclick="javascript:toggleInherit('pub_methods_structcutlass_1_1KernelLaunchConfiguration')"><img src="closed.png" alt="-"/> Public Member Functions inherited from <a class="el" href="structcutlass_1_1KernelLaunchConfiguration.html">cutlass::KernelLaunchConfiguration</a></td></tr>
|
|
<tr class="memitem:a726db328ccc8f5e186ff8e7cef568eaa inherit pub_methods_structcutlass_1_1KernelLaunchConfiguration"><td class="memItemLeft" align="right" valign="top"><a class="el" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a> </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1KernelLaunchConfiguration.html#a726db328ccc8f5e186ff8e7cef568eaa">KernelLaunchConfiguration</a> (dim3 _grid=dim3(1, 1, 1), dim3 _block=dim3(1, 1, 1), size_t _dynamic_smem=0)</td></tr>
|
|
<tr class="memdesc:a726db328ccc8f5e186ff8e7cef568eaa inherit pub_methods_structcutlass_1_1KernelLaunchConfiguration"><td class="mdescLeft"> </td><td class="mdescRight">Constructs a KernellaunchConfiguration object. <a href="structcutlass_1_1KernelLaunchConfiguration.html#a726db328ccc8f5e186ff8e7cef568eaa">More...</a><br /></td></tr>
|
|
<tr class="separator:a726db328ccc8f5e186ff8e7cef568eaa inherit pub_methods_structcutlass_1_1KernelLaunchConfiguration"><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:af2233a6ebf39788e27f051f8c614ab90"><td class="memItemLeft" align="right" valign="top"><a class="el" href="structcutlass_1_1gemm_1_1GemmCoord.html">GemmCoord</a> </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits_1_1Params.html#af2233a6ebf39788e27f051f8c614ab90">problem_size</a></td></tr>
|
|
<tr class="memdesc:af2233a6ebf39788e27f051f8c614ab90"><td class="mdescLeft"> </td><td class="mdescRight">GEMM problem size. <a href="#af2233a6ebf39788e27f051f8c614ab90">More...</a><br /></td></tr>
|
|
<tr class="separator:af2233a6ebf39788e27f051f8c614ab90"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a44b9a0d6210561d3b0d538b2c0a38886"><td class="memItemLeft" align="right" valign="top">int </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits_1_1Params.html#a44b9a0d6210561d3b0d538b2c0a38886">partitionK_range</a></td></tr>
|
|
<tr class="memdesc:a44b9a0d6210561d3b0d538b2c0a38886"><td class="mdescLeft"> </td><td class="mdescRight">The K range for every partition except the last one. <a href="#a44b9a0d6210561d3b0d538b2c0a38886">More...</a><br /></td></tr>
|
|
<tr class="separator:a44b9a0d6210561d3b0d538b2c0a38886"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a8dba1bcd9ddab830bc121afc728296c3"><td class="memItemLeft" align="right" valign="top"><a class="el" href="structcutlass_1_1gemm_1_1GlobalLoadStreamPair_1_1Params.html">GlobalLoadStream::Params</a> </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits_1_1Params.html#a8dba1bcd9ddab830bc121afc728296c3">global_to_shared_stream</a></td></tr>
|
|
<tr class="memdesc:a8dba1bcd9ddab830bc121afc728296c3"><td class="mdescLeft"> </td><td class="mdescRight">Parameters object for the global load stream. <a href="#a8dba1bcd9ddab830bc121afc728296c3">More...</a><br /></td></tr>
|
|
<tr class="separator:a8dba1bcd9ddab830bc121afc728296c3"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a50e9cc382a32abd5beab299a79b30b27"><td class="memItemLeft" align="right" valign="top"><a class="el" href="structcutlass_1_1gemm_1_1SharedStreamPair_1_1Params.html">SharedStream::Params</a> </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits_1_1Params.html#a50e9cc382a32abd5beab299a79b30b27">shared_stream</a></td></tr>
|
|
<tr class="memdesc:a50e9cc382a32abd5beab299a79b30b27"><td class="mdescLeft"> </td><td class="mdescRight">Parameters object for the shared load stream. <a href="#a50e9cc382a32abd5beab299a79b30b27">More...</a><br /></td></tr>
|
|
<tr class="separator:a50e9cc382a32abd5beab299a79b30b27"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a073430a1e8b124aec8a1f1e00f262bc8"><td class="memItemLeft" align="right" valign="top">Epilogue::Params </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits_1_1Params.html#a073430a1e8b124aec8a1f1e00f262bc8">epilogue</a></td></tr>
|
|
<tr class="memdesc:a073430a1e8b124aec8a1f1e00f262bc8"><td class="mdescLeft"> </td><td class="mdescRight">The params for the epilogue. <a href="#a073430a1e8b124aec8a1f1e00f262bc8">More...</a><br /></td></tr>
|
|
<tr class="separator:a073430a1e8b124aec8a1f1e00f262bc8"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="inherit_header pub_attribs_structcutlass_1_1KernelLaunchConfiguration"><td colspan="2" onclick="javascript:toggleInherit('pub_attribs_structcutlass_1_1KernelLaunchConfiguration')"><img src="closed.png" alt="-"/> Public Attributes inherited from <a class="el" href="structcutlass_1_1KernelLaunchConfiguration.html">cutlass::KernelLaunchConfiguration</a></td></tr>
|
|
<tr class="memitem:ab86ba1464dd9c6cd15ae0de4a552201b inherit pub_attribs_structcutlass_1_1KernelLaunchConfiguration"><td class="memItemLeft" align="right" valign="top">dim3 </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1KernelLaunchConfiguration.html#ab86ba1464dd9c6cd15ae0de4a552201b">grid</a></td></tr>
|
|
<tr class="memdesc:ab86ba1464dd9c6cd15ae0de4a552201b inherit pub_attribs_structcutlass_1_1KernelLaunchConfiguration"><td class="mdescLeft"> </td><td class="mdescRight">CUDA grid dimensions. <a href="structcutlass_1_1KernelLaunchConfiguration.html#ab86ba1464dd9c6cd15ae0de4a552201b">More...</a><br /></td></tr>
|
|
<tr class="separator:ab86ba1464dd9c6cd15ae0de4a552201b inherit pub_attribs_structcutlass_1_1KernelLaunchConfiguration"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a09535026bf08f94c6940c358d95d1edd inherit pub_attribs_structcutlass_1_1KernelLaunchConfiguration"><td class="memItemLeft" align="right" valign="top">dim3 </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1KernelLaunchConfiguration.html#a09535026bf08f94c6940c358d95d1edd">block</a></td></tr>
|
|
<tr class="memdesc:a09535026bf08f94c6940c358d95d1edd inherit pub_attribs_structcutlass_1_1KernelLaunchConfiguration"><td class="mdescLeft"> </td><td class="mdescRight">CUDA threablock dimensions. <a href="structcutlass_1_1KernelLaunchConfiguration.html#a09535026bf08f94c6940c358d95d1edd">More...</a><br /></td></tr>
|
|
<tr class="separator:a09535026bf08f94c6940c358d95d1edd inherit pub_attribs_structcutlass_1_1KernelLaunchConfiguration"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a4a6ac693d4284c84301279219623e2bc inherit pub_attribs_structcutlass_1_1KernelLaunchConfiguration"><td class="memItemLeft" align="right" valign="top">size_t </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1KernelLaunchConfiguration.html#a4a6ac693d4284c84301279219623e2bc">dynamic_smem</a></td></tr>
|
|
<tr class="memdesc:a4a6ac693d4284c84301279219623e2bc inherit pub_attribs_structcutlass_1_1KernelLaunchConfiguration"><td class="mdescLeft"> </td><td class="mdescRight">Bytes of dynamically allocated SMEM in addition to static SMEM. <a href="structcutlass_1_1KernelLaunchConfiguration.html#a4a6ac693d4284c84301279219623e2bc">More...</a><br /></td></tr>
|
|
<tr class="separator:a4a6ac693d4284c84301279219623e2bc inherit pub_attribs_structcutlass_1_1KernelLaunchConfiguration"><td class="memSeparator" colspan="2"> </td></tr>
|
|
</table>
|
|
<h2 class="groupheader">Member Function Documentation</h2>
|
|
<a id="a40023f0ffdd8bee4ccbcaac28222e983"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a40023f0ffdd8bee4ccbcaac28222e983">◆ </a></span>initialize() <span class="overload">[1/5]</span></h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmConfig_, typename GlobalLoadStreamA_, typename GlobalLoadStreamB_, typename SharedLoadStreamA_, typename SharedLoadStreamB_, typename Epilogue_, typename BlockSwizzle_ = IdentityBlockSwizzle, typename Index_ = int, typename ClearAccumulators_ = ClearAccumulators<typename GemmConfig_::Accumulators::Element>> </div>
|
|
<div class="memtemplate">
|
|
template<typename GemmDesc_ > </div>
|
|
<table class="mlabels">
|
|
<tr>
|
|
<td class="mlabels-left">
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname"><a class="el" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a> int <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html">cutlass::gemm::GemmTraits</a>< GemmConfig_, GlobalLoadStreamA_, GlobalLoadStreamB_, SharedLoadStreamA_, SharedLoadStreamB_, Epilogue_, BlockSwizzle_, Index_, ClearAccumulators_ >::Params::initialize </td>
|
|
<td>(</td>
|
|
<td class="paramtype">GemmDesc_ const & </td>
|
|
<td class="paramname"><em>desc</em></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="ac4cd0b74130ffc09e4ccb7b0acba87f8"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#ac4cd0b74130ffc09e4ccb7b0acba87f8">◆ </a></span>initialize() <span class="overload">[2/5]</span></h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmConfig_, typename GlobalLoadStreamA_, typename GlobalLoadStreamB_, typename SharedLoadStreamA_, typename SharedLoadStreamB_, typename Epilogue_, typename BlockSwizzle_ = IdentityBlockSwizzle, typename Index_ = int, typename ClearAccumulators_ = ClearAccumulators<typename GemmConfig_::Accumulators::Element>> </div>
|
|
<table class="mlabels">
|
|
<tr>
|
|
<td class="mlabels-left">
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname"><a class="el" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a> int <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html">cutlass::gemm::GemmTraits</a>< GemmConfig_, GlobalLoadStreamA_, GlobalLoadStreamB_, SharedLoadStreamA_, SharedLoadStreamB_, Epilogue_, BlockSwizzle_, Index_, ClearAccumulators_ >::Params::initialize </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> </td>
|
|
<td class="paramname"><em>m</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> </td>
|
|
<td class="paramname"><em>n</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> </td>
|
|
<td class="paramname"><em>k</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">typename Epilogue::Scalar </td>
|
|
<td class="paramname"><em>alpha</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#a96d64bdc48db4971798b620d6b49b3f6">ScalarA</a> const * </td>
|
|
<td class="paramname"><em>d_a</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> </td>
|
|
<td class="paramname"><em>lda</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#aa0e8fd28f5247764dfb7843f7670c698">ScalarB</a> const * </td>
|
|
<td class="paramname"><em>d_b</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> </td>
|
|
<td class="paramname"><em>ldb</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">typename Epilogue::Scalar </td>
|
|
<td class="paramname"><em>beta</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#a8f78d4a68817760099081523aa7fd443">ScalarC</a> const * </td>
|
|
<td class="paramname"><em>d_c</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> </td>
|
|
<td class="paramname"><em>ldc</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#a3129be75ee087603170f8367e10e070e">ScalarD</a> * </td>
|
|
<td class="paramname"><em>d_d</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> </td>
|
|
<td class="paramname"><em>ldd</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="a95314e9a9279c4870f37f68a2357e064"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a95314e9a9279c4870f37f68a2357e064">◆ </a></span>initialize() <span class="overload">[3/5]</span></h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmConfig_, typename GlobalLoadStreamA_, typename GlobalLoadStreamB_, typename SharedLoadStreamA_, typename SharedLoadStreamB_, typename Epilogue_, typename BlockSwizzle_ = IdentityBlockSwizzle, typename Index_ = int, typename ClearAccumulators_ = ClearAccumulators<typename GemmConfig_::Accumulators::Element>> </div>
|
|
<table class="mlabels">
|
|
<tr>
|
|
<td class="mlabels-left">
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname"><a class="el" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a> int <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html">cutlass::gemm::GemmTraits</a>< GemmConfig_, GlobalLoadStreamA_, GlobalLoadStreamB_, SharedLoadStreamA_, SharedLoadStreamB_, Epilogue_, BlockSwizzle_, Index_, ClearAccumulators_ >::Params::initialize </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> </td>
|
|
<td class="paramname"><em>m</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> </td>
|
|
<td class="paramname"><em>n</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> </td>
|
|
<td class="paramname"><em>k</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">typename Epilogue::Scalar </td>
|
|
<td class="paramname"><em>alpha</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#a96d64bdc48db4971798b620d6b49b3f6">ScalarA</a> const * </td>
|
|
<td class="paramname"><em>d_a</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> </td>
|
|
<td class="paramname"><em>lda</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">long long int </td>
|
|
<td class="paramname"><em>batch_stride_A</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#aa0e8fd28f5247764dfb7843f7670c698">ScalarB</a> const * </td>
|
|
<td class="paramname"><em>d_b</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> </td>
|
|
<td class="paramname"><em>ldb</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">long long int </td>
|
|
<td class="paramname"><em>batch_stride_B</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">typename Epilogue::Scalar </td>
|
|
<td class="paramname"><em>beta</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#a8f78d4a68817760099081523aa7fd443">ScalarC</a> const * </td>
|
|
<td class="paramname"><em>d_c</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> </td>
|
|
<td class="paramname"><em>ldc</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">long long int </td>
|
|
<td class="paramname"><em>batch_stride_C</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#a3129be75ee087603170f8367e10e070e">ScalarD</a> * </td>
|
|
<td class="paramname"><em>d_d</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> </td>
|
|
<td class="paramname"><em>ldd</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">long long int </td>
|
|
<td class="paramname"><em>batch_stride_D</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> </td>
|
|
<td class="paramname"><em>batch_count</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="a9f0b980cba9d6c5b3df70dd73f455985"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a9f0b980cba9d6c5b3df70dd73f455985">◆ </a></span>initialize() <span class="overload">[4/5]</span></h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmConfig_, typename GlobalLoadStreamA_, typename GlobalLoadStreamB_, typename SharedLoadStreamA_, typename SharedLoadStreamB_, typename Epilogue_, typename BlockSwizzle_ = IdentityBlockSwizzle, typename Index_ = int, typename ClearAccumulators_ = ClearAccumulators<typename GemmConfig_::Accumulators::Element>> </div>
|
|
<div class="memtemplate">
|
|
template<typename GemmDesc_ > </div>
|
|
<table class="mlabels">
|
|
<tr>
|
|
<td class="mlabels-left">
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname"><a class="el" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a> int <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html">cutlass::gemm::GemmTraits</a>< GemmConfig_, GlobalLoadStreamA_, GlobalLoadStreamB_, SharedLoadStreamA_, SharedLoadStreamB_, Epilogue_, BlockSwizzle_, Index_, ClearAccumulators_ >::Params::initialize </td>
|
|
<td>(</td>
|
|
<td class="paramtype">GemmDesc_ const & </td>
|
|
<td class="paramname"><em>partitonK_desc</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> </td>
|
|
<td class="paramname"><em>partitionK_count_</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="ae2ac243a087c64c1c5b024a0392bfc22"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#ae2ac243a087c64c1c5b024a0392bfc22">◆ </a></span>initialize() <span class="overload">[5/5]</span></h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmConfig_, typename GlobalLoadStreamA_, typename GlobalLoadStreamB_, typename SharedLoadStreamA_, typename SharedLoadStreamB_, typename Epilogue_, typename BlockSwizzle_ = IdentityBlockSwizzle, typename Index_ = int, typename ClearAccumulators_ = ClearAccumulators<typename GemmConfig_::Accumulators::Element>> </div>
|
|
<table class="mlabels">
|
|
<tr>
|
|
<td class="mlabels-left">
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname"><a class="el" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a> int <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html">cutlass::gemm::GemmTraits</a>< GemmConfig_, GlobalLoadStreamA_, GlobalLoadStreamB_, SharedLoadStreamA_, SharedLoadStreamB_, Epilogue_, BlockSwizzle_, Index_, ClearAccumulators_ >::Params::initialize </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> </td>
|
|
<td class="paramname"><em>m</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> </td>
|
|
<td class="paramname"><em>n</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> </td>
|
|
<td class="paramname"><em>k</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">typename Epilogue::Scalar </td>
|
|
<td class="paramname"><em>alpha</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#a96d64bdc48db4971798b620d6b49b3f6">ScalarA</a> const * </td>
|
|
<td class="paramname"><em>d_a</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> </td>
|
|
<td class="paramname"><em>lda</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#aa0e8fd28f5247764dfb7843f7670c698">ScalarB</a> const * </td>
|
|
<td class="paramname"><em>d_b</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> </td>
|
|
<td class="paramname"><em>ldb</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">typename Epilogue::Scalar </td>
|
|
<td class="paramname"><em>beta</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#a8f78d4a68817760099081523aa7fd443">ScalarC</a> const * </td>
|
|
<td class="paramname"><em>d_c</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> </td>
|
|
<td class="paramname"><em>ldc</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#a3129be75ee087603170f8367e10e070e">ScalarD</a> * </td>
|
|
<td class="paramname"><em>d_d</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> </td>
|
|
<td class="paramname"><em>ldd</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html#ae67227cecbe84f5c8497d9a7ff82b367">Index</a> </td>
|
|
<td class="paramname"><em>partitionK_count_</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 Data Documentation</h2>
|
|
<a id="a073430a1e8b124aec8a1f1e00f262bc8"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a073430a1e8b124aec8a1f1e00f262bc8">◆ </a></span>epilogue</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmConfig_, typename GlobalLoadStreamA_, typename GlobalLoadStreamB_, typename SharedLoadStreamA_, typename SharedLoadStreamB_, typename Epilogue_, typename BlockSwizzle_ = IdentityBlockSwizzle, typename Index_ = int, typename ClearAccumulators_ = ClearAccumulators<typename GemmConfig_::Accumulators::Element>> </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">Epilogue::Params <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html">cutlass::gemm::GemmTraits</a>< GemmConfig_, GlobalLoadStreamA_, GlobalLoadStreamB_, SharedLoadStreamA_, SharedLoadStreamB_, Epilogue_, BlockSwizzle_, Index_, ClearAccumulators_ >::Params::epilogue</td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="a8dba1bcd9ddab830bc121afc728296c3"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a8dba1bcd9ddab830bc121afc728296c3">◆ </a></span>global_to_shared_stream</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmConfig_, typename GlobalLoadStreamA_, typename GlobalLoadStreamB_, typename SharedLoadStreamA_, typename SharedLoadStreamB_, typename Epilogue_, typename BlockSwizzle_ = IdentityBlockSwizzle, typename Index_ = int, typename ClearAccumulators_ = ClearAccumulators<typename GemmConfig_::Accumulators::Element>> </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname"><a class="el" href="structcutlass_1_1gemm_1_1GlobalLoadStreamPair_1_1Params.html">GlobalLoadStream::Params</a> <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html">cutlass::gemm::GemmTraits</a>< GemmConfig_, GlobalLoadStreamA_, GlobalLoadStreamB_, SharedLoadStreamA_, SharedLoadStreamB_, Epilogue_, BlockSwizzle_, Index_, ClearAccumulators_ >::Params::global_to_shared_stream</td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="a44b9a0d6210561d3b0d538b2c0a38886"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a44b9a0d6210561d3b0d538b2c0a38886">◆ </a></span>partitionK_range</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmConfig_, typename GlobalLoadStreamA_, typename GlobalLoadStreamB_, typename SharedLoadStreamA_, typename SharedLoadStreamB_, typename Epilogue_, typename BlockSwizzle_ = IdentityBlockSwizzle, typename Index_ = int, typename ClearAccumulators_ = ClearAccumulators<typename GemmConfig_::Accumulators::Element>> </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">int <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html">cutlass::gemm::GemmTraits</a>< GemmConfig_, GlobalLoadStreamA_, GlobalLoadStreamB_, SharedLoadStreamA_, SharedLoadStreamB_, Epilogue_, BlockSwizzle_, Index_, ClearAccumulators_ >::Params::partitionK_range</td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="af2233a6ebf39788e27f051f8c614ab90"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#af2233a6ebf39788e27f051f8c614ab90">◆ </a></span>problem_size</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmConfig_, typename GlobalLoadStreamA_, typename GlobalLoadStreamB_, typename SharedLoadStreamA_, typename SharedLoadStreamB_, typename Epilogue_, typename BlockSwizzle_ = IdentityBlockSwizzle, typename Index_ = int, typename ClearAccumulators_ = ClearAccumulators<typename GemmConfig_::Accumulators::Element>> </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname"><a class="el" href="structcutlass_1_1gemm_1_1GemmCoord.html">GemmCoord</a> <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html">cutlass::gemm::GemmTraits</a>< GemmConfig_, GlobalLoadStreamA_, GlobalLoadStreamB_, SharedLoadStreamA_, SharedLoadStreamB_, Epilogue_, BlockSwizzle_, Index_, ClearAccumulators_ >::Params::problem_size</td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="a50e9cc382a32abd5beab299a79b30b27"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a50e9cc382a32abd5beab299a79b30b27">◆ </a></span>shared_stream</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmConfig_, typename GlobalLoadStreamA_, typename GlobalLoadStreamB_, typename SharedLoadStreamA_, typename SharedLoadStreamB_, typename Epilogue_, typename BlockSwizzle_ = IdentityBlockSwizzle, typename Index_ = int, typename ClearAccumulators_ = ClearAccumulators<typename GemmConfig_::Accumulators::Element>> </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname"><a class="el" href="structcutlass_1_1gemm_1_1SharedStreamPair_1_1Params.html">SharedStream::Params</a> <a class="el" href="structcutlass_1_1gemm_1_1GemmTraits.html">cutlass::gemm::GemmTraits</a>< GemmConfig_, GlobalLoadStreamA_, GlobalLoadStreamB_, SharedLoadStreamA_, SharedLoadStreamB_, Epilogue_, BlockSwizzle_, Index_, ClearAccumulators_ >::Params::shared_stream</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__traits_8h_source.html">gemm_traits.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>
|