2127 lines
118 KiB
HTML
2127 lines
118 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.11"/>
|
|
<title>CUTLASS: cutlass::reference::host Namespace 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/javascript">
|
|
$(document).ready(function() { init_search(); });
|
|
</script>
|
|
<script type="text/x-mathjax-config">
|
|
MathJax.Hub.Config({
|
|
extensions: ["tex2jax.js"],
|
|
jax: ["input/TeX","output/HTML-CSS"],
|
|
});
|
|
</script><script type="text/javascript" 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="projectlogo"><img alt="Logo" src="cutlass-logo-small.png"/></td>
|
|
<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.11 -->
|
|
<script type="text/javascript">
|
|
var searchBox = new SearchBox("searchBox", "search",false,'Search');
|
|
</script>
|
|
<div id="navrow1" class="tabs">
|
|
<ul class="tablist">
|
|
<li><a href="index.html"><span>Main Page</span></a></li>
|
|
<li><a href="modules.html"><span>Modules</span></a></li>
|
|
<li class="current"><a href="namespaces.html"><span>Namespaces</span></a></li>
|
|
<li><a href="annotated.html"><span>Classes</span></a></li>
|
|
<li><a href="files.html"><span>Files</span></a></li>
|
|
<li>
|
|
<div id="MSearchBox" class="MSearchBoxInactive">
|
|
<span class="left">
|
|
<img id="MSearchSelect" src="search/mag_sel.png"
|
|
onmouseover="return searchBox.OnSearchSelectShow()"
|
|
onmouseout="return searchBox.OnSearchSelectHide()"
|
|
alt=""/>
|
|
<input type="text" id="MSearchField" value="Search" accesskey="S"
|
|
onfocus="searchBox.OnSearchFieldFocus(true)"
|
|
onblur="searchBox.OnSearchFieldFocus(false)"
|
|
onkeyup="searchBox.OnSearchFieldChange(event)"/>
|
|
</span><span class="right">
|
|
<a id="MSearchClose" href="javascript:searchBox.CloseResultsWindow()"><img id="MSearchCloseImg" border="0" src="search/close.png" alt=""/></a>
|
|
</span>
|
|
</div>
|
|
</li>
|
|
</ul>
|
|
</div>
|
|
<div id="navrow2" class="tabs2">
|
|
<ul class="tablist">
|
|
<li><a href="namespaces.html"><span>Namespace List</span></a></li>
|
|
<li><a href="namespacemembers.html"><span>Namespace Members</span></a></li>
|
|
</ul>
|
|
</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_1reference.html">reference</a></li><li class="navelem"><a class="el" href="namespacecutlass_1_1reference_1_1host.html">host</a></li> </ul>
|
|
</div>
|
|
</div><!-- top -->
|
|
<div class="header">
|
|
<div class="summary">
|
|
<a href="#namespaces">Namespaces</a> |
|
|
<a href="#nested-classes">Classes</a> |
|
|
<a href="#func-members">Functions</a> </div>
|
|
<div class="headertitle">
|
|
<div class="title">cutlass::reference::host Namespace Reference</div> </div>
|
|
</div><!--header-->
|
|
<div class="contents">
|
|
<table class="memberdecls">
|
|
<tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="namespaces"></a>
|
|
Namespaces</h2></td></tr>
|
|
<tr class="memitem:namespacecutlass_1_1reference_1_1host_1_1detail"><td class="memItemLeft" align="right" valign="top">  </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host_1_1detail.html">detail</a></td></tr>
|
|
<tr class="memdesc:namespacecutlass_1_1reference_1_1host_1_1detail"><td class="mdescLeft"> </td><td class="mdescRight">Defines several helpers. <br /></td></tr>
|
|
<tr class="separator:"><td class="memSeparator" colspan="2"> </td></tr>
|
|
</table><table class="memberdecls">
|
|
<tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="nested-classes"></a>
|
|
Classes</h2></td></tr>
|
|
<tr class="memitem:"><td class="memItemLeft" align="right" valign="top">struct  </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1reference_1_1host_1_1BlockForEach.html">BlockForEach</a></td></tr>
|
|
<tr class="separator:"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:"><td class="memItemLeft" align="right" valign="top">struct  </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1reference_1_1host_1_1Gemm.html">Gemm</a></td></tr>
|
|
<tr class="separator:"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:"><td class="memItemLeft" align="right" valign="top">struct  </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1reference_1_1host_1_1Gemm_3_01ElementA_00_01LayoutA_00_01ElementB_00_01LayoutB_193dd3a37f00deff1e5dcd7c310afb1f.html">Gemm< ElementA, LayoutA, ElementB, LayoutB, ElementC, LayoutC, ScalarType, ComputeType, arch::OpMultiplyAdd ></a></td></tr>
|
|
<tr class="memdesc:"><td class="mdescLeft"> </td><td class="mdescRight">Partial specialization for multiply-add. <a href="structcutlass_1_1reference_1_1host_1_1Gemm_3_01ElementA_00_01LayoutA_00_01ElementB_00_01LayoutB_193dd3a37f00deff1e5dcd7c310afb1f.html#details">More...</a><br /></td></tr>
|
|
<tr class="separator:"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:"><td class="memItemLeft" align="right" valign="top">struct  </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1reference_1_1host_1_1Gemm_3_01ElementA_00_01LayoutA_00_01ElementB_00_01LayoutB_55729eac7dbd6bf311ea36f680e83e93.html">Gemm< ElementA, LayoutA, ElementB, LayoutB, ElementC, LayoutC, ScalarType, ComputeType, arch::OpMultiplyAddSaturate ></a></td></tr>
|
|
<tr class="memdesc:"><td class="mdescLeft"> </td><td class="mdescRight">Partial specialization for multiply-add-saturate. <a href="structcutlass_1_1reference_1_1host_1_1Gemm_3_01ElementA_00_01LayoutA_00_01ElementB_00_01LayoutB_55729eac7dbd6bf311ea36f680e83e93.html#details">More...</a><br /></td></tr>
|
|
<tr class="separator:"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:"><td class="memItemLeft" align="right" valign="top">struct  </td><td class="memItemRight" valign="bottom"><a class="el" href="structcutlass_1_1reference_1_1host_1_1Gemm_3_01ElementA_00_01LayoutA_00_01ElementB_00_01LayoutB_4f3f32c4b336238abfd741e87bfced46.html">Gemm< ElementA, LayoutA, ElementB, LayoutB, ElementC, LayoutC, ScalarType, ComputeType, arch::OpXorPopc ></a></td></tr>
|
|
<tr class="memdesc:"><td class="mdescLeft"> </td><td class="mdescRight">Partial specialization for XOR-popc. <a href="structcutlass_1_1reference_1_1host_1_1Gemm_3_01ElementA_00_01LayoutA_00_01ElementB_00_01LayoutB_4f3f32c4b336238abfd741e87bfced46.html#details">More...</a><br /></td></tr>
|
|
<tr class="separator:"><td class="memSeparator" colspan="2"> </td></tr>
|
|
</table><table class="memberdecls">
|
|
<tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="func-members"></a>
|
|
Functions</h2></td></tr>
|
|
<tr class="memitem:a300d68abd082150020768c0a94044a34"><td class="memTemplParams" colspan="2">template<typename ElementA , typename LayoutA , typename ElementB , typename LayoutB , typename ElementC , typename LayoutC , typename ScalarType , typename ComputeType , typename InnerProductOp = multiply_add<ComputeType>, typename ConvertOp = NumericConverter<ElementC, ScalarType>> </td></tr>
|
|
<tr class="memitem:a300d68abd082150020768c0a94044a34"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#a300d68abd082150020768c0a94044a34">compute_gemm</a> (<a class="el" href="structcutlass_1_1gemm_1_1GemmCoord.html">gemm::GemmCoord</a> problem_size, ScalarType alpha, <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementA, LayoutA > tensor_a, <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementB, LayoutB > tensor_b, ScalarType beta, <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementC, LayoutC > tensor_c, <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementC, LayoutC > tensor_d, ComputeType initial_accum)</td></tr>
|
|
<tr class="separator:a300d68abd082150020768c0a94044a34"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:aa75c5933390f3960666e97b37c854877"><td class="memTemplParams" colspan="2">template<typename ElementA , typename LayoutA , typename ElementB , typename LayoutB , typename ElementC , typename LayoutC , typename ScalarType , typename ComputeType , typename InnerProductOp = multiply_add<ComputeType>, typename ConvertOp = NumericConverter<ElementC, ScalarType>> </td></tr>
|
|
<tr class="memitem:aa75c5933390f3960666e97b37c854877"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#aa75c5933390f3960666e97b37c854877">compute_gemm</a> (<a class="el" href="structcutlass_1_1gemm_1_1GemmCoord.html">gemm::GemmCoord</a> problem_size, ScalarType alpha, <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementA, LayoutA > tensor_a, <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementB, LayoutB > tensor_b, ScalarType beta, <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementC, LayoutC > tensor_c, ComputeType initial_accum)</td></tr>
|
|
<tr class="separator:aa75c5933390f3960666e97b37c854877"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a2c1067fa5de91e2f48589120f62125c2"><td class="memTemplParams" colspan="2">template<typename TensorRefCollectionA , typename TensorRefCollectionB , typename TensorRefCollectionC , typename ScalarType , typename AccumulatorType > </td></tr>
|
|
<tr class="memitem:a2c1067fa5de91e2f48589120f62125c2"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#a2c1067fa5de91e2f48589120f62125c2">BatchedGemm</a> (<a class="el" href="structcutlass_1_1gemm_1_1GemmCoord.html">gemm::GemmCoord</a> problem_size, int batch_count, ScalarType alpha, TensorRefCollectionA const &tensor_a, TensorRefCollectionB const &tensor_b, ScalarType beta, TensorRefCollectionC &tensor_c, AccumulatorType initial_accum)</td></tr>
|
|
<tr class="memdesc:a2c1067fa5de91e2f48589120f62125c2"><td class="mdescLeft"> </td><td class="mdescRight">Computes a batch of GEMMs over a set of matrices of common dimension. <a href="#a2c1067fa5de91e2f48589120f62125c2">More...</a><br /></td></tr>
|
|
<tr class="separator:a2c1067fa5de91e2f48589120f62125c2"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a1d0a79a48353119706ffa09d570c2182"><td class="memTemplParams" colspan="2">template<typename TensorRefCollectionA , typename TensorRefCollectionB , typename TensorRefCollectionC , typename ScalarType , typename AccumulatorType > </td></tr>
|
|
<tr class="memitem:a1d0a79a48353119706ffa09d570c2182"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#a1d0a79a48353119706ffa09d570c2182">BatchedGemm</a> (<a class="el" href="structcutlass_1_1gemm_1_1GemmCoord.html">gemm::GemmCoord</a> problem_size, int batch_count, ScalarType alpha, TensorRefCollectionA const &tensor_a, TensorRefCollectionB const &tensor_b, ScalarType beta, TensorRefCollectionC &tensor_c)</td></tr>
|
|
<tr class="separator:a1d0a79a48353119706ffa09d570c2182"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a968f6ed2ed0c23c3b90b424fcf8a446e"><td class="memTemplParams" colspan="2">template<typename ElementA , typename LayoutA , typename ElementB , typename LayoutB , typename ElementC , typename LayoutC , typename ScalarType , typename ComputeType , typename ConvertOp = NumericConverter<ElementC, ScalarType>, typename InnerProductOp = multiply_add<ComputeType>> </td></tr>
|
|
<tr class="memitem:a968f6ed2ed0c23c3b90b424fcf8a446e"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#a968f6ed2ed0c23c3b90b424fcf8a446e">GemmComplex</a> (<a class="el" href="structcutlass_1_1gemm_1_1GemmCoord.html">gemm::GemmCoord</a> problem_size, ScalarType alpha, <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementA, LayoutA > tensor_a, <a class="el" href="namespacecutlass.html#a59f08b1b99c4d52257b962d35ba55cde">ComplexTransform</a> transform_a, <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementB, LayoutB > tensor_b, <a class="el" href="namespacecutlass.html#a59f08b1b99c4d52257b962d35ba55cde">ComplexTransform</a> transform_b, ScalarType beta, <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementC, LayoutC > tensor_c, ComputeType initial_accum)</td></tr>
|
|
<tr class="separator:a968f6ed2ed0c23c3b90b424fcf8a446e"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a2a802a3e81e1b473fd1e76a606b919a4"><td class="memTemplParams" colspan="2">template<typename ElementA , typename LayoutA , typename ElementB , typename LayoutB , typename ElementC , typename LayoutC , typename ScalarType > </td></tr>
|
|
<tr class="memitem:a2a802a3e81e1b473fd1e76a606b919a4"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#a2a802a3e81e1b473fd1e76a606b919a4">GemmComplex</a> (<a class="el" href="structcutlass_1_1gemm_1_1GemmCoord.html">gemm::GemmCoord</a> problem_size, ScalarType alpha, <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementA, LayoutA > tensor_a, <a class="el" href="namespacecutlass.html#a59f08b1b99c4d52257b962d35ba55cde">ComplexTransform</a> transform_a, <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementB, LayoutB > tensor_b, <a class="el" href="namespacecutlass.html#a59f08b1b99c4d52257b962d35ba55cde">ComplexTransform</a> transform_b, ScalarType beta, <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementC, LayoutC > tensor_c)</td></tr>
|
|
<tr class="separator:a2a802a3e81e1b473fd1e76a606b919a4"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a317479d2bc5b4d4ed3ec0b8e51ff1150"><td class="memTemplParams" colspan="2">template<typename Element , typename Layout > </td></tr>
|
|
<tr class="memitem:a317479d2bc5b4d4ed3ec0b8e51ff1150"><td class="memTemplItemLeft" align="right" valign="top">bool </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#a317479d2bc5b4d4ed3ec0b8e51ff1150">TensorEquals</a> (<a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > const &lhs, <a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > const &rhs)</td></tr>
|
|
<tr class="memdesc:a317479d2bc5b4d4ed3ec0b8e51ff1150"><td class="mdescLeft"> </td><td class="mdescRight">Returns true if two tensor views are equal. <a href="#a317479d2bc5b4d4ed3ec0b8e51ff1150">More...</a><br /></td></tr>
|
|
<tr class="separator:a317479d2bc5b4d4ed3ec0b8e51ff1150"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a8fecbd766b668983848255e0e2532e55"><td class="memTemplParams" colspan="2">template<typename Element , typename Layout > </td></tr>
|
|
<tr class="memitem:a8fecbd766b668983848255e0e2532e55"><td class="memTemplItemLeft" align="right" valign="top">bool </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#a8fecbd766b668983848255e0e2532e55">TensorNotEquals</a> (<a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > const &lhs, <a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > const &rhs)</td></tr>
|
|
<tr class="memdesc:a8fecbd766b668983848255e0e2532e55"><td class="mdescLeft"> </td><td class="mdescRight">Returns true if two tensor views are NOT equal. <a href="#a8fecbd766b668983848255e0e2532e55">More...</a><br /></td></tr>
|
|
<tr class="separator:a8fecbd766b668983848255e0e2532e55"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:adcfbba47411196d82bb8cc4fb0f70402"><td class="memTemplParams" colspan="2">template<typename Element , typename Layout > </td></tr>
|
|
<tr class="memitem:adcfbba47411196d82bb8cc4fb0f70402"><td class="memTemplItemLeft" align="right" valign="top">bool </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#adcfbba47411196d82bb8cc4fb0f70402">TensorContains</a> (<a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > const &view, Element value)</td></tr>
|
|
<tr class="memdesc:adcfbba47411196d82bb8cc4fb0f70402"><td class="mdescLeft"> </td><td class="mdescRight">Returns true if a value is present in a tensor. <a href="#adcfbba47411196d82bb8cc4fb0f70402">More...</a><br /></td></tr>
|
|
<tr class="separator:adcfbba47411196d82bb8cc4fb0f70402"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:aa92b096c37d8a366390c368d17f21b85"><td class="memTemplParams" colspan="2">template<typename Element , typename Layout > </td></tr>
|
|
<tr class="memitem:aa92b096c37d8a366390c368d17f21b85"><td class="memTemplItemLeft" align="right" valign="top">std::pair< bool, <a class="el" href="structcutlass_1_1Coord.html">Coord</a>< Layout::kRank > > </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#aa92b096c37d8a366390c368d17f21b85">TensorFind</a> (<a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > const &view, Element value)</td></tr>
|
|
<tr class="memdesc:aa92b096c37d8a366390c368d17f21b85"><td class="mdescLeft"> </td><td class="mdescRight">< Layout function <a href="#aa92b096c37d8a366390c368d17f21b85">More...</a><br /></td></tr>
|
|
<tr class="separator:aa92b096c37d8a366390c368d17f21b85"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:ab32bea7b552a408f93fc2f153b44fcb8"><td class="memTemplParams" colspan="2">template<typename DstElement , typename DstLayout , typename SrcElement , typename SrcLayout , typename F > </td></tr>
|
|
<tr class="memitem:ab32bea7b552a408f93fc2f153b44fcb8"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#ab32bea7b552a408f93fc2f153b44fcb8">TensorCopy</a> (<a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< DstElement, DstLayout > dst, <a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< SrcElement, SrcLayout > src, F const &transform)</td></tr>
|
|
<tr class="memdesc:ab32bea7b552a408f93fc2f153b44fcb8"><td class="mdescLeft"> </td><td class="mdescRight">Copies elements from one tensor view into another, satisfying bounds of each tensor. <a href="#ab32bea7b552a408f93fc2f153b44fcb8">More...</a><br /></td></tr>
|
|
<tr class="separator:ab32bea7b552a408f93fc2f153b44fcb8"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a32903a34034cfe040157a5cd48c325ce"><td class="memTemplParams" colspan="2">template<typename DstElement , typename DstLayout , typename SrcElement , typename SrcLayout , typename F > </td></tr>
|
|
<tr class="memitem:a32903a34034cfe040157a5cd48c325ce"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#a32903a34034cfe040157a5cd48c325ce">TensorCopy</a> (<a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< DstElement, DstLayout > dst, <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< SrcElement, SrcLayout > src, F const &transform)</td></tr>
|
|
<tr class="separator:a32903a34034cfe040157a5cd48c325ce"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a723c4026b4a73a6050aa203aee95de84"><td class="memTemplParams" colspan="2">template<typename DstElement , typename DstLayout , typename SrcElement , typename SrcLayout , typename F > </td></tr>
|
|
<tr class="memitem:a723c4026b4a73a6050aa203aee95de84"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#a723c4026b4a73a6050aa203aee95de84">TensorCopy</a> (<a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< DstElement, DstLayout > dst, <a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< SrcElement, SrcLayout > src, F const &transform)</td></tr>
|
|
<tr class="separator:a723c4026b4a73a6050aa203aee95de84"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a1ffea013419e1ac514797633dd46d6a6"><td class="memTemplParams" colspan="2">template<typename DstElement , typename DstLayout , typename SrcElement , typename SrcLayout > </td></tr>
|
|
<tr class="memitem:a1ffea013419e1ac514797633dd46d6a6"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#a1ffea013419e1ac514797633dd46d6a6">TensorCopy</a> (<a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< DstElement, DstLayout > dst, <a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< SrcElement, SrcLayout > src)</td></tr>
|
|
<tr class="separator:a1ffea013419e1ac514797633dd46d6a6"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:aae15ab711d7be5074953df3fa62bae0d"><td class="memTemplParams" colspan="2">template<typename DstElement , typename DstLayout , typename SrcElement , typename SrcLayout , typename F > </td></tr>
|
|
<tr class="memitem:aae15ab711d7be5074953df3fa62bae0d"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#aae15ab711d7be5074953df3fa62bae0d">TensorCopy</a> (<a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< DstElement, DstLayout > dst, <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< SrcElement, SrcLayout > src)</td></tr>
|
|
<tr class="separator:aae15ab711d7be5074953df3fa62bae0d"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a513f4fef76bc1a6b34b39f0d6bd7c48a"><td class="memTemplParams" colspan="2">template<typename DstElement , typename DstLayout , typename SrcElement , typename SrcLayout > </td></tr>
|
|
<tr class="memitem:a513f4fef76bc1a6b34b39f0d6bd7c48a"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#a513f4fef76bc1a6b34b39f0d6bd7c48a">TensorCopy</a> (<a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< DstElement, DstLayout > dst, <a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< SrcElement, SrcLayout > src)</td></tr>
|
|
<tr class="separator:a513f4fef76bc1a6b34b39f0d6bd7c48a"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a17840338fea08affd4bbf4a450140dff"><td class="memTemplParams" colspan="2">template<typename ElementD , typename LayoutD , typename ElementA , typename LayoutA , typename ElementB , typename LayoutB > </td></tr>
|
|
<tr class="memitem:a17840338fea08affd4bbf4a450140dff"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#a17840338fea08affd4bbf4a450140dff">TensorAdd</a> (<a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< ElementD, LayoutD > d, <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementA, LayoutA > a, <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementB, LayoutB > b)</td></tr>
|
|
<tr class="memdesc:a17840338fea08affd4bbf4a450140dff"><td class="mdescLeft"> </td><td class="mdescRight">Adds two tensors and stores in the destination tensor: d = a + b. <a href="#a17840338fea08affd4bbf4a450140dff">More...</a><br /></td></tr>
|
|
<tr class="separator:a17840338fea08affd4bbf4a450140dff"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:adf0e1dc76c868172d941a5091dee8dde"><td class="memTemplParams" colspan="2">template<typename ElementD , typename LayoutD , typename ElementA , typename LayoutA > </td></tr>
|
|
<tr class="memitem:adf0e1dc76c868172d941a5091dee8dde"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#adf0e1dc76c868172d941a5091dee8dde">TensorAdd</a> (<a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< ElementD, LayoutD > d, <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementA, LayoutA > a)</td></tr>
|
|
<tr class="memdesc:adf0e1dc76c868172d941a5091dee8dde"><td class="mdescLeft"> </td><td class="mdescRight">Adds a tensor in place: d = d .+ a. <a href="#adf0e1dc76c868172d941a5091dee8dde">More...</a><br /></td></tr>
|
|
<tr class="separator:adf0e1dc76c868172d941a5091dee8dde"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a3b2eede84fbc860b8f0d178c14e4cbb1"><td class="memTemplParams" colspan="2">template<typename ElementD , typename LayoutD , typename ElementA , typename LayoutA , typename ElementB , typename LayoutB > </td></tr>
|
|
<tr class="memitem:a3b2eede84fbc860b8f0d178c14e4cbb1"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#a3b2eede84fbc860b8f0d178c14e4cbb1">TensorSub</a> (<a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< ElementD, LayoutD > d, <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementA, LayoutA > a, <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementB, LayoutB > b)</td></tr>
|
|
<tr class="memdesc:a3b2eede84fbc860b8f0d178c14e4cbb1"><td class="mdescLeft"> </td><td class="mdescRight">Subtracts two tensors and stores in the destination tensor: d = a - b. <a href="#a3b2eede84fbc860b8f0d178c14e4cbb1">More...</a><br /></td></tr>
|
|
<tr class="separator:a3b2eede84fbc860b8f0d178c14e4cbb1"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a30294715f2c97a2f3f8adca4a16a97ff"><td class="memTemplParams" colspan="2">template<typename ElementD , typename LayoutD , typename ElementA , typename LayoutA , typename ElementB , typename LayoutB > </td></tr>
|
|
<tr class="memitem:a30294715f2c97a2f3f8adca4a16a97ff"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#a30294715f2c97a2f3f8adca4a16a97ff">TensorSub</a> (<a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< ElementD, LayoutD > d, <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementA, LayoutA > a)</td></tr>
|
|
<tr class="memdesc:a30294715f2c97a2f3f8adca4a16a97ff"><td class="mdescLeft"> </td><td class="mdescRight">Subtracts two tensors in place: d = d .- a. <a href="#a30294715f2c97a2f3f8adca4a16a97ff">More...</a><br /></td></tr>
|
|
<tr class="separator:a30294715f2c97a2f3f8adca4a16a97ff"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:affc066e432cb8213ce84a81371623f7e"><td class="memTemplParams" colspan="2">template<typename ElementD , typename LayoutD , typename ElementA , typename LayoutA , typename ElementB , typename LayoutB > </td></tr>
|
|
<tr class="memitem:affc066e432cb8213ce84a81371623f7e"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#affc066e432cb8213ce84a81371623f7e">TensorMul</a> (<a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< ElementD, LayoutD > d, <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementA, LayoutA > a, <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementB, LayoutB > b)</td></tr>
|
|
<tr class="memdesc:affc066e432cb8213ce84a81371623f7e"><td class="mdescLeft"> </td><td class="mdescRight">Multiplies two tensors and stores in the destination tensor: d = a .* b. <a href="#affc066e432cb8213ce84a81371623f7e">More...</a><br /></td></tr>
|
|
<tr class="separator:affc066e432cb8213ce84a81371623f7e"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:ac11f9fadfa4526ba8e6fa62f806eae7c"><td class="memTemplParams" colspan="2">template<typename ElementD , typename LayoutD , typename ElementA , typename LayoutA > </td></tr>
|
|
<tr class="memitem:ac11f9fadfa4526ba8e6fa62f806eae7c"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#ac11f9fadfa4526ba8e6fa62f806eae7c">TensorMul</a> (<a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< ElementD, LayoutD > d, <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementA, LayoutA > a)</td></tr>
|
|
<tr class="memdesc:ac11f9fadfa4526ba8e6fa62f806eae7c"><td class="mdescLeft"> </td><td class="mdescRight">Multiplies tensors in place: d = d .* a. <a href="#ac11f9fadfa4526ba8e6fa62f806eae7c">More...</a><br /></td></tr>
|
|
<tr class="separator:ac11f9fadfa4526ba8e6fa62f806eae7c"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a35db5f5e1946616fd626991cfe689c6c"><td class="memTemplParams" colspan="2">template<typename ElementD , typename LayoutD , typename ElementA , typename LayoutA , typename ElementB , typename LayoutB > </td></tr>
|
|
<tr class="memitem:a35db5f5e1946616fd626991cfe689c6c"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#a35db5f5e1946616fd626991cfe689c6c">TensorDiv</a> (<a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< ElementD, LayoutD > d, <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementA, LayoutA > a, <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementB, LayoutB > b)</td></tr>
|
|
<tr class="memdesc:a35db5f5e1946616fd626991cfe689c6c"><td class="mdescLeft"> </td><td class="mdescRight">Divides two tensors and stores in the destination tensor: d = a ./ b. <a href="#a35db5f5e1946616fd626991cfe689c6c">More...</a><br /></td></tr>
|
|
<tr class="separator:a35db5f5e1946616fd626991cfe689c6c"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:ae1e20de08fb5d034abe1b5d64041514c"><td class="memTemplParams" colspan="2">template<typename ElementD , typename LayoutD , typename ElementA , typename LayoutA > </td></tr>
|
|
<tr class="memitem:ae1e20de08fb5d034abe1b5d64041514c"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#ae1e20de08fb5d034abe1b5d64041514c">TensorDiv</a> (<a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< ElementD, LayoutD > d, <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementA, LayoutA > a)</td></tr>
|
|
<tr class="memdesc:ae1e20de08fb5d034abe1b5d64041514c"><td class="mdescLeft"> </td><td class="mdescRight">Divides tensors in place: d = d ./ a. <a href="#ae1e20de08fb5d034abe1b5d64041514c">More...</a><br /></td></tr>
|
|
<tr class="separator:ae1e20de08fb5d034abe1b5d64041514c"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:ac9ea3d1c0a265daf9f43c4d32b2d3d4c"><td class="memTemplParams" colspan="2">template<typename ElementD , typename LayoutD , typename ElementA , typename LayoutA , typename ElementB , typename LayoutB > </td></tr>
|
|
<tr class="memitem:ac9ea3d1c0a265daf9f43c4d32b2d3d4c"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#ac9ea3d1c0a265daf9f43c4d32b2d3d4c">TensorModulus</a> (<a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< ElementD, LayoutD > d, <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementA, LayoutA > a, <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementB, LayoutB > b)</td></tr>
|
|
<tr class="memdesc:ac9ea3d1c0a265daf9f43c4d32b2d3d4c"><td class="mdescLeft"> </td><td class="mdescRight">Divides two tensors and stores in the destination tensor: d = a ./ b. <a href="#ac9ea3d1c0a265daf9f43c4d32b2d3d4c">More...</a><br /></td></tr>
|
|
<tr class="separator:ac9ea3d1c0a265daf9f43c4d32b2d3d4c"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a50b72c70c26ff46dff8747010e3a92b9"><td class="memTemplParams" colspan="2">template<typename ElementD , typename LayoutD , typename ElementA , typename LayoutA > </td></tr>
|
|
<tr class="memitem:a50b72c70c26ff46dff8747010e3a92b9"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#a50b72c70c26ff46dff8747010e3a92b9">TensorModulus</a> (<a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< ElementD, LayoutD > d, <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementA, LayoutA > a)</td></tr>
|
|
<tr class="memdesc:a50b72c70c26ff46dff8747010e3a92b9"><td class="mdescLeft"> </td><td class="mdescRight">Divides tensors in place: d = d ./ a. <a href="#a50b72c70c26ff46dff8747010e3a92b9">More...</a><br /></td></tr>
|
|
<tr class="separator:a50b72c70c26ff46dff8747010e3a92b9"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a81b0f81940ce0ecc0d486d2f45659f6e"><td class="memTemplParams" colspan="2">template<typename Element , typename Layout > </td></tr>
|
|
<tr class="memitem:a81b0f81940ce0ecc0d486d2f45659f6e"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#a81b0f81940ce0ecc0d486d2f45659f6e">TensorFill</a> (<a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > dst, Element val=Element(0))</td></tr>
|
|
<tr class="memdesc:a81b0f81940ce0ecc0d486d2f45659f6e"><td class="mdescLeft"> </td><td class="mdescRight">Fills a tensor with a uniform value. <a href="#a81b0f81940ce0ecc0d486d2f45659f6e">More...</a><br /></td></tr>
|
|
<tr class="separator:a81b0f81940ce0ecc0d486d2f45659f6e"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a5187d9f07f2b1edebfbad067c7a84826"><td class="memTemplParams" colspan="2">template<typename Element , typename Layout > </td></tr>
|
|
<tr class="memitem:a5187d9f07f2b1edebfbad067c7a84826"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#a5187d9f07f2b1edebfbad067c7a84826">TensorFillRandomGaussian</a> (<a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > dst, uint64_t seed, double mean=0, double stddev=1, int bits=-1)</td></tr>
|
|
<tr class="memdesc:a5187d9f07f2b1edebfbad067c7a84826"><td class="mdescLeft"> </td><td class="mdescRight">Fills a tensor with random values with a Gaussian distribution. <a href="#a5187d9f07f2b1edebfbad067c7a84826">More...</a><br /></td></tr>
|
|
<tr class="separator:a5187d9f07f2b1edebfbad067c7a84826"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a121079d5cb24dd0e0339cee552a854de"><td class="memTemplParams" colspan="2">template<typename Element > </td></tr>
|
|
<tr class="memitem:a121079d5cb24dd0e0339cee552a854de"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#a121079d5cb24dd0e0339cee552a854de">BlockFillRandomGaussian</a> (Element *ptr, size_t capacity, uint64_t seed, double mean=0, double stddev=1, int bits=-1)</td></tr>
|
|
<tr class="memdesc:a121079d5cb24dd0e0339cee552a854de"><td class="mdescLeft"> </td><td class="mdescRight">Fills a tensor with random values with a Gaussian distribution. <a href="#a121079d5cb24dd0e0339cee552a854de">More...</a><br /></td></tr>
|
|
<tr class="separator:a121079d5cb24dd0e0339cee552a854de"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a8f1500a3e2cb694323bf9d25bafc0fbc"><td class="memTemplParams" colspan="2">template<typename Element , typename Layout > </td></tr>
|
|
<tr class="memitem:a8f1500a3e2cb694323bf9d25bafc0fbc"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#a8f1500a3e2cb694323bf9d25bafc0fbc">TensorFillRandomUniform</a> (<a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > dst, uint64_t seed, double max=1, double min=0, int bits=-1)</td></tr>
|
|
<tr class="memdesc:a8f1500a3e2cb694323bf9d25bafc0fbc"><td class="mdescLeft"> </td><td class="mdescRight">Fills a tensor with random values with a uniform random distribution. <a href="#a8f1500a3e2cb694323bf9d25bafc0fbc">More...</a><br /></td></tr>
|
|
<tr class="separator:a8f1500a3e2cb694323bf9d25bafc0fbc"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a417152b59865d2ef6995ee2398bcea8d"><td class="memTemplParams" colspan="2">template<typename Element > </td></tr>
|
|
<tr class="memitem:a417152b59865d2ef6995ee2398bcea8d"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#a417152b59865d2ef6995ee2398bcea8d">BlockFillRandomUniform</a> (Element *ptr, size_t capacity, uint64_t seed, double max=1, double min=0, int bits=-1)</td></tr>
|
|
<tr class="memdesc:a417152b59865d2ef6995ee2398bcea8d"><td class="mdescLeft"> </td><td class="mdescRight">Fills a tensor with random values with a uniform random distribution. <a href="#a417152b59865d2ef6995ee2398bcea8d">More...</a><br /></td></tr>
|
|
<tr class="separator:a417152b59865d2ef6995ee2398bcea8d"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a1c81144ca36832a48d04d1b5b6498080"><td class="memTemplParams" colspan="2">template<typename Element , typename Layout > </td></tr>
|
|
<tr class="memitem:a1c81144ca36832a48d04d1b5b6498080"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#a1c81144ca36832a48d04d1b5b6498080">TensorFillDiagonal</a> (<a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > dst, Element diag=Element(1), Element other=Element(0))</td></tr>
|
|
<tr class="memdesc:a1c81144ca36832a48d04d1b5b6498080"><td class="mdescLeft"> </td><td class="mdescRight">Fills a tensor everywhere with a unique value for its diagonal. <a href="#a1c81144ca36832a48d04d1b5b6498080">More...</a><br /></td></tr>
|
|
<tr class="separator:a1c81144ca36832a48d04d1b5b6498080"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a29548cb522d9c147cf34263ecac75d89"><td class="memTemplParams" colspan="2">template<typename Element , typename Layout > </td></tr>
|
|
<tr class="memitem:a29548cb522d9c147cf34263ecac75d89"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#a29548cb522d9c147cf34263ecac75d89">TensorFillIdentity</a> (<a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > dst)</td></tr>
|
|
<tr class="memdesc:a29548cb522d9c147cf34263ecac75d89"><td class="mdescLeft"> </td><td class="mdescRight">Helper to fill a tensor's diagonal with 1 and 0 everywhere else. <a href="#a29548cb522d9c147cf34263ecac75d89">More...</a><br /></td></tr>
|
|
<tr class="separator:a29548cb522d9c147cf34263ecac75d89"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:acbf747241e8ac6ef9b1702b735a7913e"><td class="memTemplParams" colspan="2">template<typename Element , typename Layout > </td></tr>
|
|
<tr class="memitem:acbf747241e8ac6ef9b1702b735a7913e"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#acbf747241e8ac6ef9b1702b735a7913e">TensorUpdateDiagonal</a> (<a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > dst, Element val=Element(1))</td></tr>
|
|
<tr class="memdesc:acbf747241e8ac6ef9b1702b735a7913e"><td class="mdescLeft"> </td><td class="mdescRight">Writes a uniform value to the diagonal of a tensor without modifying off-diagonal elements. <a href="#acbf747241e8ac6ef9b1702b735a7913e">More...</a><br /></td></tr>
|
|
<tr class="separator:acbf747241e8ac6ef9b1702b735a7913e"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a6d146dc0390e4c045f08b5d2adfcf48a"><td class="memTemplParams" colspan="2">template<typename Element , typename Layout > </td></tr>
|
|
<tr class="memitem:a6d146dc0390e4c045f08b5d2adfcf48a"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#a6d146dc0390e4c045f08b5d2adfcf48a">TensorUpdateOffDiagonal</a> (<a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > dst, Element other=Element(1))</td></tr>
|
|
<tr class="memdesc:a6d146dc0390e4c045f08b5d2adfcf48a"><td class="mdescLeft"> </td><td class="mdescRight">Writes a uniform value to all elements in the tensor without modifying diagonal elements. <a href="#a6d146dc0390e4c045f08b5d2adfcf48a">More...</a><br /></td></tr>
|
|
<tr class="separator:a6d146dc0390e4c045f08b5d2adfcf48a"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:ac23a650217ddd1640807c2e91e69ab91"><td class="memTemplParams" colspan="2">template<typename Element , typename Layout > </td></tr>
|
|
<tr class="memitem:ac23a650217ddd1640807c2e91e69ab91"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#ac23a650217ddd1640807c2e91e69ab91">TensorFillLinear</a> (<a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > dst, Array< Element, Layout::kRank > const &v, Element s=Element(0))</td></tr>
|
|
<tr class="memdesc:ac23a650217ddd1640807c2e91e69ab91"><td class="mdescLeft"> </td><td class="mdescRight">Fills tensor with a linear combination of its coordinate and another vector. <a href="#ac23a650217ddd1640807c2e91e69ab91">More...</a><br /></td></tr>
|
|
<tr class="separator:ac23a650217ddd1640807c2e91e69ab91"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:ad6e2cd99b8096eaaf79c0a6edbd3e420"><td class="memTemplParams" colspan="2">template<typename Element , typename Layout > </td></tr>
|
|
<tr class="memitem:ad6e2cd99b8096eaaf79c0a6edbd3e420"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#ad6e2cd99b8096eaaf79c0a6edbd3e420">TensorFillSequential</a> (<a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > dst, Element s=Element(0))</td></tr>
|
|
<tr class="memdesc:ad6e2cd99b8096eaaf79c0a6edbd3e420"><td class="mdescLeft"> </td><td class="mdescRight">Fills tensor with a linear combination of its coordinate and another vector. <a href="#ad6e2cd99b8096eaaf79c0a6edbd3e420">More...</a><br /></td></tr>
|
|
<tr class="separator:ad6e2cd99b8096eaaf79c0a6edbd3e420"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a1808624141976837e298340c9f6c0f6b"><td class="memTemplParams" colspan="2">template<typename Element > </td></tr>
|
|
<tr class="memitem:a1808624141976837e298340c9f6c0f6b"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#a1808624141976837e298340c9f6c0f6b">BlockFillSequential</a> (Element *ptr, int64_t capacity, Element v=Element(1), Element s=Element(0))</td></tr>
|
|
<tr class="memdesc:a1808624141976837e298340c9f6c0f6b"><td class="mdescLeft"> </td><td class="mdescRight">Fills a block of data with sequential elements. <a href="#a1808624141976837e298340c9f6c0f6b">More...</a><br /></td></tr>
|
|
<tr class="separator:a1808624141976837e298340c9f6c0f6b"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:ae6171d78c959aefff277cec4cad8fdb3"><td class="memTemplParams" colspan="2">template<typename Element > </td></tr>
|
|
<tr class="memitem:ae6171d78c959aefff277cec4cad8fdb3"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#ae6171d78c959aefff277cec4cad8fdb3">BlockFillRandom</a> (Element *ptr, size_t capacity, uint64_t seed, <a class="el" href="structcutlass_1_1Distribution.html">Distribution</a> dist)</td></tr>
|
|
<tr class="memdesc:ae6171d78c959aefff277cec4cad8fdb3"><td class="mdescLeft"> </td><td class="mdescRight">Fills a block of data with sequential elements. <a href="#ae6171d78c959aefff277cec4cad8fdb3">More...</a><br /></td></tr>
|
|
<tr class="separator:ae6171d78c959aefff277cec4cad8fdb3"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a224f376c3199603bde4ededa4357eea3"><td class="memTemplParams" colspan="2">template<typename Element , typename Layout > </td></tr>
|
|
<tr class="memitem:a224f376c3199603bde4ededa4357eea3"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#a224f376c3199603bde4ededa4357eea3">TensorCopyDiagonalIn</a> (<a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > dst, Element const *ptr)</td></tr>
|
|
<tr class="memdesc:a224f376c3199603bde4ededa4357eea3"><td class="mdescLeft"> </td><td class="mdescRight">Copies a diagonal in from host memory without modifying off-diagonal elements. <a href="#a224f376c3199603bde4ededa4357eea3">More...</a><br /></td></tr>
|
|
<tr class="separator:a224f376c3199603bde4ededa4357eea3"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a63f6629dd3aabb499ba430d84cb98e05"><td class="memTemplParams" colspan="2">template<typename Element , typename Layout > </td></tr>
|
|
<tr class="memitem:a63f6629dd3aabb499ba430d84cb98e05"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#a63f6629dd3aabb499ba430d84cb98e05">TensorCopyDiagonalOut</a> (Element *ptr, <a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > src)</td></tr>
|
|
<tr class="memdesc:a63f6629dd3aabb499ba430d84cb98e05"><td class="mdescLeft"> </td><td class="mdescRight">Copies the diagonal of a tensor into a dense buffer in host memory. <a href="#a63f6629dd3aabb499ba430d84cb98e05">More...</a><br /></td></tr>
|
|
<tr class="separator:a63f6629dd3aabb499ba430d84cb98e05"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a8c798c04df572b34e3ed3976d69f993d"><td class="memTemplParams" colspan="2">template<typename Func , int Rank> </td></tr>
|
|
<tr class="memitem:a8c798c04df572b34e3ed3976d69f993d"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#a8c798c04df572b34e3ed3976d69f993d">TensorForEach</a> (<a class="el" href="structcutlass_1_1Coord.html">Coord</a>< Rank > extent, Func &func)</td></tr>
|
|
<tr class="memdesc:a8c798c04df572b34e3ed3976d69f993d"><td class="mdescLeft"> </td><td class="mdescRight">Iterates over the index space of a tensor. <a href="#a8c798c04df572b34e3ed3976d69f993d">More...</a><br /></td></tr>
|
|
<tr class="separator:a8c798c04df572b34e3ed3976d69f993d"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a3825b1aaaf5e5abf0de5f427e3481ada"><td class="memTemplParams" colspan="2">template<typename Func , int Rank> </td></tr>
|
|
<tr class="memitem:a3825b1aaaf5e5abf0de5f427e3481ada"><td class="memTemplItemLeft" align="right" valign="top">void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#a3825b1aaaf5e5abf0de5f427e3481ada">TensorForEachLambda</a> (<a class="el" href="structcutlass_1_1Coord.html">Coord</a>< Rank > extent, Func func)</td></tr>
|
|
<tr class="memdesc:a3825b1aaaf5e5abf0de5f427e3481ada"><td class="mdescLeft"> </td><td class="mdescRight">Iterates over the index space of a tensor and calls a C++ lambda. <a href="#a3825b1aaaf5e5abf0de5f427e3481ada">More...</a><br /></td></tr>
|
|
<tr class="separator:a3825b1aaaf5e5abf0de5f427e3481ada"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a0995c2093dcb952de0a8b9fb199352a1"><td class="memTemplParams" colspan="2">template<typename Element , typename Layout , typename ElementReduction > </td></tr>
|
|
<tr class="memitem:a0995c2093dcb952de0a8b9fb199352a1"><td class="memTemplItemLeft" align="right" valign="top">ElementReduction </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#a0995c2093dcb952de0a8b9fb199352a1">TensorNorm</a> (<a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > view, ElementReduction accumulator)</td></tr>
|
|
<tr class="memdesc:a0995c2093dcb952de0a8b9fb199352a1"><td class="mdescLeft"> </td><td class="mdescRight">Computes the p=2 norm of the elements of a tensor with arbitrary reduction data type. <a href="#a0995c2093dcb952de0a8b9fb199352a1">More...</a><br /></td></tr>
|
|
<tr class="separator:a0995c2093dcb952de0a8b9fb199352a1"><td class="memSeparator" colspan="2"> </td></tr>
|
|
<tr class="memitem:a8e1741fda1d41ab51a960e5f6fe5b59e"><td class="memTemplParams" colspan="2">template<typename Element , typename Layout > </td></tr>
|
|
<tr class="memitem:a8e1741fda1d41ab51a960e5f6fe5b59e"><td class="memTemplItemLeft" align="right" valign="top">double </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacecutlass_1_1reference_1_1host.html#a8e1741fda1d41ab51a960e5f6fe5b59e">TensorNorm</a> (<a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > view)</td></tr>
|
|
<tr class="memdesc:a8e1741fda1d41ab51a960e5f6fe5b59e"><td class="mdescLeft"> </td><td class="mdescRight">Computes the p=2 norm of the elements of a tensor. <a href="#a8e1741fda1d41ab51a960e5f6fe5b59e">More...</a><br /></td></tr>
|
|
<tr class="separator:a8e1741fda1d41ab51a960e5f6fe5b59e"><td class="memSeparator" colspan="2"> </td></tr>
|
|
</table>
|
|
<h2 class="groupheader">Function Documentation</h2>
|
|
<a class="anchor" id="a2c1067fa5de91e2f48589120f62125c2"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename TensorRefCollectionA , typename TensorRefCollectionB , typename TensorRefCollectionC , typename ScalarType , typename AccumulatorType > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::BatchedGemm </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmCoord.html">gemm::GemmCoord</a> </td>
|
|
<td class="paramname"><em>problem_size</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">int </td>
|
|
<td class="paramname"><em>batch_count</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">ScalarType </td>
|
|
<td class="paramname"><em>alpha</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">TensorRefCollectionA const & </td>
|
|
<td class="paramname"><em>tensor_a</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">TensorRefCollectionB const & </td>
|
|
<td class="paramname"><em>tensor_b</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">ScalarType </td>
|
|
<td class="paramname"><em>beta</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">TensorRefCollectionC & </td>
|
|
<td class="paramname"><em>tensor_c</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">AccumulatorType </td>
|
|
<td class="paramname"><em>initial_accum</em> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="a1d0a79a48353119706ffa09d570c2182"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename TensorRefCollectionA , typename TensorRefCollectionB , typename TensorRefCollectionC , typename ScalarType , typename AccumulatorType > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::BatchedGemm </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmCoord.html">gemm::GemmCoord</a> </td>
|
|
<td class="paramname"><em>problem_size</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">int </td>
|
|
<td class="paramname"><em>batch_count</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">ScalarType </td>
|
|
<td class="paramname"><em>alpha</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">TensorRefCollectionA const & </td>
|
|
<td class="paramname"><em>tensor_a</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">TensorRefCollectionB const & </td>
|
|
<td class="paramname"><em>tensor_b</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">ScalarType </td>
|
|
<td class="paramname"><em>beta</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">TensorRefCollectionC & </td>
|
|
<td class="paramname"><em>tensor_c</em> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<p>Computes a general matrix product among matrices (tensors of rank=2) pointed to by <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a> objects. </p>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="ae6171d78c959aefff277cec4cad8fdb3"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename Element > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::BlockFillRandom </td>
|
|
<td>(</td>
|
|
<td class="paramtype">Element * </td>
|
|
<td class="paramname"><em>ptr</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">size_t </td>
|
|
<td class="paramname"><em>capacity</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">uint64_t </td>
|
|
<td class="paramname"><em>seed</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1Distribution.html">Distribution</a> </td>
|
|
<td class="paramname"><em>dist</em> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="a121079d5cb24dd0e0339cee552a854de"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename Element > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::BlockFillRandomGaussian </td>
|
|
<td>(</td>
|
|
<td class="paramtype">Element * </td>
|
|
<td class="paramname"><em>ptr</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">size_t </td>
|
|
<td class="paramname"><em>capacity</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">uint64_t </td>
|
|
<td class="paramname"><em>seed</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">double </td>
|
|
<td class="paramname"><em>mean</em> = <code>0</code>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">double </td>
|
|
<td class="paramname"><em>stddev</em> = <code>1</code>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">int </td>
|
|
<td class="paramname"><em>bits</em> = <code>-1</code> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<p>< If non-negative, specifies number of fractional bits that are not truncated to zero. Permits reducing precision of data. </p>
|
|
<dl class="params"><dt>Parameters</dt><dd>
|
|
<table class="params">
|
|
<tr><td class="paramname">ptr</td><td>destination buffer </td></tr>
|
|
<tr><td class="paramname">capacity</td><td>number of elements </td></tr>
|
|
<tr><td class="paramname">seed</td><td>seed for RNG </td></tr>
|
|
<tr><td class="paramname">mean</td><td>Gaussian distribution's mean </td></tr>
|
|
<tr><td class="paramname">stddev</td><td>Gaussian distribution's standard deviation </td></tr>
|
|
</table>
|
|
</dd>
|
|
</dl>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="a417152b59865d2ef6995ee2398bcea8d"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename Element > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::BlockFillRandomUniform </td>
|
|
<td>(</td>
|
|
<td class="paramtype">Element * </td>
|
|
<td class="paramname"><em>ptr</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">size_t </td>
|
|
<td class="paramname"><em>capacity</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">uint64_t </td>
|
|
<td class="paramname"><em>seed</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">double </td>
|
|
<td class="paramname"><em>max</em> = <code>1</code>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">double </td>
|
|
<td class="paramname"><em>min</em> = <code>0</code>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">int </td>
|
|
<td class="paramname"><em>bits</em> = <code>-1</code> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<p>< If non-negative, specifies number of fractional bits that are not truncated to zero. Permits reducing precision of data. </p>
|
|
<dl class="params"><dt>Parameters</dt><dd>
|
|
<table class="params">
|
|
<tr><td class="paramname">seed</td><td>seed for RNG </td></tr>
|
|
<tr><td class="paramname">max</td><td>upper bound of distribution </td></tr>
|
|
<tr><td class="paramname">min</td><td>lower bound for distribution </td></tr>
|
|
</table>
|
|
</dd>
|
|
</dl>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="a1808624141976837e298340c9f6c0f6b"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename Element > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::BlockFillSequential </td>
|
|
<td>(</td>
|
|
<td class="paramtype">Element * </td>
|
|
<td class="paramname"><em>ptr</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">int64_t </td>
|
|
<td class="paramname"><em>capacity</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">Element </td>
|
|
<td class="paramname"><em>v</em> = <code>Element(1)</code>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">Element </td>
|
|
<td class="paramname"><em>s</em> = <code>Element(0)</code> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="a300d68abd082150020768c0a94044a34"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename ElementA , typename LayoutA , typename ElementB , typename LayoutB , typename ElementC , typename LayoutC , typename ScalarType , typename ComputeType , typename InnerProductOp = multiply_add<ComputeType>, typename ConvertOp = NumericConverter<ElementC, ScalarType>> </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::compute_gemm </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmCoord.html">gemm::GemmCoord</a> </td>
|
|
<td class="paramname"><em>problem_size</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">ScalarType </td>
|
|
<td class="paramname"><em>alpha</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementA, LayoutA > </td>
|
|
<td class="paramname"><em>tensor_a</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementB, LayoutB > </td>
|
|
<td class="paramname"><em>tensor_b</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">ScalarType </td>
|
|
<td class="paramname"><em>beta</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementC, LayoutC > </td>
|
|
<td class="paramname"><em>tensor_c</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementC, LayoutC > </td>
|
|
<td class="paramname"><em>tensor_d</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">ComputeType </td>
|
|
<td class="paramname"><em>initial_accum</em> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<p>Computes a general matrix product among matrices (tensors of rank=2) pointed to by <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a> objects. </p>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="aa75c5933390f3960666e97b37c854877"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename ElementA , typename LayoutA , typename ElementB , typename LayoutB , typename ElementC , typename LayoutC , typename ScalarType , typename ComputeType , typename InnerProductOp = multiply_add<ComputeType>, typename ConvertOp = NumericConverter<ElementC, ScalarType>> </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::compute_gemm </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmCoord.html">gemm::GemmCoord</a> </td>
|
|
<td class="paramname"><em>problem_size</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">ScalarType </td>
|
|
<td class="paramname"><em>alpha</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementA, LayoutA > </td>
|
|
<td class="paramname"><em>tensor_a</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementB, LayoutB > </td>
|
|
<td class="paramname"><em>tensor_b</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">ScalarType </td>
|
|
<td class="paramname"><em>beta</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementC, LayoutC > </td>
|
|
<td class="paramname"><em>tensor_c</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">ComputeType </td>
|
|
<td class="paramname"><em>initial_accum</em> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<p>Computes a general matrix product among matrices (tensors of rank=2) pointed to by <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a> objects. </p>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="a968f6ed2ed0c23c3b90b424fcf8a446e"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename ElementA , typename LayoutA , typename ElementB , typename LayoutB , typename ElementC , typename LayoutC , typename ScalarType , typename ComputeType , typename ConvertOp = NumericConverter<ElementC, ScalarType>, typename InnerProductOp = multiply_add<ComputeType>> </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::GemmComplex </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmCoord.html">gemm::GemmCoord</a> </td>
|
|
<td class="paramname"><em>problem_size</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">ScalarType </td>
|
|
<td class="paramname"><em>alpha</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementA, LayoutA > </td>
|
|
<td class="paramname"><em>tensor_a</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="namespacecutlass.html#a59f08b1b99c4d52257b962d35ba55cde">ComplexTransform</a> </td>
|
|
<td class="paramname"><em>transform_a</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementB, LayoutB > </td>
|
|
<td class="paramname"><em>tensor_b</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="namespacecutlass.html#a59f08b1b99c4d52257b962d35ba55cde">ComplexTransform</a> </td>
|
|
<td class="paramname"><em>transform_b</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">ScalarType </td>
|
|
<td class="paramname"><em>beta</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementC, LayoutC > </td>
|
|
<td class="paramname"><em>tensor_c</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">ComputeType </td>
|
|
<td class="paramname"><em>initial_accum</em> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<p>Computes a general matrix product among matrices (tensors of rank=2) pointed to by <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a> objects.</p>
|
|
<p>Explicitly naming types needed by this template can be cumbersome, particularly for the accumulator type, so a function argument 'initial_accum' is exposed. Passing AccumulatorType(0) as the last function argument can be easier than naming all template arguments explicitly. </p>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="a2a802a3e81e1b473fd1e76a606b919a4"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename ElementA , typename LayoutA , typename ElementB , typename LayoutB , typename ElementC , typename LayoutC , typename ScalarType > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::GemmComplex </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1gemm_1_1GemmCoord.html">gemm::GemmCoord</a> </td>
|
|
<td class="paramname"><em>problem_size</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">ScalarType </td>
|
|
<td class="paramname"><em>alpha</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementA, LayoutA > </td>
|
|
<td class="paramname"><em>tensor_a</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="namespacecutlass.html#a59f08b1b99c4d52257b962d35ba55cde">ComplexTransform</a> </td>
|
|
<td class="paramname"><em>transform_a</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementB, LayoutB > </td>
|
|
<td class="paramname"><em>tensor_b</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="namespacecutlass.html#a59f08b1b99c4d52257b962d35ba55cde">ComplexTransform</a> </td>
|
|
<td class="paramname"><em>transform_b</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">ScalarType </td>
|
|
<td class="paramname"><em>beta</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementC, LayoutC > </td>
|
|
<td class="paramname"><em>tensor_c</em> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<p>Computes a general matrix product among matrices (tensors of rank=2) pointed to by <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a> objects.</p>
|
|
<p>This assumes the accumulator type is the same type as the scalars. </p>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="a17840338fea08affd4bbf4a450140dff"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename ElementD , typename LayoutD , typename ElementA , typename LayoutA , typename ElementB , typename LayoutB > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::TensorAdd </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< ElementD, LayoutD > </td>
|
|
<td class="paramname"><em>d</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementA, LayoutA > </td>
|
|
<td class="paramname"><em>a</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementB, LayoutB > </td>
|
|
<td class="paramname"><em>b</em> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<dl class="params"><dt>Parameters</dt><dd>
|
|
<table class="params">
|
|
<tr><td class="paramname">d</td><td>destination tensor view </td></tr>
|
|
<tr><td class="paramname">a</td><td>A tensor reference </td></tr>
|
|
<tr><td class="paramname">b</td><td>B tensor reference </td></tr>
|
|
</table>
|
|
</dd>
|
|
</dl>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="adf0e1dc76c868172d941a5091dee8dde"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename ElementD , typename LayoutD , typename ElementA , typename LayoutA > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::TensorAdd </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< ElementD, LayoutD > </td>
|
|
<td class="paramname"><em>d</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementA, LayoutA > </td>
|
|
<td class="paramname"><em>a</em> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<dl class="params"><dt>Parameters</dt><dd>
|
|
<table class="params">
|
|
<tr><td class="paramname">d</td><td>destination tensor view </td></tr>
|
|
<tr><td class="paramname">a</td><td>A tensor reference </td></tr>
|
|
</table>
|
|
</dd>
|
|
</dl>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="adcfbba47411196d82bb8cc4fb0f70402"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename Element , typename Layout > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">bool cutlass::reference::host::TensorContains </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > const & </td>
|
|
<td class="paramname"><em>view</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">Element </td>
|
|
<td class="paramname"><em>value</em> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<p>< Layout function </p>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="ab32bea7b552a408f93fc2f153b44fcb8"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename DstElement , typename DstLayout , typename SrcElement , typename SrcLayout , typename F > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::TensorCopy </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< DstElement, DstLayout > </td>
|
|
<td class="paramname"><em>dst</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< SrcElement, SrcLayout > </td>
|
|
<td class="paramname"><em>src</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">F const & </td>
|
|
<td class="paramname"><em>transform</em> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="a32903a34034cfe040157a5cd48c325ce"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename DstElement , typename DstLayout , typename SrcElement , typename SrcLayout , typename F > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::TensorCopy </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< DstElement, DstLayout > </td>
|
|
<td class="paramname"><em>dst</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< SrcElement, SrcLayout > </td>
|
|
<td class="paramname"><em>src</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">F const & </td>
|
|
<td class="paramname"><em>transform</em> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<p>Copies elements from a <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a> into a <a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>. Assumes source tensor has sufficient extent to avoid out of bounds accesses. </p>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="a723c4026b4a73a6050aa203aee95de84"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename DstElement , typename DstLayout , typename SrcElement , typename SrcLayout , typename F > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::TensorCopy </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< DstElement, DstLayout > </td>
|
|
<td class="paramname"><em>dst</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< SrcElement, SrcLayout > </td>
|
|
<td class="paramname"><em>src</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">F const & </td>
|
|
<td class="paramname"><em>transform</em> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<p>Copies elements from a <a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a> into a <a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>. Assumes source tensor has sufficient extent to avoid out of bounds accesses. </p>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="a1ffea013419e1ac514797633dd46d6a6"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename DstElement , typename DstLayout , typename SrcElement , typename SrcLayout > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::TensorCopy </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< DstElement, DstLayout > </td>
|
|
<td class="paramname"><em>dst</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< SrcElement, SrcLayout > </td>
|
|
<td class="paramname"><em>src</em> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<p>Copies elements from one tensor view into another, satisfying bounds of each tensor. Succeeds if SrcElement can be converted to DstElement. </p>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="aae15ab711d7be5074953df3fa62bae0d"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename DstElement , typename DstLayout , typename SrcElement , typename SrcLayout , typename F > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::TensorCopy </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< DstElement, DstLayout > </td>
|
|
<td class="paramname"><em>dst</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< SrcElement, SrcLayout > </td>
|
|
<td class="paramname"><em>src</em> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<p>Copies elements from one tensor view into another, satisfying bounds of each tensor. Succeeds if SrcElement can be converted to DstElement. </p>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="a513f4fef76bc1a6b34b39f0d6bd7c48a"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename DstElement , typename DstLayout , typename SrcElement , typename SrcLayout > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::TensorCopy </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< DstElement, DstLayout > </td>
|
|
<td class="paramname"><em>dst</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< SrcElement, SrcLayout > </td>
|
|
<td class="paramname"><em>src</em> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<p>Copies elements from one tensor view into another, satisfying bounds of each tensor. Succeeds if SrcElement can be converted to DstElement. </p>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="a224f376c3199603bde4ededa4357eea3"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename Element , typename Layout > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::TensorCopyDiagonalIn </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > </td>
|
|
<td class="paramname"><em>dst</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">Element const * </td>
|
|
<td class="paramname"><em>ptr</em> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<p>< Layout function </p>
|
|
<p>< dense buffer of elements </p>
|
|
<dl class="params"><dt>Parameters</dt><dd>
|
|
<table class="params">
|
|
<tr><td class="paramname">dst</td><td>destination tensor </td></tr>
|
|
</table>
|
|
</dd>
|
|
</dl>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="a63f6629dd3aabb499ba430d84cb98e05"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename Element , typename Layout > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::TensorCopyDiagonalOut </td>
|
|
<td>(</td>
|
|
<td class="paramtype">Element * </td>
|
|
<td class="paramname"><em>ptr</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > </td>
|
|
<td class="paramname"><em>src</em> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<p>< Layout function </p>
|
|
<p>< source tensor </p>
|
|
<dl class="params"><dt>Parameters</dt><dd>
|
|
<table class="params">
|
|
<tr><td class="paramname">ptr</td><td>dense buffer of elements </td></tr>
|
|
</table>
|
|
</dd>
|
|
</dl>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="a35db5f5e1946616fd626991cfe689c6c"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename ElementD , typename LayoutD , typename ElementA , typename LayoutA , typename ElementB , typename LayoutB > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::TensorDiv </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< ElementD, LayoutD > </td>
|
|
<td class="paramname"><em>d</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementA, LayoutA > </td>
|
|
<td class="paramname"><em>a</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementB, LayoutB > </td>
|
|
<td class="paramname"><em>b</em> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<dl class="params"><dt>Parameters</dt><dd>
|
|
<table class="params">
|
|
<tr><td class="paramname">d</td><td>destination tensor view </td></tr>
|
|
<tr><td class="paramname">a</td><td>A tensor reference </td></tr>
|
|
<tr><td class="paramname">b</td><td>B tensor reference </td></tr>
|
|
</table>
|
|
</dd>
|
|
</dl>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="ae1e20de08fb5d034abe1b5d64041514c"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename ElementD , typename LayoutD , typename ElementA , typename LayoutA > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::TensorDiv </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< ElementD, LayoutD > </td>
|
|
<td class="paramname"><em>d</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementA, LayoutA > </td>
|
|
<td class="paramname"><em>a</em> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<dl class="params"><dt>Parameters</dt><dd>
|
|
<table class="params">
|
|
<tr><td class="paramname">d</td><td>destination tensor view </td></tr>
|
|
<tr><td class="paramname">a</td><td>A tensor reference </td></tr>
|
|
</table>
|
|
</dd>
|
|
</dl>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="a317479d2bc5b4d4ed3ec0b8e51ff1150"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename Element , typename Layout > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">bool cutlass::reference::host::TensorEquals </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > const & </td>
|
|
<td class="paramname"><em>lhs</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > const & </td>
|
|
<td class="paramname"><em>rhs</em> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<p>< Layout function </p>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="a81b0f81940ce0ecc0d486d2f45659f6e"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename Element , typename Layout > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::TensorFill </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > </td>
|
|
<td class="paramname"><em>dst</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">Element </td>
|
|
<td class="paramname"><em>val</em> = <code>Element(0)</code> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<p>< Layout function </p>
|
|
<p>< value to uniformly fill it with </p>
|
|
<dl class="params"><dt>Parameters</dt><dd>
|
|
<table class="params">
|
|
<tr><td class="paramname">dst</td><td>destination tensor </td></tr>
|
|
</table>
|
|
</dd>
|
|
</dl>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="a1c81144ca36832a48d04d1b5b6498080"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename Element , typename Layout > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::TensorFillDiagonal </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > </td>
|
|
<td class="paramname"><em>dst</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">Element </td>
|
|
<td class="paramname"><em>diag</em> = <code>Element(1)</code>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">Element </td>
|
|
<td class="paramname"><em>other</em> = <code>Element(0)</code> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<p>< Layout function </p>
|
|
<p>< value to write off the diagonal </p>
|
|
<dl class="params"><dt>Parameters</dt><dd>
|
|
<table class="params">
|
|
<tr><td class="paramname">dst</td><td>destination tensor </td></tr>
|
|
<tr><td class="paramname">diag</td><td>value to write in the diagonal </td></tr>
|
|
</table>
|
|
</dd>
|
|
</dl>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="a29548cb522d9c147cf34263ecac75d89"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename Element , typename Layout > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::TensorFillIdentity </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > </td>
|
|
<td class="paramname"><em>dst</em></td><td>)</td>
|
|
<td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<p>< Layout function </p>
|
|
<p>< destination tensor </p>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="ac23a650217ddd1640807c2e91e69ab91"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename Element , typename Layout > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::TensorFillLinear </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > </td>
|
|
<td class="paramname"><em>dst</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">Array< Element, Layout::kRank > const & </td>
|
|
<td class="paramname"><em>v</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">Element </td>
|
|
<td class="paramname"><em>s</em> = <code>Element(0)</code> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<p>< Layout function </p>
|
|
<dl class="params"><dt>Parameters</dt><dd>
|
|
<table class="params">
|
|
<tr><td class="paramname">dst</td><td>destination tensor </td></tr>
|
|
</table>
|
|
</dd>
|
|
</dl>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="a5187d9f07f2b1edebfbad067c7a84826"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename Element , typename Layout > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::TensorFillRandomGaussian </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > </td>
|
|
<td class="paramname"><em>dst</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">uint64_t </td>
|
|
<td class="paramname"><em>seed</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">double </td>
|
|
<td class="paramname"><em>mean</em> = <code>0</code>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">double </td>
|
|
<td class="paramname"><em>stddev</em> = <code>1</code>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">int </td>
|
|
<td class="paramname"><em>bits</em> = <code>-1</code> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<p>< Layout function </p>
|
|
<p>< If non-negative, specifies number of fractional bits that are not truncated to zero. Permits reducing precision of data. </p>
|
|
<dl class="params"><dt>Parameters</dt><dd>
|
|
<table class="params">
|
|
<tr><td class="paramname">dst</td><td>destination tensor </td></tr>
|
|
<tr><td class="paramname">seed</td><td>seed for RNG </td></tr>
|
|
<tr><td class="paramname">mean</td><td>Gaussian distribution's mean </td></tr>
|
|
<tr><td class="paramname">stddev</td><td>Gaussian distribution's standard deviation </td></tr>
|
|
</table>
|
|
</dd>
|
|
</dl>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="a8f1500a3e2cb694323bf9d25bafc0fbc"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename Element , typename Layout > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::TensorFillRandomUniform </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > </td>
|
|
<td class="paramname"><em>dst</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">uint64_t </td>
|
|
<td class="paramname"><em>seed</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">double </td>
|
|
<td class="paramname"><em>max</em> = <code>1</code>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">double </td>
|
|
<td class="paramname"><em>min</em> = <code>0</code>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">int </td>
|
|
<td class="paramname"><em>bits</em> = <code>-1</code> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<p>< Layout function </p>
|
|
<p>< If non-negative, specifies number of fractional bits that are not truncated to zero. Permits reducing precision of data. </p>
|
|
<dl class="params"><dt>Parameters</dt><dd>
|
|
<table class="params">
|
|
<tr><td class="paramname">dst</td><td>destination tensor </td></tr>
|
|
<tr><td class="paramname">seed</td><td>seed for RNG </td></tr>
|
|
<tr><td class="paramname">max</td><td>upper bound of distribution </td></tr>
|
|
<tr><td class="paramname">min</td><td>lower bound for distribution </td></tr>
|
|
</table>
|
|
</dd>
|
|
</dl>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="ad6e2cd99b8096eaaf79c0a6edbd3e420"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename Element , typename Layout > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::TensorFillSequential </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > </td>
|
|
<td class="paramname"><em>dst</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">Element </td>
|
|
<td class="paramname"><em>s</em> = <code>Element(0)</code> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<p>< Layout function </p>
|
|
<dl class="params"><dt>Parameters</dt><dd>
|
|
<table class="params">
|
|
<tr><td class="paramname">dst</td><td>destination tensor </td></tr>
|
|
</table>
|
|
</dd>
|
|
</dl>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="aa92b096c37d8a366390c368d17f21b85"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename Element , typename Layout > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">std::pair<bool, <a class="el" href="structcutlass_1_1Coord.html">Coord</a><Layout::kRank> > cutlass::reference::host::TensorFind </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > const & </td>
|
|
<td class="paramname"><em>view</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">Element </td>
|
|
<td class="paramname"><em>value</em> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<p>Returns a pair containing a boolean of whether a value exists in a tensor and the location of the first occurrence. If the value is not contained in the tensor, the second element of the pair is undefined. </p>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="a8c798c04df572b34e3ed3976d69f993d"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename Func , int Rank> </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::TensorForEach </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1Coord.html">Coord</a>< Rank > </td>
|
|
<td class="paramname"><em>extent</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">Func & </td>
|
|
<td class="paramname"><em>func</em> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<p>< rank of index space </p>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="a3825b1aaaf5e5abf0de5f427e3481ada"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename Func , int Rank> </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::TensorForEachLambda </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="structcutlass_1_1Coord.html">Coord</a>< Rank > </td>
|
|
<td class="paramname"><em>extent</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">Func </td>
|
|
<td class="paramname"><em>func</em> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<p>< rank of index space </p>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="ac9ea3d1c0a265daf9f43c4d32b2d3d4c"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename ElementD , typename LayoutD , typename ElementA , typename LayoutA , typename ElementB , typename LayoutB > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::TensorModulus </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< ElementD, LayoutD > </td>
|
|
<td class="paramname"><em>d</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementA, LayoutA > </td>
|
|
<td class="paramname"><em>a</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementB, LayoutB > </td>
|
|
<td class="paramname"><em>b</em> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<dl class="params"><dt>Parameters</dt><dd>
|
|
<table class="params">
|
|
<tr><td class="paramname">d</td><td>destination tensor view </td></tr>
|
|
<tr><td class="paramname">a</td><td>A tensor reference </td></tr>
|
|
<tr><td class="paramname">b</td><td>B tensor reference </td></tr>
|
|
</table>
|
|
</dd>
|
|
</dl>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="a50b72c70c26ff46dff8747010e3a92b9"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename ElementD , typename LayoutD , typename ElementA , typename LayoutA > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::TensorModulus </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< ElementD, LayoutD > </td>
|
|
<td class="paramname"><em>d</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementA, LayoutA > </td>
|
|
<td class="paramname"><em>a</em> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<dl class="params"><dt>Parameters</dt><dd>
|
|
<table class="params">
|
|
<tr><td class="paramname">d</td><td>destination tensor view </td></tr>
|
|
<tr><td class="paramname">a</td><td>A tensor reference </td></tr>
|
|
</table>
|
|
</dd>
|
|
</dl>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="affc066e432cb8213ce84a81371623f7e"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename ElementD , typename LayoutD , typename ElementA , typename LayoutA , typename ElementB , typename LayoutB > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::TensorMul </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< ElementD, LayoutD > </td>
|
|
<td class="paramname"><em>d</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementA, LayoutA > </td>
|
|
<td class="paramname"><em>a</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementB, LayoutB > </td>
|
|
<td class="paramname"><em>b</em> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<dl class="params"><dt>Parameters</dt><dd>
|
|
<table class="params">
|
|
<tr><td class="paramname">d</td><td>destination tensor view </td></tr>
|
|
<tr><td class="paramname">a</td><td>A tensor reference </td></tr>
|
|
<tr><td class="paramname">b</td><td>B tensor reference </td></tr>
|
|
</table>
|
|
</dd>
|
|
</dl>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="ac11f9fadfa4526ba8e6fa62f806eae7c"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename ElementD , typename LayoutD , typename ElementA , typename LayoutA > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::TensorMul </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< ElementD, LayoutD > </td>
|
|
<td class="paramname"><em>d</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementA, LayoutA > </td>
|
|
<td class="paramname"><em>a</em> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<dl class="params"><dt>Parameters</dt><dd>
|
|
<table class="params">
|
|
<tr><td class="paramname">d</td><td>destination tensor view </td></tr>
|
|
<tr><td class="paramname">a</td><td>A tensor reference </td></tr>
|
|
</table>
|
|
</dd>
|
|
</dl>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="a0995c2093dcb952de0a8b9fb199352a1"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename Element , typename Layout , typename ElementReduction > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">ElementReduction cutlass::reference::host::TensorNorm </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > </td>
|
|
<td class="paramname"><em>view</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">ElementReduction </td>
|
|
<td class="paramname"><em>accumulator</em> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="a8e1741fda1d41ab51a960e5f6fe5b59e"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename Element , typename Layout > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">double cutlass::reference::host::TensorNorm </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > </td>
|
|
<td class="paramname"><em>view</em></td><td>)</td>
|
|
<td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="a8fecbd766b668983848255e0e2532e55"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename Element , typename Layout > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">bool cutlass::reference::host::TensorNotEquals </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > const & </td>
|
|
<td class="paramname"><em>lhs</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > const & </td>
|
|
<td class="paramname"><em>rhs</em> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<p>< Layout function </p>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="a3b2eede84fbc860b8f0d178c14e4cbb1"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename ElementD , typename LayoutD , typename ElementA , typename LayoutA , typename ElementB , typename LayoutB > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::TensorSub </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< ElementD, LayoutD > </td>
|
|
<td class="paramname"><em>d</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementA, LayoutA > </td>
|
|
<td class="paramname"><em>a</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementB, LayoutB > </td>
|
|
<td class="paramname"><em>b</em> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<dl class="params"><dt>Parameters</dt><dd>
|
|
<table class="params">
|
|
<tr><td class="paramname">d</td><td>destination tensor view </td></tr>
|
|
<tr><td class="paramname">a</td><td>A tensor reference </td></tr>
|
|
<tr><td class="paramname">b</td><td>B tensor reference </td></tr>
|
|
</table>
|
|
</dd>
|
|
</dl>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="a30294715f2c97a2f3f8adca4a16a97ff"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename ElementD , typename LayoutD , typename ElementA , typename LayoutA , typename ElementB , typename LayoutB > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::TensorSub </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< ElementD, LayoutD > </td>
|
|
<td class="paramname"><em>d</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorRef.html">TensorRef</a>< ElementA, LayoutA > </td>
|
|
<td class="paramname"><em>a</em> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<dl class="params"><dt>Parameters</dt><dd>
|
|
<table class="params">
|
|
<tr><td class="paramname">d</td><td>destination tensor view </td></tr>
|
|
<tr><td class="paramname">a</td><td>A tensor reference </td></tr>
|
|
</table>
|
|
</dd>
|
|
</dl>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="acbf747241e8ac6ef9b1702b735a7913e"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename Element , typename Layout > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::TensorUpdateDiagonal </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > </td>
|
|
<td class="paramname"><em>dst</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">Element </td>
|
|
<td class="paramname"><em>val</em> = <code>Element(1)</code> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<p>< Layout function </p>
|
|
<dl class="params"><dt>Parameters</dt><dd>
|
|
<table class="params">
|
|
<tr><td class="paramname">dst</td><td>destination tensor </td></tr>
|
|
</table>
|
|
</dd>
|
|
</dl>
|
|
|
|
</div>
|
|
</div>
|
|
<a class="anchor" id="a6d146dc0390e4c045f08b5d2adfcf48a"></a>
|
|
<div class="memitem">
|
|
<div class="memproto">
|
|
<div class="memtemplate">
|
|
template<typename Element , typename Layout > </div>
|
|
<table class="memname">
|
|
<tr>
|
|
<td class="memname">void cutlass::reference::host::TensorUpdateOffDiagonal </td>
|
|
<td>(</td>
|
|
<td class="paramtype"><a class="el" href="classcutlass_1_1TensorView.html">TensorView</a>< Element, Layout > </td>
|
|
<td class="paramname"><em>dst</em>, </td>
|
|
</tr>
|
|
<tr>
|
|
<td class="paramkey"></td>
|
|
<td></td>
|
|
<td class="paramtype">Element </td>
|
|
<td class="paramname"><em>other</em> = <code>Element(1)</code> </td>
|
|
</tr>
|
|
<tr>
|
|
<td></td>
|
|
<td>)</td>
|
|
<td></td><td></td>
|
|
</tr>
|
|
</table>
|
|
</div><div class="memdoc">
|
|
<p>< Layout function </p>
|
|
<dl class="params"><dt>Parameters</dt><dd>
|
|
<table class="params">
|
|
<tr><td class="paramname">dst</td><td>destination tensor </td></tr>
|
|
</table>
|
|
</dd>
|
|
</dl>
|
|
|
|
</div>
|
|
</div>
|
|
</div><!-- contents -->
|
|
<!-- start footer part -->
|
|
<hr class="footer"/><address class="footer"><small>
|
|
Generated by  <a href="http://www.doxygen.org/index.html">
|
|
<img class="footer" src="doxygen.png" alt="doxygen"/>
|
|
</a> 1.8.11
|
|
</small></address>
|
|
</body>
|
|
</html>
|