780 lines
51 KiB
HTML
780 lines
51 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::GemmDesc< AType_, BType_, CType_, DType_, SType_, Index_ > 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_1GemmDesc.html">GemmDesc</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_1GemmDesc-members.html">List of all members</a> </div>
|
|
<div class="headertitle">
|
|
<div class="title">cutlass::gemm::GemmDesc< AType_, BType_, CType_, DType_, SType_, Index_ > Struct Template Reference</div> </div>
|
|
</div><!--header-->
|
|
<div class="contents">
|
|
|
|
<p>GEMM problem description.
|
|
</p>
|
|
|
|
<p><code>#include <<a class="el" href="gemm__desc_8h_source.html">gemm_desc.h</a>></code></p>
|
|
<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:abd12fea9779ada02c0f2266909602171"><td class="memItemLeft" align="right" valign="top">typedef Index_ </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#abd12fea9779ada02c0f2266909602171">Index</a></td></tr>
|
|
<tr class="memdesc:abd12fea9779ada02c0f2266909602171"><td class="mdescLeft"> </td><td class="mdescRight">Index type for dimensions and strides. <a href="#abd12fea9779ada02c0f2266909602171">More...</a><br /></td></tr>
|
|
<tr class="separator:abd12fea9779ada02c0f2266909602171"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a22642bd88ccb24fec3df87230537c037"><td class="memItemLeft" align="right" valign="top">typedef AType_ </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a22642bd88ccb24fec3df87230537c037">AType</a></td></tr>
|
|
<tr class="memdesc:a22642bd88ccb24fec3df87230537c037"><td class="mdescLeft"> </td><td class="mdescRight">Source accumulator matrix type. <a href="#a22642bd88ccb24fec3df87230537c037">More...</a><br /></td></tr>
|
|
<tr class="separator:a22642bd88ccb24fec3df87230537c037"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:aea3a523662dc801bf9abf205a8e400fe"><td class="memItemLeft" align="right" valign="top">typedef <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a22642bd88ccb24fec3df87230537c037">AType</a> const, 2 > </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#aea3a523662dc801bf9abf205a8e400fe">TensorRefA</a></td></tr>
|
|
<tr class="memdesc:aea3a523662dc801bf9abf205a8e400fe"><td class="mdescLeft"> </td><td class="mdescRight">Tensor reference to A operand. <a href="#aea3a523662dc801bf9abf205a8e400fe">More...</a><br /></td></tr>
|
|
<tr class="separator:aea3a523662dc801bf9abf205a8e400fe"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a23317c22122ea2721a7a3e0c12e07a49"><td class="memItemLeft" align="right" valign="top">typedef BType_ </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a23317c22122ea2721a7a3e0c12e07a49">BType</a></td></tr>
|
|
<tr class="memdesc:a23317c22122ea2721a7a3e0c12e07a49"><td class="mdescLeft"> </td><td class="mdescRight">Destination accumulator type. <a href="#a23317c22122ea2721a7a3e0c12e07a49">More...</a><br /></td></tr>
|
|
<tr class="separator:a23317c22122ea2721a7a3e0c12e07a49"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a7de7de97d237d9dcc0d89de7b04342ab"><td class="memItemLeft" align="right" valign="top">typedef <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a23317c22122ea2721a7a3e0c12e07a49">BType</a> const, 2 > </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a7de7de97d237d9dcc0d89de7b04342ab">TensorRefB</a></td></tr>
|
|
<tr class="memdesc:a7de7de97d237d9dcc0d89de7b04342ab"><td class="mdescLeft"> </td><td class="mdescRight">Tensor reference to B operand. <a href="#a7de7de97d237d9dcc0d89de7b04342ab">More...</a><br /></td></tr>
|
|
<tr class="separator:a7de7de97d237d9dcc0d89de7b04342ab"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:aa46907b69a3b1d0db5c3db1709867704"><td class="memItemLeft" align="right" valign="top">typedef CType_ </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#aa46907b69a3b1d0db5c3db1709867704">CType</a></td></tr>
|
|
<tr class="memdesc:aa46907b69a3b1d0db5c3db1709867704"><td class="mdescLeft"> </td><td class="mdescRight">Source accumulator matrix type. <a href="#aa46907b69a3b1d0db5c3db1709867704">More...</a><br /></td></tr>
|
|
<tr class="separator:aa46907b69a3b1d0db5c3db1709867704"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a51f77118d545308068b0dc287abf4e8b"><td class="memItemLeft" align="right" valign="top">typedef <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#aa46907b69a3b1d0db5c3db1709867704">CType</a> const, 2 > </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a51f77118d545308068b0dc287abf4e8b">TensorRefC</a></td></tr>
|
|
<tr class="memdesc:a51f77118d545308068b0dc287abf4e8b"><td class="mdescLeft"> </td><td class="mdescRight">Tensor reference to C operand. <a href="#a51f77118d545308068b0dc287abf4e8b">More...</a><br /></td></tr>
|
|
<tr class="separator:a51f77118d545308068b0dc287abf4e8b"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a837a1c513f71ddb2729f8d2e6320b8cd"><td class="memItemLeft" align="right" valign="top">typedef DType_ </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a837a1c513f71ddb2729f8d2e6320b8cd">DType</a></td></tr>
|
|
<tr class="memdesc:a837a1c513f71ddb2729f8d2e6320b8cd"><td class="mdescLeft"> </td><td class="mdescRight">Destination accumulator type. <a href="#a837a1c513f71ddb2729f8d2e6320b8cd">More...</a><br /></td></tr>
|
|
<tr class="separator:a837a1c513f71ddb2729f8d2e6320b8cd"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a1f59ad60a529aa1db6e66ade3cb65052"><td class="memItemLeft" align="right" valign="top">typedef <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a837a1c513f71ddb2729f8d2e6320b8cd">DType</a>, 2 > </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a1f59ad60a529aa1db6e66ade3cb65052">TensorRefD</a></td></tr>
|
|
<tr class="memdesc:a1f59ad60a529aa1db6e66ade3cb65052"><td class="mdescLeft"> </td><td class="mdescRight">Tensor reference to D operand. <a href="#a1f59ad60a529aa1db6e66ade3cb65052">More...</a><br /></td></tr>
|
|
<tr class="separator:a1f59ad60a529aa1db6e66ade3cb65052"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a258e8c008db365b88654f46df3017ec2"><td class="memItemLeft" align="right" valign="top">typedef SType_ </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a258e8c008db365b88654f46df3017ec2">SType</a></td></tr>
|
|
<tr class="memdesc:a258e8c008db365b88654f46df3017ec2"><td class="mdescLeft"> </td><td class="mdescRight">Scalar type for alpha and beta. <a href="#a258e8c008db365b88654f46df3017ec2">More...</a><br /></td></tr>
|
|
<tr class="separator:a258e8c008db365b88654f46df3017ec2"><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:ae2708b731cbb99d3e638382ecf599425"><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_1gemm_1_1GemmDesc.html#ae2708b731cbb99d3e638382ecf599425">GemmDesc</a> ()</td></tr>
|
|
<tr class="memdesc:ae2708b731cbb99d3e638382ecf599425"><td class="mdescLeft"> </td><td class="mdescRight">Default ctor. <a href="#ae2708b731cbb99d3e638382ecf599425">More...</a><br /></td></tr>
|
|
<tr class="separator:ae2708b731cbb99d3e638382ecf599425"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a948af4a974f1aa74d3b6da9cd3e185de"><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_1gemm_1_1GemmDesc.html#a948af4a974f1aa74d3b6da9cd3e185de">GemmDesc</a> (<a class="el" href="structcutlass_1_1Coord.html">Coord</a>< 3 > _problem_size, <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a258e8c008db365b88654f46df3017ec2">SType</a> _alpha, <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#aea3a523662dc801bf9abf205a8e400fe">TensorRefA</a> const &_A, <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a7de7de97d237d9dcc0d89de7b04342ab">TensorRefB</a> const &_B, <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a258e8c008db365b88654f46df3017ec2">SType</a> _beta, <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a51f77118d545308068b0dc287abf4e8b">TensorRefC</a> const &_C, <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a1f59ad60a529aa1db6e66ade3cb65052">TensorRefD</a> const &_D)</td></tr>
|
|
<tr class="memdesc:a948af4a974f1aa74d3b6da9cd3e185de"><td class="mdescLeft"> </td><td class="mdescRight">Constructor for basic GEMM with batch count = 1. <a href="#a948af4a974f1aa74d3b6da9cd3e185de">More...</a><br /></td></tr>
|
|
<tr class="separator:a948af4a974f1aa74d3b6da9cd3e185de"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a7292ac0196f22525b13f223a02271bc3"><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_1gemm_1_1GemmDesc.html#a7292ac0196f22525b13f223a02271bc3">GemmDesc</a> (<a class="el" href="structcutlass_1_1gemm_1_1GemmCoord.html">GemmCoord</a> _problem_size, <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a258e8c008db365b88654f46df3017ec2">SType</a> _alpha, <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#aea3a523662dc801bf9abf205a8e400fe">TensorRefA</a> const &_A, <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a7de7de97d237d9dcc0d89de7b04342ab">TensorRefB</a> const &_B, <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a258e8c008db365b88654f46df3017ec2">SType</a> _beta, <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a51f77118d545308068b0dc287abf4e8b">TensorRefC</a> const &_C, <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a1f59ad60a529aa1db6e66ade3cb65052">TensorRefD</a> const &_D)</td></tr>
|
|
<tr class="memdesc:a7292ac0196f22525b13f223a02271bc3"><td class="mdescLeft"> </td><td class="mdescRight">Constructor for basic GEMM with batch count = 1. <a href="#a7292ac0196f22525b13f223a02271bc3">More...</a><br /></td></tr>
|
|
<tr class="separator:a7292ac0196f22525b13f223a02271bc3"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:af96cd9c8f2454ea118dfa7358dd24824"><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_1gemm_1_1GemmDesc.html#af96cd9c8f2454ea118dfa7358dd24824">GemmDesc</a> (<a class="el" href="structcutlass_1_1gemm_1_1GemmCoord.html">GemmCoord</a> _problem_size, <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a258e8c008db365b88654f46df3017ec2">SType</a> _alpha, <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#aea3a523662dc801bf9abf205a8e400fe">TensorRefA</a> const &_A, long long _batch_stride_A, <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a7de7de97d237d9dcc0d89de7b04342ab">TensorRefB</a> const &_B, long long _batch_stride_B, <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a258e8c008db365b88654f46df3017ec2">SType</a> _beta, <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a51f77118d545308068b0dc287abf4e8b">TensorRefC</a> const &_C, long long _batch_stride_C, <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a1f59ad60a529aa1db6e66ade3cb65052">TensorRefD</a> const &_D, long long _batch_stride_D)</td></tr>
|
|
<tr class="memdesc:af96cd9c8f2454ea118dfa7358dd24824"><td class="mdescLeft"> </td><td class="mdescRight">Constructor for strided batch GEMM GEMM. <a href="#af96cd9c8f2454ea118dfa7358dd24824">More...</a><br /></td></tr>
|
|
<tr class="separator:af96cd9c8f2454ea118dfa7358dd24824"><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:a52a5538b1b5208090417cf98d8735e3f"><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_1GemmDesc.html#a52a5538b1b5208090417cf98d8735e3f">problem_size</a></td></tr>
|
|
<tr class="memdesc:a52a5538b1b5208090417cf98d8735e3f"><td class="mdescLeft"> </td><td class="mdescRight">The dimensions of the GEMM. <a href="#a52a5538b1b5208090417cf98d8735e3f">More...</a><br /></td></tr>
|
|
<tr class="separator:a52a5538b1b5208090417cf98d8735e3f"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:aa82600c82e17ea1233f2f74be4aa3785"><td class="memItemLeft" align="right" valign="top"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a258e8c008db365b88654f46df3017ec2">SType</a> </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#aa82600c82e17ea1233f2f74be4aa3785">alpha</a></td></tr>
|
|
<tr class="memdesc:aa82600c82e17ea1233f2f74be4aa3785"><td class="mdescLeft"> </td><td class="mdescRight">The alpha scaling values. <a href="#aa82600c82e17ea1233f2f74be4aa3785">More...</a><br /></td></tr>
|
|
<tr class="separator:aa82600c82e17ea1233f2f74be4aa3785"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a80b0aae6e67b733ae5bf289d979a7c9b"><td class="memItemLeft" align="right" valign="top"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#aea3a523662dc801bf9abf205a8e400fe">TensorRefA</a> </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a80b0aae6e67b733ae5bf289d979a7c9b">A</a></td></tr>
|
|
<tr class="memdesc:a80b0aae6e67b733ae5bf289d979a7c9b"><td class="mdescLeft"> </td><td class="mdescRight">The source matrix A. <a href="#a80b0aae6e67b733ae5bf289d979a7c9b">More...</a><br /></td></tr>
|
|
<tr class="separator:a80b0aae6e67b733ae5bf289d979a7c9b"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a8b5a86d14eba0d3c5173753212d62599"><td class="memItemLeft" align="right" valign="top">long long </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a8b5a86d14eba0d3c5173753212d62599">batch_stride_A</a></td></tr>
|
|
<tr class="memdesc:a8b5a86d14eba0d3c5173753212d62599"><td class="mdescLeft"> </td><td class="mdescRight">batch stride for A operand <a href="#a8b5a86d14eba0d3c5173753212d62599">More...</a><br /></td></tr>
|
|
<tr class="separator:a8b5a86d14eba0d3c5173753212d62599"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:af0ac89b161f9cad96307f1ff3c80a774"><td class="memItemLeft" align="right" valign="top"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a7de7de97d237d9dcc0d89de7b04342ab">TensorRefB</a> </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#af0ac89b161f9cad96307f1ff3c80a774">B</a></td></tr>
|
|
<tr class="memdesc:af0ac89b161f9cad96307f1ff3c80a774"><td class="mdescLeft"> </td><td class="mdescRight">The source matrix B. <a href="#af0ac89b161f9cad96307f1ff3c80a774">More...</a><br /></td></tr>
|
|
<tr class="separator:af0ac89b161f9cad96307f1ff3c80a774"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a95e3fe05e4ca0d4019cbef2b1a54419a"><td class="memItemLeft" align="right" valign="top">long long </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a95e3fe05e4ca0d4019cbef2b1a54419a">batch_stride_B</a></td></tr>
|
|
<tr class="memdesc:a95e3fe05e4ca0d4019cbef2b1a54419a"><td class="mdescLeft"> </td><td class="mdescRight">batch stride for B operand <a href="#a95e3fe05e4ca0d4019cbef2b1a54419a">More...</a><br /></td></tr>
|
|
<tr class="separator:a95e3fe05e4ca0d4019cbef2b1a54419a"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:adac41a0baad9e65aa4a6fe12d249a02b"><td class="memItemLeft" align="right" valign="top"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a258e8c008db365b88654f46df3017ec2">SType</a> </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#adac41a0baad9e65aa4a6fe12d249a02b">beta</a></td></tr>
|
|
<tr class="memdesc:adac41a0baad9e65aa4a6fe12d249a02b"><td class="mdescLeft"> </td><td class="mdescRight">The beta scaling values. <a href="#adac41a0baad9e65aa4a6fe12d249a02b">More...</a><br /></td></tr>
|
|
<tr class="separator:adac41a0baad9e65aa4a6fe12d249a02b"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:ab384a226ab370fa5e25468fb99c63e30"><td class="memItemLeft" align="right" valign="top"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a51f77118d545308068b0dc287abf4e8b">TensorRefC</a> </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#ab384a226ab370fa5e25468fb99c63e30">C</a></td></tr>
|
|
<tr class="memdesc:ab384a226ab370fa5e25468fb99c63e30"><td class="mdescLeft"> </td><td class="mdescRight">The source matrix C. <a href="#ab384a226ab370fa5e25468fb99c63e30">More...</a><br /></td></tr>
|
|
<tr class="separator:ab384a226ab370fa5e25468fb99c63e30"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:aad3590dffa2e1ba82c834efae6b35ad2"><td class="memItemLeft" align="right" valign="top">long long </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#aad3590dffa2e1ba82c834efae6b35ad2">batch_stride_C</a></td></tr>
|
|
<tr class="memdesc:aad3590dffa2e1ba82c834efae6b35ad2"><td class="mdescLeft"> </td><td class="mdescRight">batch stride for C operand <a href="#aad3590dffa2e1ba82c834efae6b35ad2">More...</a><br /></td></tr>
|
|
<tr class="separator:aad3590dffa2e1ba82c834efae6b35ad2"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:adf95451422c529587d55aac0fecf0d9f"><td class="memItemLeft" align="right" valign="top"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a1f59ad60a529aa1db6e66ade3cb65052">TensorRefD</a> </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#adf95451422c529587d55aac0fecf0d9f">D</a></td></tr>
|
|
<tr class="memdesc:adf95451422c529587d55aac0fecf0d9f"><td class="mdescLeft"> </td><td class="mdescRight">The destination matrix D. <a href="#adf95451422c529587d55aac0fecf0d9f">More...</a><br /></td></tr>
|
|
<tr class="separator:adf95451422c529587d55aac0fecf0d9f"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a7b9afcc7e3105da1d002b1baa68d83de"><td class="memItemLeft" align="right" valign="top">long long </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a7b9afcc7e3105da1d002b1baa68d83de">batch_stride_D</a></td></tr>
|
|
<tr class="memdesc:a7b9afcc7e3105da1d002b1baa68d83de"><td class="mdescLeft"> </td><td class="mdescRight">batch stride for D operand <a href="#a7b9afcc7e3105da1d002b1baa68d83de">More...</a><br /></td></tr>
|
|
<tr class="separator:a7b9afcc7e3105da1d002b1baa68d83de"><td class="memSeparator" colspan="2"> </td></tr>
|
|
</table>
|
|
<h2 class="groupheader">Member Typedef Documentation</h2>
|
|
<a id="a22642bd88ccb24fec3df87230537c037"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a22642bd88ccb24fec3df87230537c037">◆ </a></span>AType</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int> </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">typedef AType_ <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html">cutlass::gemm::GemmDesc</a>< AType_, BType_, CType_, DType_, SType_, Index_ >::<a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a22642bd88ccb24fec3df87230537c037">AType</a></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="a23317c22122ea2721a7a3e0c12e07a49"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a23317c22122ea2721a7a3e0c12e07a49">◆ </a></span>BType</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int> </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">typedef BType_ <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html">cutlass::gemm::GemmDesc</a>< AType_, BType_, CType_, DType_, SType_, Index_ >::<a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a23317c22122ea2721a7a3e0c12e07a49">BType</a></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="aa46907b69a3b1d0db5c3db1709867704"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#aa46907b69a3b1d0db5c3db1709867704">◆ </a></span>CType</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int> </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">typedef CType_ <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html">cutlass::gemm::GemmDesc</a>< AType_, BType_, CType_, DType_, SType_, Index_ >::<a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#aa46907b69a3b1d0db5c3db1709867704">CType</a></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="a837a1c513f71ddb2729f8d2e6320b8cd"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a837a1c513f71ddb2729f8d2e6320b8cd">◆ </a></span>DType</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int> </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">typedef DType_ <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html">cutlass::gemm::GemmDesc</a>< AType_, BType_, CType_, DType_, SType_, Index_ >::<a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a837a1c513f71ddb2729f8d2e6320b8cd">DType</a></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="abd12fea9779ada02c0f2266909602171"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#abd12fea9779ada02c0f2266909602171">◆ </a></span>Index</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int> </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">typedef Index_ <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html">cutlass::gemm::GemmDesc</a>< AType_, BType_, CType_, DType_, SType_, Index_ >::<a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#abd12fea9779ada02c0f2266909602171">Index</a></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="a258e8c008db365b88654f46df3017ec2"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a258e8c008db365b88654f46df3017ec2">◆ </a></span>SType</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int> </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">typedef SType_ <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html">cutlass::gemm::GemmDesc</a>< AType_, BType_, CType_, DType_, SType_, Index_ >::<a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a258e8c008db365b88654f46df3017ec2">SType</a></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="aea3a523662dc801bf9abf205a8e400fe"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#aea3a523662dc801bf9abf205a8e400fe">◆ </a></span>TensorRefA</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int> </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">typedef <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a><<a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a22642bd88ccb24fec3df87230537c037">AType</a> const, 2> <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html">cutlass::gemm::GemmDesc</a>< AType_, BType_, CType_, DType_, SType_, Index_ >::<a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#aea3a523662dc801bf9abf205a8e400fe">TensorRefA</a></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="a7de7de97d237d9dcc0d89de7b04342ab"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a7de7de97d237d9dcc0d89de7b04342ab">◆ </a></span>TensorRefB</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int> </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">typedef <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a><<a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a23317c22122ea2721a7a3e0c12e07a49">BType</a> const, 2> <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html">cutlass::gemm::GemmDesc</a>< AType_, BType_, CType_, DType_, SType_, Index_ >::<a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a7de7de97d237d9dcc0d89de7b04342ab">TensorRefB</a></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="a51f77118d545308068b0dc287abf4e8b"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a51f77118d545308068b0dc287abf4e8b">◆ </a></span>TensorRefC</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int> </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">typedef <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a><<a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#aa46907b69a3b1d0db5c3db1709867704">CType</a> const, 2> <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html">cutlass::gemm::GemmDesc</a>< AType_, BType_, CType_, DType_, SType_, Index_ >::<a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a51f77118d545308068b0dc287abf4e8b">TensorRefC</a></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="a1f59ad60a529aa1db6e66ade3cb65052"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a1f59ad60a529aa1db6e66ade3cb65052">◆ </a></span>TensorRefD</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int> </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">typedef <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a><<a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a837a1c513f71ddb2729f8d2e6320b8cd">DType</a>, 2> <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html">cutlass::gemm::GemmDesc</a>< AType_, BType_, CType_, DType_, SType_, Index_ >::<a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a1f59ad60a529aa1db6e66ade3cb65052">TensorRefD</a></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<h2 class="groupheader">Constructor & Destructor Documentation</h2>
|
|
<a id="ae2708b731cbb99d3e638382ecf599425"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#ae2708b731cbb99d3e638382ecf599425">◆ </a></span>GemmDesc() <span class="overload">[1/4]</span></h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int> </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> <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html">cutlass::gemm::GemmDesc</a>< AType_, BType_, CType_, DType_, SType_, Index_ >::<a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html">GemmDesc</a> </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="a948af4a974f1aa74d3b6da9cd3e185de"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a948af4a974f1aa74d3b6da9cd3e185de">◆ </a></span>GemmDesc() <span class="overload">[2/4]</span></h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int> </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> <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html">cutlass::gemm::GemmDesc</a>< AType_, BType_, CType_, DType_, SType_, Index_ >::<a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html">GemmDesc</a> </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1Coord.html">Coord</a>< 3 > </td>
|
|
<td class="paramname"><em>_problem_size</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a258e8c008db365b88654f46df3017ec2">SType</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_1GemmDesc.html#aea3a523662dc801bf9abf205a8e400fe">TensorRefA</a> const & </td>
|
|
<td class="paramname"><em>_A</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a7de7de97d237d9dcc0d89de7b04342ab">TensorRefB</a> const & </td>
|
|
<td class="paramname"><em>_B</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a258e8c008db365b88654f46df3017ec2">SType</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_1GemmDesc.html#a51f77118d545308068b0dc287abf4e8b">TensorRefC</a> const & </td>
|
|
<td class="paramname"><em>_C</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a1f59ad60a529aa1db6e66ade3cb65052">TensorRefD</a> const & </td>
|
|
<td class="paramname"><em>_D</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="a7292ac0196f22525b13f223a02271bc3"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a7292ac0196f22525b13f223a02271bc3">◆ </a></span>GemmDesc() <span class="overload">[3/4]</span></h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int> </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> <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html">cutlass::gemm::GemmDesc</a>< AType_, BType_, CType_, DType_, SType_, Index_ >::<a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html">GemmDesc</a> </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmCoord.html">GemmCoord</a> </td>
|
|
<td class="paramname"><em>_problem_size</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a258e8c008db365b88654f46df3017ec2">SType</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_1GemmDesc.html#aea3a523662dc801bf9abf205a8e400fe">TensorRefA</a> const & </td>
|
|
<td class="paramname"><em>_A</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a7de7de97d237d9dcc0d89de7b04342ab">TensorRefB</a> const & </td>
|
|
<td class="paramname"><em>_B</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a258e8c008db365b88654f46df3017ec2">SType</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_1GemmDesc.html#a51f77118d545308068b0dc287abf4e8b">TensorRefC</a> const & </td>
|
|
<td class="paramname"><em>_C</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a1f59ad60a529aa1db6e66ade3cb65052">TensorRefD</a> const & </td>
|
|
<td class="paramname"><em>_D</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="af96cd9c8f2454ea118dfa7358dd24824"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#af96cd9c8f2454ea118dfa7358dd24824">◆ </a></span>GemmDesc() <span class="overload">[4/4]</span></h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int> </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> <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html">cutlass::gemm::GemmDesc</a>< AType_, BType_, CType_, DType_, SType_, Index_ >::<a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html">GemmDesc</a> </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmCoord.html">GemmCoord</a> </td>
|
|
<td class="paramname"><em>_problem_size</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a258e8c008db365b88654f46df3017ec2">SType</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_1GemmDesc.html#aea3a523662dc801bf9abf205a8e400fe">TensorRefA</a> const & </td>
|
|
<td class="paramname"><em>_A</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">long long </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_1GemmDesc.html#a7de7de97d237d9dcc0d89de7b04342ab">TensorRefB</a> const & </td>
|
|
<td class="paramname"><em>_B</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">long long </td>
|
|
<td class="paramname"><em>_batch_stride_B</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a258e8c008db365b88654f46df3017ec2">SType</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_1GemmDesc.html#a51f77118d545308068b0dc287abf4e8b">TensorRefC</a> const & </td>
|
|
<td class="paramname"><em>_C</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">long long </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_1GemmDesc.html#a1f59ad60a529aa1db6e66ade3cb65052">TensorRefD</a> const & </td>
|
|
<td class="paramname"><em>_D</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">long long </td>
|
|
<td class="paramname"><em>_batch_stride_D</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="a80b0aae6e67b733ae5bf289d979a7c9b"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a80b0aae6e67b733ae5bf289d979a7c9b">◆ </a></span>A</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int> </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#aea3a523662dc801bf9abf205a8e400fe">TensorRefA</a> <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html">cutlass::gemm::GemmDesc</a>< AType_, BType_, CType_, DType_, SType_, Index_ >::A</td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="aa82600c82e17ea1233f2f74be4aa3785"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#aa82600c82e17ea1233f2f74be4aa3785">◆ </a></span>alpha</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int> </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a258e8c008db365b88654f46df3017ec2">SType</a> <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html">cutlass::gemm::GemmDesc</a>< AType_, BType_, CType_, DType_, SType_, Index_ >::alpha</td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="af0ac89b161f9cad96307f1ff3c80a774"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#af0ac89b161f9cad96307f1ff3c80a774">◆ </a></span>B</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int> </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a7de7de97d237d9dcc0d89de7b04342ab">TensorRefB</a> <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html">cutlass::gemm::GemmDesc</a>< AType_, BType_, CType_, DType_, SType_, Index_ >::B</td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="a8b5a86d14eba0d3c5173753212d62599"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a8b5a86d14eba0d3c5173753212d62599">◆ </a></span>batch_stride_A</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int> </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">long long <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html">cutlass::gemm::GemmDesc</a>< AType_, BType_, CType_, DType_, SType_, Index_ >::batch_stride_A</td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="a95e3fe05e4ca0d4019cbef2b1a54419a"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a95e3fe05e4ca0d4019cbef2b1a54419a">◆ </a></span>batch_stride_B</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int> </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">long long <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html">cutlass::gemm::GemmDesc</a>< AType_, BType_, CType_, DType_, SType_, Index_ >::batch_stride_B</td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="aad3590dffa2e1ba82c834efae6b35ad2"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#aad3590dffa2e1ba82c834efae6b35ad2">◆ </a></span>batch_stride_C</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int> </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">long long <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html">cutlass::gemm::GemmDesc</a>< AType_, BType_, CType_, DType_, SType_, Index_ >::batch_stride_C</td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="a7b9afcc7e3105da1d002b1baa68d83de"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a7b9afcc7e3105da1d002b1baa68d83de">◆ </a></span>batch_stride_D</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int> </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">long long <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html">cutlass::gemm::GemmDesc</a>< AType_, BType_, CType_, DType_, SType_, Index_ >::batch_stride_D</td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="adac41a0baad9e65aa4a6fe12d249a02b"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#adac41a0baad9e65aa4a6fe12d249a02b">◆ </a></span>beta</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int> </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a258e8c008db365b88654f46df3017ec2">SType</a> <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html">cutlass::gemm::GemmDesc</a>< AType_, BType_, CType_, DType_, SType_, Index_ >::beta</td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="ab384a226ab370fa5e25468fb99c63e30"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#ab384a226ab370fa5e25468fb99c63e30">◆ </a></span>C</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int> </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a51f77118d545308068b0dc287abf4e8b">TensorRefC</a> <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html">cutlass::gemm::GemmDesc</a>< AType_, BType_, CType_, DType_, SType_, Index_ >::C</td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="adf95451422c529587d55aac0fecf0d9f"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#adf95451422c529587d55aac0fecf0d9f">◆ </a></span>D</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int> </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname"><a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html#a1f59ad60a529aa1db6e66ade3cb65052">TensorRefD</a> <a class="el" href="structcutlass_1_1gemm_1_1GemmDesc.html">cutlass::gemm::GemmDesc</a>< AType_, BType_, CType_, DType_, SType_, Index_ >::D</td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a id="a52a5538b1b5208090417cf98d8735e3f"></a>
|
|
<h2 class="memtitle"><span class="permalink"><a href="#a52a5538b1b5208090417cf98d8735e3f">◆ </a></span>problem_size</h2>
|
|
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename AType_, typename BType_, typename CType_, typename DType_, typename SType_, typename Index_ = int> </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_1GemmDesc.html">cutlass::gemm::GemmDesc</a>< AType_, BType_, CType_, DType_, SType_, Index_ >::problem_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="gemm__desc_8h_source.html">gemm_desc.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>
|