467 lines
27 KiB
HTML
467 lines
27 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::SplitkPIGemmTraits< GemmTraits_, ReductionTraits_ >::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_1SplitkPIGemmTraits.html">SplitkPIGemmTraits</a></li><li class="navelem"><a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits_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_1SplitkPIGemmTraits_1_1Params-members.html">List of all members</a> </div>
|
|
<div class="headertitle">
|
|
<div class="title">cutlass::gemm::SplitkPIGemmTraits< GemmTraits_, ReductionTraits_ >::Params Struct Reference</div> </div>
|
|
</div><!--header-->
|
|
<div class="contents">
|
|
|
|
<p><code>#include <<a class="el" href="device__gemm__traits_8h_source.html">device_gemm_traits.h</a>></code></p>
|
|
<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:a9cf3f82207f65e6178d4c3764577b657"><td class="memItemLeft" align="right" valign="top"> </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits_1_1Params.html#a9cf3f82207f65e6178d4c3764577b657">Params</a> ()</td></tr>
|
|
<tr class="memdesc:a9cf3f82207f65e6178d4c3764577b657"><td class="mdescLeft"> </td><td class="mdescRight">ctor <a href="#a9cf3f82207f65e6178d4c3764577b657">More...</a><br /></td></tr>
|
|
<tr class="separator:a9cf3f82207f65e6178d4c3764577b657"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:ae6143cae5ebb1ee8804a55b34f3912b1"><td class="memItemLeft" align="right" valign="top"> </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits_1_1Params.html#ae6143cae5ebb1ee8804a55b34f3912b1">Params</a> (<a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits.html#a3037291a463c6daa06e9a3ef2120f9c7">Index</a> m_, <a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits.html#a3037291a463c6daa06e9a3ef2120f9c7">Index</a> n_, <a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits.html#a3037291a463c6daa06e9a3ef2120f9c7">Index</a> k_)</td></tr>
|
|
<tr class="memdesc:ae6143cae5ebb1ee8804a55b34f3912b1"><td class="mdescLeft"> </td><td class="mdescRight">ctor <a href="#ae6143cae5ebb1ee8804a55b34f3912b1">More...</a><br /></td></tr>
|
|
<tr class="separator:ae6143cae5ebb1ee8804a55b34f3912b1"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:ae447cd662efb8099f040fb0dffc4f104"><td class="memItemLeft" align="right" valign="top">void </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits_1_1Params.html#ae447cd662efb8099f040fb0dffc4f104">init_problem</a> (<a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits.html#a3037291a463c6daa06e9a3ef2120f9c7">Index</a> m_, <a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits.html#a3037291a463c6daa06e9a3ef2120f9c7">Index</a> n_, <a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits.html#a3037291a463c6daa06e9a3ef2120f9c7">Index</a> k_)</td></tr>
|
|
<tr class="memdesc:ae447cd662efb8099f040fb0dffc4f104"><td class="mdescLeft"> </td><td class="mdescRight">init problem is needed if using default ctor <a href="#ae447cd662efb8099f040fb0dffc4f104">More...</a><br /></td></tr>
|
|
<tr class="separator:ae447cd662efb8099f040fb0dffc4f104"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:aec998017d6046436cd4e4c933a2a89a6"><td class="memItemLeft" align="right" valign="top">int </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits_1_1Params.html#aec998017d6046436cd4e4c933a2a89a6">initialize</a> (<a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits.html#a98dd6bfe225412dc86f850b07bf8f232">Scalar</a> alpha_, <a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits.html#a79afe0be2f0ac59c4e755784651aa332">ScalarA</a> const *d_a_, <a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits.html#a3037291a463c6daa06e9a3ef2120f9c7">Index</a> lda_, <a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits.html#a748e224a8789694e4f6bfc36a1262103">ScalarB</a> const *d_b_, <a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits.html#a3037291a463c6daa06e9a3ef2120f9c7">Index</a> ldb_, <a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits.html#a98dd6bfe225412dc86f850b07bf8f232">Scalar</a> beta_, <a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits.html#aa7f85820b1392f80157e14bbbb639689">ScalarC</a> const *d_c_, <a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits.html#a3037291a463c6daa06e9a3ef2120f9c7">Index</a> ldc_, <a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits.html#a36d747ce477872482920402f00b604be">ScalarD</a> *d_d_, <a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits.html#a3037291a463c6daa06e9a3ef2120f9c7">Index</a> ldd_, <a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits.html#a8cffb7f3d493d56bc2bfc113db29bad1">ScalarAccum</a> *workspace_ptr_)</td></tr>
|
|
<tr class="separator:aec998017d6046436cd4e4c933a2a89a6"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a6fe1868939c6aeba97d232ac693a5264"><td class="memItemLeft" align="right" valign="top">int </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits_1_1Params.html#a6fe1868939c6aeba97d232ac693a5264">required_workspace_memory_in_byte</a> ()</td></tr>
|
|
<tr class="separator:a6fe1868939c6aeba97d232ac693a5264"><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:a542a371bf39a9e58889d78e782af3aae"><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_1SplitkPIGemmTraits_1_1Params.html#a542a371bf39a9e58889d78e782af3aae">problem_size</a></td></tr>
|
|
<tr class="memdesc:a542a371bf39a9e58889d78e782af3aae"><td class="mdescLeft"> </td><td class="mdescRight">The dimensions of the GEMM in K, N, M order. <a href="#a542a371bf39a9e58889d78e782af3aae">More...</a><br /></td></tr>
|
|
<tr class="separator:a542a371bf39a9e58889d78e782af3aae"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a966f73ad90754f53eed5faf68fc77d87"><td class="memItemLeft" align="right" valign="top">bool </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits_1_1Params.html#a966f73ad90754f53eed5faf68fc77d87">problem_size_initialized</a></td></tr>
|
|
<tr class="memdesc:a966f73ad90754f53eed5faf68fc77d87"><td class="mdescLeft"> </td><td class="mdescRight">Check if params are init. <a href="#a966f73ad90754f53eed5faf68fc77d87">More...</a><br /></td></tr>
|
|
<tr class="separator:a966f73ad90754f53eed5faf68fc77d87"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a38de876c29aaccc45ec8d194814b102a"><td class="memItemLeft" align="right" valign="top"><a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits.html#a8cffb7f3d493d56bc2bfc113db29bad1">ScalarAccum</a> * </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits_1_1Params.html#a38de876c29aaccc45ec8d194814b102a">workspace_ptr</a></td></tr>
|
|
<tr class="memdesc:a38de876c29aaccc45ec8d194814b102a"><td class="mdescLeft"> </td><td class="mdescRight">The pointer to workspace memory. <a href="#a38de876c29aaccc45ec8d194814b102a">More...</a><br /></td></tr>
|
|
<tr class="separator:a38de876c29aaccc45ec8d194814b102a"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:aff11608efe9ced3a4862ce78399aa86e"><td class="memItemLeft" align="right" valign="top">int </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits_1_1Params.html#aff11608efe9ced3a4862ce78399aa86e">workspace_size</a></td></tr>
|
|
<tr class="separator:aff11608efe9ced3a4862ce78399aa86e"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a52264bdfeb4d95313a8dbda81e87ede0"><td class="memItemLeft" align="right" valign="top"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits_1_1Params.html">GemmTraits::Params</a> </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits_1_1Params.html#a52264bdfeb4d95313a8dbda81e87ede0">GemmParams</a></td></tr>
|
|
<tr class="memdesc:a52264bdfeb4d95313a8dbda81e87ede0"><td class="mdescLeft"> </td><td class="mdescRight">The <a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits_1_1Params.html">Params</a> for the first kernel. <a href="#a52264bdfeb4d95313a8dbda81e87ede0">More...</a><br /></td></tr>
|
|
<tr class="separator:a52264bdfeb4d95313a8dbda81e87ede0"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a62adba1d7d9c309049c636f3e0379527"><td class="memItemLeft" align="right" valign="top">ReductionTraits::Params </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits_1_1Params.html#a62adba1d7d9c309049c636f3e0379527">ReductionParams</a></td></tr>
|
|
<tr class="memdesc:a62adba1d7d9c309049c636f3e0379527"><td class="mdescLeft"> </td><td class="mdescRight">The <a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits_1_1Params.html">Params</a> for the second kernel. <a href="#a62adba1d7d9c309049c636f3e0379527">More...</a><br /></td></tr>
|
|
<tr class="separator:a62adba1d7d9c309049c636f3e0379527"><td class="memSeparator" colspan="2"> </td></tr>
|
|
</table>
|
|
<h2 class="groupheader">Constructor & Destructor Documentation</h2>
|
|
<a id="a9cf3f82207f65e6178d4c3764577b657"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a9cf3f82207f65e6178d4c3764577b657">◆ </a></span>Params() <span class="overload">[1/2]</span></h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmTraits_ , typename ReductionTraits_ > </div>
|
|
<table class="mlabels">
|
|
<tr>
|
|
<td class="mlabels-left">
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname"><a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits.html">cutlass::gemm::SplitkPIGemmTraits</a>< GemmTraits_, ReductionTraits_ >::Params::Params </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="ae6143cae5ebb1ee8804a55b34f3912b1"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#ae6143cae5ebb1ee8804a55b34f3912b1">◆ </a></span>Params() <span class="overload">[2/2]</span></h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmTraits_ , typename ReductionTraits_ > </div>
|
|
<table class="mlabels">
|
|
<tr>
|
|
<td class="mlabels-left">
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname"><a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits.html">cutlass::gemm::SplitkPIGemmTraits</a>< GemmTraits_, ReductionTraits_ >::Params::Params </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits.html#a3037291a463c6daa06e9a3ef2120f9c7">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_1SplitkPIGemmTraits.html#a3037291a463c6daa06e9a3ef2120f9c7">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_1SplitkPIGemmTraits.html#a3037291a463c6daa06e9a3ef2120f9c7">Index</a> </td>
|
|
<td class="paramname"><em>k_</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="ae447cd662efb8099f040fb0dffc4f104"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#ae447cd662efb8099f040fb0dffc4f104">◆ </a></span>init_problem()</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmTraits_ , typename ReductionTraits_ > </div>
|
|
<table class="mlabels">
|
|
<tr>
|
|
<td class="mlabels-left">
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void <a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits.html">cutlass::gemm::SplitkPIGemmTraits</a>< GemmTraits_, ReductionTraits_ >::Params::init_problem </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits.html#a3037291a463c6daa06e9a3ef2120f9c7">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_1SplitkPIGemmTraits.html#a3037291a463c6daa06e9a3ef2120f9c7">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_1SplitkPIGemmTraits.html#a3037291a463c6daa06e9a3ef2120f9c7">Index</a> </td>
|
|
<td class="paramname"><em>k_</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="aec998017d6046436cd4e4c933a2a89a6"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#aec998017d6046436cd4e4c933a2a89a6">◆ </a></span>initialize()</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmTraits_ , typename ReductionTraits_ > </div>
|
|
<table class="mlabels">
|
|
<tr>
|
|
<td class="mlabels-left">
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">int <a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits.html">cutlass::gemm::SplitkPIGemmTraits</a>< GemmTraits_, ReductionTraits_ >::Params::initialize </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits.html#a98dd6bfe225412dc86f850b07bf8f232">Scalar</a> </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_1SplitkPIGemmTraits.html#a79afe0be2f0ac59c4e755784651aa332">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_1SplitkPIGemmTraits.html#a3037291a463c6daa06e9a3ef2120f9c7">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_1SplitkPIGemmTraits.html#a748e224a8789694e4f6bfc36a1262103">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_1SplitkPIGemmTraits.html#a3037291a463c6daa06e9a3ef2120f9c7">Index</a> </td>
|
|
<td class="paramname"><em>ldb_</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits.html#a98dd6bfe225412dc86f850b07bf8f232">Scalar</a> </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_1SplitkPIGemmTraits.html#aa7f85820b1392f80157e14bbbb639689">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_1SplitkPIGemmTraits.html#a3037291a463c6daa06e9a3ef2120f9c7">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_1SplitkPIGemmTraits.html#a36d747ce477872482920402f00b604be">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_1SplitkPIGemmTraits.html#a3037291a463c6daa06e9a3ef2120f9c7">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_1SplitkPIGemmTraits.html#a8cffb7f3d493d56bc2bfc113db29bad1">ScalarAccum</a> * </td>
|
|
<td class="paramname"><em>workspace_ptr_</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="a6fe1868939c6aeba97d232ac693a5264"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a6fe1868939c6aeba97d232ac693a5264">◆ </a></span>required_workspace_memory_in_byte()</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmTraits_ , typename ReductionTraits_ > </div>
|
|
<table class="mlabels">
|
|
<tr>
|
|
<td class="mlabels-left">
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">int <a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits.html">cutlass::gemm::SplitkPIGemmTraits</a>< GemmTraits_, ReductionTraits_ >::Params::required_workspace_memory_in_byte </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="a52264bdfeb4d95313a8dbda81e87ede0"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a52264bdfeb4d95313a8dbda81e87ede0">◆ </a></span>GemmParams</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmTraits_ , typename ReductionTraits_ > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname"><a class="el" href="structcutlass_1_1gemm_1_1GemmTraits_1_1Params.html">GemmTraits::Params</a> <a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits.html">cutlass::gemm::SplitkPIGemmTraits</a>< GemmTraits_, ReductionTraits_ >::Params::GemmParams</td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="a542a371bf39a9e58889d78e782af3aae"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a542a371bf39a9e58889d78e782af3aae">◆ </a></span>problem_size</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmTraits_ , typename ReductionTraits_ > </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_1SplitkPIGemmTraits.html">cutlass::gemm::SplitkPIGemmTraits</a>< GemmTraits_, ReductionTraits_ >::Params::problem_size</td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="a966f73ad90754f53eed5faf68fc77d87"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a966f73ad90754f53eed5faf68fc77d87">◆ </a></span>problem_size_initialized</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmTraits_ , typename ReductionTraits_ > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">bool <a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits.html">cutlass::gemm::SplitkPIGemmTraits</a>< GemmTraits_, ReductionTraits_ >::Params::problem_size_initialized</td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="a62adba1d7d9c309049c636f3e0379527"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a62adba1d7d9c309049c636f3e0379527">◆ </a></span>ReductionParams</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmTraits_ , typename ReductionTraits_ > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">ReductionTraits::Params <a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits.html">cutlass::gemm::SplitkPIGemmTraits</a>< GemmTraits_, ReductionTraits_ >::Params::ReductionParams</td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="a38de876c29aaccc45ec8d194814b102a"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a38de876c29aaccc45ec8d194814b102a">◆ </a></span>workspace_ptr</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmTraits_ , typename ReductionTraits_ > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname"><a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits.html#a8cffb7f3d493d56bc2bfc113db29bad1">ScalarAccum</a>* <a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits.html">cutlass::gemm::SplitkPIGemmTraits</a>< GemmTraits_, ReductionTraits_ >::Params::workspace_ptr</td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="aff11608efe9ced3a4862ce78399aa86e"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#aff11608efe9ced3a4862ce78399aa86e">◆ </a></span>workspace_size</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename GemmTraits_ , typename ReductionTraits_ > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">int <a class="el" href="structcutlass_1_1gemm_1_1SplitkPIGemmTraits.html">cutlass::gemm::SplitkPIGemmTraits</a>< GemmTraits_, ReductionTraits_ >::Params::workspace_size</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="device__gemm__traits_8h_source.html">device_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:40 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>
|