162 lines
149 KiB
HTML
162 lines
149 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: predicated_tile_iterator_2dthreadtile.h Source File</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><a href="namespaces.html"><span>Namespaces</span></a></li>
|
|
<li><a href="annotated.html"><span>Classes</span></a></li>
|
|
<li class="current"><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="files.html"><span>File List</span></a></li>
|
|
<li><a href="globals.html"><span>File 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="dir_d44c64559bbebec7f509842c48db8b23.html">include</a></li><li class="navelem"><a class="el" href="dir_6baf2bb612a2f0daa69af3101ede80a1.html">cutlass</a></li><li class="navelem"><a class="el" href="dir_568e97a0eb81cc0d3daf98cef30c9135.html">transform</a></li><li class="navelem"><a class="el" href="dir_5a68e39c181f2defa4dd959f7500739b.html">threadblock</a></li> </ul>
|
|
</div>
|
|
</div><!-- top -->
|
|
<div class="header">
|
|
<div class="headertitle">
|
|
<div class="title">predicated_tile_iterator_2dthreadtile.h</div> </div>
|
|
</div><!--header-->
|
|
<div class="contents">
|
|
<a href="predicated__tile__iterator__2dthreadtile_8h.html">Go to the documentation of this file.</a><div class="fragment"><div class="line"><a name="l00001"></a><span class="lineno"> 1</span> <span class="comment">/***************************************************************************************************</span></div><div class="line"><a name="l00002"></a><span class="lineno"> 2</span> <span class="comment"> * Copyright (c) 2017-2019, NVIDIA CORPORATION. All rights reserved.</span></div><div class="line"><a name="l00003"></a><span class="lineno"> 3</span> <span class="comment"> *</span></div><div class="line"><a name="l00004"></a><span class="lineno"> 4</span> <span class="comment"> * Redistribution and use in source and binary forms, with or without modification, are permitted</span></div><div class="line"><a name="l00005"></a><span class="lineno"> 5</span> <span class="comment"> * provided that the following conditions are met:</span></div><div class="line"><a name="l00006"></a><span class="lineno"> 6</span> <span class="comment"> * * Redistributions of source code must retain the above copyright notice, this list of</span></div><div class="line"><a name="l00007"></a><span class="lineno"> 7</span> <span class="comment"> * conditions and the following disclaimer.</span></div><div class="line"><a name="l00008"></a><span class="lineno"> 8</span> <span class="comment"> * * Redistributions in binary form must reproduce the above copyright notice, this list of</span></div><div class="line"><a name="l00009"></a><span class="lineno"> 9</span> <span class="comment"> * conditions and the following disclaimer in the documentation and/or other materials</span></div><div class="line"><a name="l00010"></a><span class="lineno"> 10</span> <span class="comment"> * provided with the distribution.</span></div><div class="line"><a name="l00011"></a><span class="lineno"> 11</span> <span class="comment"> * * Neither the name of the NVIDIA CORPORATION nor the names of its contributors may be used</span></div><div class="line"><a name="l00012"></a><span class="lineno"> 12</span> <span class="comment"> * to endorse or promote products derived from this software without specific prior written</span></div><div class="line"><a name="l00013"></a><span class="lineno"> 13</span> <span class="comment"> * permission.</span></div><div class="line"><a name="l00014"></a><span class="lineno"> 14</span> <span class="comment"> *</span></div><div class="line"><a name="l00015"></a><span class="lineno"> 15</span> <span class="comment"> * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR</span></div><div class="line"><a name="l00016"></a><span class="lineno"> 16</span> <span class="comment"> * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND</span></div><div class="line"><a name="l00017"></a><span class="lineno"> 17</span> <span class="comment"> * FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL NVIDIA CORPORATION BE LIABLE</span></div><div class="line"><a name="l00018"></a><span class="lineno"> 18</span> <span class="comment"> * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,</span></div><div class="line"><a name="l00019"></a><span class="lineno"> 19</span> <span class="comment"> * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;</span></div><div class="line"><a name="l00020"></a><span class="lineno"> 20</span> <span class="comment"> * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,</span></div><div class="line"><a name="l00021"></a><span class="lineno"> 21</span> <span class="comment"> * STRICT LIABILITY, OR TOR (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE</span></div><div class="line"><a name="l00022"></a><span class="lineno"> 22</span> <span class="comment"> * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.</span></div><div class="line"><a name="l00023"></a><span class="lineno"> 23</span> <span class="comment"> *</span></div><div class="line"><a name="l00024"></a><span class="lineno"> 24</span> <span class="comment"> **************************************************************************************************/</span></div><div class="line"><a name="l00035"></a><span class="lineno"> 35</span> <span class="preprocessor">#pragma once</span></div><div class="line"><a name="l00036"></a><span class="lineno"> 36</span> </div><div class="line"><a name="l00037"></a><span class="lineno"> 37</span> <span class="preprocessor">#include "<a class="code" href="predicated__tile__access__iterator__2dthreadtile_8h.html">cutlass/transform/threadblock/predicated_tile_access_iterator_2dthreadtile.h</a>"</span></div><div class="line"><a name="l00038"></a><span class="lineno"> 38</span> <span class="preprocessor">#include "<a class="code" href="transpose_8h.html">cutlass/transform/thread/transpose.h</a>"</span></div><div class="line"><a name="l00039"></a><span class="lineno"> 39</span> </div><div class="line"><a name="l00041"></a><span class="lineno"> 41</span> </div><div class="line"><a name="l00042"></a><span class="lineno"> 42</span> <span class="keyword">namespace </span><a class="code" href="namespacecutlass.html">cutlass</a> {</div><div class="line"><a name="l00043"></a><span class="lineno"> 43</span> <span class="keyword">namespace </span>transform {</div><div class="line"><a name="l00044"></a><span class="lineno"> 44</span> <span class="keyword">namespace </span>threadblock {</div><div class="line"><a name="l00045"></a><span class="lineno"> 45</span> </div><div class="line"><a name="l00047"></a><span class="lineno"> 47</span> </div><div class="line"><a name="l00084"></a><span class="lineno"> 84</span> <span class="comment">// template <typename Iterator></span></div><div class="line"><a name="l00085"></a><span class="lineno"> 85</span> <span class="comment">// __global__ void kernel(</span></div><div class="line"><a name="l00086"></a><span class="lineno"> 86</span> <span class="comment">// typename Iterator::Params params, </span></div><div class="line"><a name="l00087"></a><span class="lineno"> 87</span> <span class="comment">// typename Iterator::Element *ptr,</span></div><div class="line"><a name="l00088"></a><span class="lineno"> 88</span> <span class="comment">// TensorCoord extent) {</span></div><div class="line"><a name="l00089"></a><span class="lineno"> 89</span> <span class="comment">//</span></div><div class="line"><a name="l00090"></a><span class="lineno"> 90</span> <span class="comment">// typename Iterator::Fragment fragment;</span></div><div class="line"><a name="l00091"></a><span class="lineno"> 91</span> <span class="comment">//</span></div><div class="line"><a name="l00092"></a><span class="lineno"> 92</span> <span class="comment">// TensorCoord threadblock_offset(0, 0);</span></div><div class="line"><a name="l00093"></a><span class="lineno"> 93</span> <span class="comment">//</span></div><div class="line"><a name="l00094"></a><span class="lineno"> 94</span> <span class="comment">// Iterator iter(params, ptr, extent, threadIdx.x, threadblock_offsets);</span></div><div class="line"><a name="l00095"></a><span class="lineno"> 95</span> <span class="comment">//</span></div><div class="line"><a name="l00096"></a><span class="lineno"> 96</span> <span class="comment">//</span></div><div class="line"><a name="l00097"></a><span class="lineno"> 97</span> <span class="comment">// fragment = *iter; // load "residue" tile first</span></div><div class="line"><a name="l00098"></a><span class="lineno"> 98</span> <span class="comment">// ++iter; // advance to first "steady state" tile and update internal masks</span></div><div class="line"><a name="l00099"></a><span class="lineno"> 99</span> <span class="comment">//</span></div><div class="line"><a name="l00100"></a><span class="lineno"> 100</span> <span class="comment">//</span></div><div class="line"><a name="l00101"></a><span class="lineno"> 101</span> <span class="comment">// #pragma unroll</span></div><div class="line"><a name="l00102"></a><span class="lineno"> 102</span> <span class="comment">// for (int i = Remaining - 1; i >= 0; --i) {</span></div><div class="line"><a name="l00103"></a><span class="lineno"> 103</span> <span class="comment">//</span></div><div class="line"><a name="l00104"></a><span class="lineno"> 104</span> <span class="comment">// f(fragment);</span></div><div class="line"><a name="l00105"></a><span class="lineno"> 105</span> <span class="comment">//</span></div><div class="line"><a name="l00106"></a><span class="lineno"> 106</span> <span class="comment">// if (!i) {</span></div><div class="line"><a name="l00107"></a><span class="lineno"> 107</span> <span class="comment">// iter.clear_mask(); // light-weight operation to clear masks - subsequent loads become NO-OPs.</span></div><div class="line"><a name="l00108"></a><span class="lineno"> 108</span> <span class="comment">// }</span></div><div class="line"><a name="l00109"></a><span class="lineno"> 109</span> <span class="comment">// </span></div><div class="line"><a name="l00110"></a><span class="lineno"> 110</span> <span class="comment">// fragment = *iter; // load tile during "steady state" phase</span></div><div class="line"><a name="l00111"></a><span class="lineno"> 111</span> <span class="comment">// ++iter; // advance to next tile - lightweight due to steady-state masks</span></div><div class="line"><a name="l00112"></a><span class="lineno"> 112</span> <span class="comment">// }</span></div><div class="line"><a name="l00113"></a><span class="lineno"> 113</span> <span class="comment">// }</span></div><div class="line"><a name="l00114"></a><span class="lineno"> 114</span> <span class="comment">//</span></div><div class="line"><a name="l00115"></a><span class="lineno"> 115</span> <span class="comment">// void host(TensorView<Element, 2, layout::PitchLinear> view) {</span></div><div class="line"><a name="l00116"></a><span class="lineno"> 116</span> <span class="comment">//</span></div><div class="line"><a name="l00117"></a><span class="lineno"> 117</span> <span class="comment">// using Iterator = transform::threadblock::PredicatedTileIterator2dThreadTile;</span></div><div class="line"><a name="l00118"></a><span class="lineno"> 118</span> <span class="comment">//</span></div><div class="line"><a name="l00119"></a><span class="lineno"> 119</span> <span class="comment">// typename Iterator::Params params(view.layout());</span></div><div class="line"><a name="l00120"></a><span class="lineno"> 120</span> <span class="comment">//</span></div><div class="line"><a name="l00121"></a><span class="lineno"> 121</span> <span class="comment">// kernel<Iterator>(params, view.data());</span></div><div class="line"><a name="l00122"></a><span class="lineno"> 122</span> <span class="comment">// }</span></div><div class="line"><a name="l00125"></a><span class="lineno"> 125</span> <span class="comment"></span><span class="keyword">template</span> <</div><div class="line"><a name="l00126"></a><span class="lineno"> 126</span>  <span class="keyword">typename</span> Shape,</div><div class="line"><a name="l00127"></a><span class="lineno"> 127</span>  <span class="keyword">typename</span> Element,</div><div class="line"><a name="l00128"></a><span class="lineno"> 128</span>  <span class="keyword">typename</span> Layout,</div><div class="line"><a name="l00129"></a><span class="lineno"> 129</span>  <span class="keywordtype">int</span> AdvanceRank,</div><div class="line"><a name="l00130"></a><span class="lineno"> 130</span>  <span class="keyword">typename</span> ThreadMap,</div><div class="line"><a name="l00131"></a><span class="lineno"> 131</span>  <span class="keywordtype">bool</span> Transpose = <span class="keyword">false</span></div><div class="line"><a name="l00132"></a><span class="lineno"> 132</span> ></div><div class="line"><a name="l00133"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile.html"> 133</a></span> <span class="keyword">class </span><a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile.html">PredicatedTileIterator2dThreadTile</a>;</div><div class="line"><a name="l00134"></a><span class="lineno"> 134</span> </div><div class="line"><a name="l00136"></a><span class="lineno"> 136</span> </div><div class="line"><a name="l00144"></a><span class="lineno"> 144</span> <span class="keyword">template</span> <<span class="keyword">typename</span> Shape_, <span class="keyword">typename</span> Element_, <span class="keywordtype">int</span> AdvanceRank, <span class="keyword">typename</span> ThreadMap_, <span class="keywordtype">bool</span> Transpose_></div><div class="line"><a name="l00145"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html"> 145</a></span> <span class="keyword">class </span><a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile.html">PredicatedTileIterator2dThreadTile</a><Shape_, Element_, layout::PitchLinear, AdvanceRank, ThreadMap_, Transpose_> {</div><div class="line"><a name="l00146"></a><span class="lineno"> 146</span>  <span class="keyword">public</span>:</div><div class="line"><a name="l00147"></a><span class="lineno"> 147</span>  <a class="code" href="platform_8h.html#adde4c9ea91b753491851361a4198c009">static_assert</a>(</div><div class="line"><a name="l00148"></a><span class="lineno"> 148</span>  AdvanceRank == 0 || AdvanceRank == 1,</div><div class="line"><a name="l00149"></a><span class="lineno"> 149</span>  <span class="stringliteral">"Specialization for pitch-linear iterator may along advance along the "</span></div><div class="line"><a name="l00150"></a><span class="lineno"> 150</span>  <span class="stringliteral">"contiguous(rank=0) or strided(rank=1) dimension."</span>);</div><div class="line"><a name="l00151"></a><span class="lineno"> 151</span> </div><div class="line"><a name="l00152"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a837b46168668a29afce4f36f5c4c598f"> 152</a></span>  <span class="keyword">using</span> <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a837b46168668a29afce4f36f5c4c598f">Shape</a> = Shape_;</div><div class="line"><a name="l00153"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#ad3370aac1cc87c44babbc34f3c9d1d7b"> 153</a></span>  <span class="keyword">using</span> Element = Element_;</div><div class="line"><a name="l00154"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a412ae2569a2339172a1facddd9c9fcb3"> 154</a></span>  <span class="keyword">using</span> Layout = <a class="code" href="classcutlass_1_1layout_1_1PitchLinear.html">layout::PitchLinear</a>;</div><div class="line"><a name="l00155"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a8ea7d8c5b2d1cb79ecec2a5b114c42c7"> 155</a></span>  <span class="keyword">static</span> <span class="keywordtype">int</span> <span class="keyword">const</span> kAdvanceRank = AdvanceRank;</div><div class="line"><a name="l00156"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a56d8fc35653cdaf60e5a1d54df504008"> 156</a></span>  <span class="keyword">using</span> ThreadMap = ThreadMap_;</div><div class="line"><a name="l00157"></a><span class="lineno"> 157</span> </div><div class="line"><a name="l00158"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a715b78e9d8587c4c59e5337d33eb7025"> 158</a></span>  <span class="keyword">using</span> <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a715b78e9d8587c4c59e5337d33eb7025">Index</a> = <span class="keyword">typename</span> <a class="code" href="classcutlass_1_1layout_1_1PitchLinear.html#a9d1dfd7b6d3b2b651009dcba8f5fd5cd">Layout::Index</a>;</div><div class="line"><a name="l00159"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a560fc3bce354a657f12b423969eb5b37"> 159</a></span>  <span class="keyword">using</span> <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a560fc3bce354a657f12b423969eb5b37">LongIndex</a> = <span class="keyword">typename</span> <a class="code" href="classcutlass_1_1layout_1_1PitchLinear.html#a1a4b31740e77b3c03925f507650978ea">Layout::LongIndex</a>;</div><div class="line"><a name="l00160"></a><span class="lineno"> 160</span> </div><div class="line"><a name="l00161"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#ae6c24e1b5ca733c12766da023a24c122"> 161</a></span>  <span class="keyword">using</span> <a class="code" href="classcutlass_1_1TensorRef.html">TensorRef</a> = <a class="code" href="classcutlass_1_1TensorRef.html">TensorRef<Element, Layout></a>;</div><div class="line"><a name="l00162"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a831fe297019ef0d80ec2863f931712bd"> 162</a></span>  <span class="keyword">using</span> <a class="code" href="classcutlass_1_1TensorView.html">TensorView</a> = <a class="code" href="classcutlass_1_1TensorView.html">TensorView<Element, Layout></a>;</div><div class="line"><a name="l00163"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a587fc2ae48d8793d5e563249242c16fa"> 163</a></span>  <span class="keyword">using</span> <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a587fc2ae48d8793d5e563249242c16fa">TensorCoord</a> = <span class="keyword">typename</span> <a class="code" href="structcutlass_1_1layout_1_1PitchLinearCoord.html">Layout::TensorCoord</a>;</div><div class="line"><a name="l00164"></a><span class="lineno"> 164</span> </div><div class="line"><a name="l00165"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#aeae69750c7c478c23653af6e785892af"> 165</a></span>  <span class="keyword">using</span> <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#aeae69750c7c478c23653af6e785892af">Pointer</a> = Element *;</div><div class="line"><a name="l00166"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a1065b487ba57b47e4d03442a91af8f67"> 166</a></span>  <span class="keyword">using</span> <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a1065b487ba57b47e4d03442a91af8f67">NonConstPointer</a> = <span class="keyword">typename</span> <a class="code" href="structcutlass_1_1platform_1_1remove__const.html#ac3662947fa50251daf58240a9c798085">platform::remove_const<Element>::type</a> *;</div><div class="line"><a name="l00167"></a><span class="lineno"> 167</span> </div><div class="line"><a name="l00170"></a><span class="lineno"><a class="line" href="structcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0b878062cc0cd214bf7e17d74ff17e246.html"> 170</a></span>  <span class="keyword">struct </span><span class="keyword">alignas</span>((ThreadMap::kElementsPerAccess * sizeof_bits<Element>::value /</div><div class="line"><a name="l00172"></a><span class="lineno"> 172</span>  8)) AccessType {</div><div class="line"><a name="l00173"></a><span class="lineno"><a class="line" href="structcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0b878062cc0cd214bf7e17d74ff17e246.html#aa973b588e9d9ed9065d5e1e86eb2a3ea"> 173</a></span> </div><div class="line"><a name="l00174"></a><span class="lineno"> 174</span>  Array<Element, ThreadMap::kElementsPerAccess> storage;</div><div class="line"><a name="l00175"></a><span class="lineno"><a class="line" href="structcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0b878062cc0cd214bf7e17d74ff17e246.html#a7e9c43dcd79254f5b71bec6a67c5d08b"> 175</a></span> </div><div class="line"><a name="l00176"></a><span class="lineno"> 176</span>  <span class="keyword">static</span> <span class="keywordtype">int</span> <span class="keyword">const</span> kElements = ThreadMap::kElementsPerAccess;</div><div class="line"><a name="l00177"></a><span class="lineno"> 177</span>  };</div><div class="line"><a name="l00178"></a><span class="lineno"> 178</span> </div><div class="line"><a name="l00180"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#ac2e6fecd986cc087c0c4a6087aa29e88"> 180</a></span>  <span class="keyword">using</span> <a class="code" href="classcutlass_1_1transform_1_1thread_1_1Transpose.html">Transform</a> = <a class="code" href="classcutlass_1_1transform_1_1thread_1_1Transpose.html">thread::Transpose< ThreadMap::Iterations::kCount * ThreadMap::ThreadAccessShape::kCount , layout::PitchLinearShape<4,4></a>, Element>;</div><div class="line"><a name="l00181"></a><span class="lineno"> 181</span>  <span class="keyword">static</span> <span class="keywordtype">bool</span> <span class="keyword">const</span> transpose = Transpose_;</div><div class="line"><a name="l00182"></a><span class="lineno"> 182</span> </div><div class="line"><a name="l00184"></a><span class="lineno"> 184</span>  <span class="keyword">using</span> <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileAccessIterator2dThreadTile.html">TileAccessIterator</a> =</div><div class="line"><a name="l00185"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a8417f78d0d4278304d0ef595094d06bc"> 185</a></span>  <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileAccessIterator2dThreadTile.html">PredicatedTileAccessIterator2dThreadTile</a><<a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a837b46168668a29afce4f36f5c4c598f">Shape</a>, Element, Layout, kAdvanceRank,</div><div class="line"><a name="l00186"></a><span class="lineno"> 186</span>  ThreadMap, AccessType>;</div><div class="line"><a name="l00187"></a><span class="lineno"> 187</span> </div><div class="line"><a name="l00189"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a0c7351825135690fa82c4fba5f9447c7"> 189</a></span>  <span class="keyword">using</span> <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a0c7351825135690fa82c4fba5f9447c7">Fragment</a> = cutlass::Array<Element, ThreadMap::Iterations::kCount *</div><div class="line"><a name="l00190"></a><span class="lineno"> 190</span>  ThreadMap::ThreadAccessShape::kCount>;</div><div class="line"><a name="l00191"></a><span class="lineno"> 191</span> </div><div class="line"><a name="l00193"></a><span class="lineno"> 193</span>  <span class="keyword">using</span> <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a20a0f1a208b301ecd6f29cced4f7cbb7">Mask</a> = <span class="keyword">typename</span> TileAccessIterator::Mask;</div><div class="line"><a name="l00194"></a><span class="lineno"> 194</span> </div><div class="line"><a name="l00196"></a><span class="lineno"> 196</span>  <span class="keyword">class </span>Params {</div><div class="line"><a name="l00197"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0145ef045e8f7d57dc718098adcb00cf3d.html#a3f0c8d2081d8d23cd6cb81627a77390a"> 197</a></span>  <span class="keyword">public</span>:</div><div class="line"><a name="l00198"></a><span class="lineno"> 198</span>  <span class="keyword">friend</span> <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile.html">PredicatedTileIterator2dThreadTile</a>;</div><div class="line"><a name="l00199"></a><span class="lineno"> 199</span> </div><div class="line"><a name="l00200"></a><span class="lineno"> 200</span>  <span class="keyword">private</span>:</div><div class="line"><a name="l00202"></a><span class="lineno"> 202</span>  <span class="keyword">typename</span> TileAccessIterator::Params params_;</div><div class="line"><a name="l00203"></a><span class="lineno"> 203</span> </div><div class="line"><a name="l00204"></a><span class="lineno"> 204</span>  <span class="keyword">public</span>:</div><div class="line"><a name="l00206"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0145ef045e8f7d57dc718098adcb00cf3d.html#ad7de484ae6fd7e295771003ff2dc155f"> 206</a></span>  <a class="code" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a></div><div class="line"><a name="l00207"></a><span class="lineno"> 207</span>  Params(Layout <span class="keyword">const</span> &layout) : params_(layout) { }</div><div class="line"><a name="l00208"></a><span class="lineno"> 208</span>  </div><div class="line"><a name="l00209"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0145ef045e8f7d57dc718098adcb00cf3d.html#abb10040d141b685cdbabd41f3e222aa4"> 209</a></span>  <a class="code" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a></div><div class="line"><a name="l00210"></a><span class="lineno"> 210</span>  Params() { }</div><div class="line"><a name="l00211"></a><span class="lineno"> 211</span>  };</div><div class="line"><a name="l00212"></a><span class="lineno"> 212</span> </div><div class="line"><a name="l00213"></a><span class="lineno"> 213</span>  <span class="keyword">private</span>:</div><div class="line"><a name="l00215"></a><span class="lineno"> 215</span>  <span class="keyword">using</span> BytePointer = <span class="keywordtype">char</span> *;</div><div class="line"><a name="l00216"></a><span class="lineno"> 216</span> </div><div class="line"><a name="l00217"></a><span class="lineno"> 217</span>  <span class="keyword">private</span>:</div><div class="line"><a name="l00218"></a><span class="lineno"> 218</span>  <span class="comment">//</span></div><div class="line"><a name="l00219"></a><span class="lineno"> 219</span>  <span class="comment">// Data members</span></div><div class="line"><a name="l00220"></a><span class="lineno"> 220</span>  <span class="comment">//</span></div><div class="line"><a name="l00221"></a><span class="lineno"> 221</span> </div><div class="line"><a name="l00223"></a><span class="lineno"> 223</span>  <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileAccessIterator2dThreadTile.html">TileAccessIterator</a> address_iterator_;</div><div class="line"><a name="l00224"></a><span class="lineno"> 224</span> </div><div class="line"><a name="l00225"></a><span class="lineno"> 225</span>  <span class="keyword">public</span>:</div><div class="line"><a name="l00228"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#aef7f87f11e85c20cfedb2c15c1bcdc1b"> 228</a></span>  <a class="code" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a></div><div class="line"><a name="l00229"></a><span class="lineno"> 229</span>  <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile.html">PredicatedTileIterator2dThreadTile</a>(</div><div class="line"><a name="l00231"></a><span class="lineno"> 231</span>  Params <span class="keyword">const</span> &params,</div><div class="line"><a name="l00233"></a><span class="lineno"> 233</span>  <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#aeae69750c7c478c23653af6e785892af">Pointer</a> pointer,</div><div class="line"><a name="l00235"></a><span class="lineno"> 235</span>  <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a587fc2ae48d8793d5e563249242c16fa">TensorCoord</a> extent,</div><div class="line"><a name="l00237"></a><span class="lineno"> 237</span>  <span class="keywordtype">int</span> thread_id,</div><div class="line"><a name="l00239"></a><span class="lineno"> 239</span>  <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a587fc2ae48d8793d5e563249242c16fa">TensorCoord</a> <span class="keyword">const</span> &threadblock_offset)</div><div class="line"><a name="l00240"></a><span class="lineno"> 240</span>  : address_iterator_(params.params_, pointer, extent, thread_id,</div><div class="line"><a name="l00241"></a><span class="lineno"> 241</span>  threadblock_offset) {}</div><div class="line"><a name="l00242"></a><span class="lineno"> 242</span> </div><div class="line"><a name="l00244"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a344627cd2525bf07e4c631d2cc3ca1d9"> 244</a></span>  <a class="code" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a></div><div class="line"><a name="l00245"></a><span class="lineno"> 245</span>  <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile.html">PredicatedTileIterator2dThreadTile</a>(</div><div class="line"><a name="l00246"></a><span class="lineno"> 246</span>  Params <span class="keyword">const</span> &params, </div><div class="line"><a name="l00247"></a><span class="lineno"> 247</span>  <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#aeae69750c7c478c23653af6e785892af">Pointer</a> pointer, </div><div class="line"><a name="l00248"></a><span class="lineno"> 248</span>  <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a587fc2ae48d8793d5e563249242c16fa">TensorCoord</a> extent, </div><div class="line"><a name="l00249"></a><span class="lineno"> 249</span>  <span class="keywordtype">int</span> thread_id </div><div class="line"><a name="l00250"></a><span class="lineno"> 250</span>  )</div><div class="line"><a name="l00251"></a><span class="lineno"> 251</span>  : PredicatedTileIterator2dThreadTile(params, pointer, extent, thread_id,</div><div class="line"><a name="l00252"></a><span class="lineno"> 252</span>  <a class="code" href="namespacecutlass.html#a7419519fa453a121dfa5f26bf87318d9">make_Coord</a>(0, 0)) {}</div><div class="line"><a name="l00253"></a><span class="lineno"> 253</span> </div><div class="line"><a name="l00255"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a760970f90ee5c75ff49fc4e8a14aea0a"> 255</a></span>  <a class="code" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a></div><div class="line"><a name="l00256"></a><span class="lineno"> 256</span>  <span class="keywordtype">void</span> add_pointer_offset(<a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a560fc3bce354a657f12b423969eb5b37">LongIndex</a> pointer_offset) {</div><div class="line"><a name="l00257"></a><span class="lineno"> 257</span>  address_iterator_.add_pointer_offset(pointer_offset);</div><div class="line"><a name="l00258"></a><span class="lineno"> 258</span>  }</div><div class="line"><a name="l00259"></a><span class="lineno"> 259</span> </div><div class="line"><a name="l00266"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a928b6720a862da0315fb5fef5da5921f"> 266</a></span>  <a class="code" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a></div><div class="line"><a name="l00267"></a><span class="lineno"> 267</span>  <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile.html">PredicatedTileIterator2dThreadTile</a> &<a class="code" href="namespacecutlass.html#a1c7a9e66ca7b5dc7413ea3b8f349530a">operator++</a>() {</div><div class="line"><a name="l00268"></a><span class="lineno"> 268</span>  <span class="keywordflow">if</span> (kAdvanceRank)</div><div class="line"><a name="l00269"></a><span class="lineno"> 269</span>  address_iterator_.add_tile_offset({0, 1});</div><div class="line"><a name="l00270"></a><span class="lineno"> 270</span>  <span class="keywordflow">else</span></div><div class="line"><a name="l00271"></a><span class="lineno"> 271</span>  address_iterator_.add_tile_offset({1, 0});</div><div class="line"><a name="l00272"></a><span class="lineno"> 272</span> </div><div class="line"><a name="l00273"></a><span class="lineno"> 273</span>  <span class="keywordflow">return</span> *<span class="keyword">this</span>;</div><div class="line"><a name="l00274"></a><span class="lineno"> 274</span>  }</div><div class="line"><a name="l00275"></a><span class="lineno"> 275</span> </div><div class="line"><a name="l00282"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a20bdafdd3866620b4aeddc1a8a2601fe"> 282</a></span>  <a class="code" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a></div><div class="line"><a name="l00283"></a><span class="lineno"> 283</span>  <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile.html">PredicatedTileIterator2dThreadTile</a> <a class="code" href="namespacecutlass.html#a1c7a9e66ca7b5dc7413ea3b8f349530a">operator++</a>(<span class="keywordtype">int</span>) {</div><div class="line"><a name="l00284"></a><span class="lineno"> 284</span>  <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile.html">PredicatedTileIterator2dThreadTile</a> <span class="keyword">self</span>(*this);</div><div class="line"><a name="l00285"></a><span class="lineno"> 285</span>  <a class="code" href="namespacecutlass.html#a1c7a9e66ca7b5dc7413ea3b8f349530a">operator++</a>();</div><div class="line"><a name="l00286"></a><span class="lineno"> 286</span>  <span class="keywordflow">return</span> <span class="keyword">self</span>;</div><div class="line"><a name="l00287"></a><span class="lineno"> 287</span>  }</div><div class="line"><a name="l00288"></a><span class="lineno"> 288</span> </div><div class="line"><a name="l00290"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a24d17e4be377b870202dd8524047af8d"> 290</a></span>  <a class="code" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a></div><div class="line"><a name="l00291"></a><span class="lineno"> 291</span>  <span class="keywordtype">void</span> clear_mask() { address_iterator_.clear_mask(); }</div><div class="line"><a name="l00292"></a><span class="lineno"> 292</span> </div><div class="line"><a name="l00294"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#aab5504906f1b3e4acf25db2c47869961"> 294</a></span>  <a class="code" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a></div><div class="line"><a name="l00295"></a><span class="lineno"> 295</span>  <span class="keywordtype">void</span> enable_mask() { address_iterator_.enable_mask(); }</div><div class="line"><a name="l00296"></a><span class="lineno"> 296</span> </div><div class="line"><a name="l00298"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#adfca5bcfb511847a8c11e74b446c563a"> 298</a></span>  <a class="code" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a></div><div class="line"><a name="l00299"></a><span class="lineno"> 299</span>  <span class="keywordtype">void</span> set_mask(<a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a20a0f1a208b301ecd6f29cced4f7cbb7">Mask</a> <span class="keyword">const</span> &mask) { address_iterator_.set_mask(mask); }</div><div class="line"><a name="l00300"></a><span class="lineno"> 300</span> </div><div class="line"><a name="l00302"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a73852507077f6058a1efe2bd85bb402d"> 302</a></span>  <a class="code" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a></div><div class="line"><a name="l00303"></a><span class="lineno"> 303</span>  <span class="keywordtype">void</span> get_mask(<a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a20a0f1a208b301ecd6f29cced4f7cbb7">Mask</a> &mask) { address_iterator_.get_mask(mask); }</div><div class="line"><a name="l00304"></a><span class="lineno"> 304</span> </div><div class="line"><a name="l00306"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#aed8636c2750ecb8d6901d658017c1fd2"> 306</a></span>  CUTLASS_DEVICE</div><div class="line"><a name="l00307"></a><span class="lineno"> 307</span>  <span class="keywordtype">void</span> load_with_pointer_offset(<a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a0c7351825135690fa82c4fba5f9447c7">Fragment</a> &frag, <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a715b78e9d8587c4c59e5337d33eb7025">Index</a> pointer_offset) {</div><div class="line"><a name="l00308"></a><span class="lineno"> 308</span> </div><div class="line"><a name="l00309"></a><span class="lineno"> 309</span>  AccessType *frag_ptr = <span class="keyword">reinterpret_cast<</span>AccessType *<span class="keyword">></span>(&frag);</div><div class="line"><a name="l00310"></a><span class="lineno"> 310</span> </div><div class="line"><a name="l00311"></a><span class="lineno"> 311</span>  <a class="code" href="cutlass_8h.html#a4b1c9f25ab6eaa25e1f2258dd63e6ce4">CUTLASS_PRAGMA_UNROLL</a></div><div class="line"><a name="l00312"></a><span class="lineno"> 312</span>  <span class="keywordflow">for</span> (<span class="keywordtype">int</span> s = 0; s < ThreadMap::Iterations::kStrided; ++s) {</div><div class="line"><a name="l00313"></a><span class="lineno"> 313</span>  <a class="code" href="cutlass_8h.html#a4b1c9f25ab6eaa25e1f2258dd63e6ce4">CUTLASS_PRAGMA_UNROLL</a></div><div class="line"><a name="l00314"></a><span class="lineno"> 314</span>  <span class="keywordflow">for</span> (<span class="keywordtype">int</span> c = 0; c < ThreadMap::Iterations::kContiguous; ++c) {</div><div class="line"><a name="l00315"></a><span class="lineno"> 315</span>  <a class="code" href="cutlass_8h.html#a4b1c9f25ab6eaa25e1f2258dd63e6ce4">CUTLASS_PRAGMA_UNROLL</a></div><div class="line"><a name="l00316"></a><span class="lineno"> 316</span>  <span class="keywordflow">for</span> (<span class="keywordtype">int</span> ts = 0; ts < ThreadMap::ThreadAccessShape::kStrided; ts++){</div><div class="line"><a name="l00317"></a><span class="lineno"> 317</span> </div><div class="line"><a name="l00318"></a><span class="lineno"> 318</span>  <span class="keywordtype">int</span> access_idx = ts + c * ThreadMap::ThreadAccessShape::kStrided + \</div><div class="line"><a name="l00319"></a><span class="lineno"> 319</span>  s * ThreadMap::Iterations::kContiguous * ThreadMap::ThreadAccessShape::kStrided;</div><div class="line"><a name="l00320"></a><span class="lineno"> 320</span> </div><div class="line"><a name="l00321"></a><span class="lineno"> 321</span>  address_iterator_.set_iteration_index(access_idx);</div><div class="line"><a name="l00322"></a><span class="lineno"> 322</span>  <span class="keywordflow">if</span> (address_iterator_.valid()) {</div><div class="line"><a name="l00323"></a><span class="lineno"> 323</span> </div><div class="line"><a name="l00324"></a><span class="lineno"> 324</span>  frag_ptr[access_idx] =</div><div class="line"><a name="l00325"></a><span class="lineno"> 325</span>  *(address_iterator_.get() + pointer_offset);</div><div class="line"><a name="l00326"></a><span class="lineno"> 326</span>  }</div><div class="line"><a name="l00327"></a><span class="lineno"> 327</span> </div><div class="line"><a name="l00328"></a><span class="lineno"> 328</span>  ++address_iterator_;</div><div class="line"><a name="l00329"></a><span class="lineno"> 329</span>  }</div><div class="line"><a name="l00330"></a><span class="lineno"> 330</span>  }</div><div class="line"><a name="l00331"></a><span class="lineno"> 331</span>  }</div><div class="line"><a name="l00332"></a><span class="lineno"> 332</span> </div><div class="line"><a name="l00333"></a><span class="lineno"> 333</span>  <span class="keywordflow">if</span> (transpose) {</div><div class="line"><a name="l00334"></a><span class="lineno"> 334</span>  <a class="code" href="classcutlass_1_1transform_1_1thread_1_1Transpose.html">Transform</a> t;</div><div class="line"><a name="l00335"></a><span class="lineno"> 335</span>  t.transform(frag, frag);</div><div class="line"><a name="l00336"></a><span class="lineno"> 336</span>  }</div><div class="line"><a name="l00337"></a><span class="lineno"> 337</span>  }</div><div class="line"><a name="l00338"></a><span class="lineno"> 338</span> </div><div class="line"><a name="l00340"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a858a49a0b14d760b3d8d1edf74426183"> 340</a></span>  CUTLASS_DEVICE</div><div class="line"><a name="l00341"></a><span class="lineno"> 341</span>  <span class="keywordtype">void</span> load(<a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a0c7351825135690fa82c4fba5f9447c7">Fragment</a> &frag) { load_with_pointer_offset(frag, 0); }</div><div class="line"><a name="l00342"></a><span class="lineno"> 342</span> </div><div class="line"><a name="l00344"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a96941b886a38cdabe034866fd2db9e94"> 344</a></span>  CUTLASS_DEVICE</div><div class="line"><a name="l00345"></a><span class="lineno"> 345</span>  <span class="keywordtype">void</span> store_with_pointer_offset(<a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a0c7351825135690fa82c4fba5f9447c7">Fragment</a> <span class="keyword">const</span> &frag, <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a715b78e9d8587c4c59e5337d33eb7025">Index</a> pointer_offset) {</div><div class="line"><a name="l00346"></a><span class="lineno"> 346</span>  </div><div class="line"><a name="l00347"></a><span class="lineno"> 347</span>  AccessType <span class="keyword">const</span> *frag_ptr = <span class="keyword">reinterpret_cast<</span>AccessType <span class="keyword">const </span>*<span class="keyword">></span>(&frag);</div><div class="line"><a name="l00348"></a><span class="lineno"> 348</span> </div><div class="line"><a name="l00349"></a><span class="lineno"> 349</span>  <a class="code" href="cutlass_8h.html#a4b1c9f25ab6eaa25e1f2258dd63e6ce4">CUTLASS_PRAGMA_UNROLL</a></div><div class="line"><a name="l00350"></a><span class="lineno"> 350</span>  <span class="keywordflow">for</span> (<span class="keywordtype">int</span> s = 0; s < ThreadMap::Iterations::kStrided; ++s) {</div><div class="line"><a name="l00351"></a><span class="lineno"> 351</span>  <a class="code" href="cutlass_8h.html#a4b1c9f25ab6eaa25e1f2258dd63e6ce4">CUTLASS_PRAGMA_UNROLL</a></div><div class="line"><a name="l00352"></a><span class="lineno"> 352</span>  <span class="keywordflow">for</span> (<span class="keywordtype">int</span> c = 0; c < ThreadMap::Iterations::kContiguous; ++c) {</div><div class="line"><a name="l00353"></a><span class="lineno"> 353</span>  <a class="code" href="cutlass_8h.html#a4b1c9f25ab6eaa25e1f2258dd63e6ce4">CUTLASS_PRAGMA_UNROLL</a></div><div class="line"><a name="l00354"></a><span class="lineno"> 354</span>  <span class="keywordflow">for</span> (<span class="keywordtype">int</span> ts = 0; ts < ThreadMap::ThreadAccessShape::kStrided; ts++){</div><div class="line"><a name="l00355"></a><span class="lineno"> 355</span> </div><div class="line"><a name="l00356"></a><span class="lineno"> 356</span>  <span class="keywordtype">int</span> access_idx = ts + c * ThreadMap::ThreadAccessShape::kStrided + \</div><div class="line"><a name="l00357"></a><span class="lineno"> 357</span>  s * ThreadMap::Iterations::kContiguous * ThreadMap::ThreadAccessShape::kStrided;</div><div class="line"><a name="l00358"></a><span class="lineno"> 358</span> </div><div class="line"><a name="l00359"></a><span class="lineno"> 359</span>  address_iterator_.set_iteration_index(access_idx);</div><div class="line"><a name="l00360"></a><span class="lineno"> 360</span>  <span class="keywordflow">if</span> (address_iterator_.valid()) {</div><div class="line"><a name="l00361"></a><span class="lineno"> 361</span>  *(address_iterator_.get() + pointer_offset) = frag_ptr[access_idx];</div><div class="line"><a name="l00362"></a><span class="lineno"> 362</span>  }</div><div class="line"><a name="l00363"></a><span class="lineno"> 363</span>  ++address_iterator_;</div><div class="line"><a name="l00364"></a><span class="lineno"> 364</span>  }</div><div class="line"><a name="l00365"></a><span class="lineno"> 365</span>  }</div><div class="line"><a name="l00366"></a><span class="lineno"> 366</span>  }</div><div class="line"><a name="l00367"></a><span class="lineno"> 367</span>  }</div><div class="line"><a name="l00368"></a><span class="lineno"> 368</span> </div><div class="line"><a name="l00370"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#ab2f86d2e6cbd41ff0df73d4e1459c59f"> 370</a></span>  CUTLASS_DEVICE</div><div class="line"><a name="l00371"></a><span class="lineno"> 371</span>  <span class="keywordtype">void</span> store(<a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a0c7351825135690fa82c4fba5f9447c7">Fragment</a> <span class="keyword">const</span> &frag) { store_with_pointer_offset(frag, 0); }</div><div class="line"><a name="l00372"></a><span class="lineno"> 372</span> };</div><div class="line"><a name="l00373"></a><span class="lineno"> 373</span> </div><div class="line"><a name="l00375"></a><span class="lineno"> 375</span> </div><div class="line"><a name="l00383"></a><span class="lineno"> 383</span> <span class="keyword">template</span> <</div><div class="line"><a name="l00384"></a><span class="lineno"> 384</span>  <span class="keyword">typename</span> Shape_,</div><div class="line"><a name="l00385"></a><span class="lineno"> 385</span>  <span class="keyword">typename</span> Element_,</div><div class="line"><a name="l00386"></a><span class="lineno"> 386</span>  <span class="keywordtype">int</span> AdvanceRank,</div><div class="line"><a name="l00387"></a><span class="lineno"> 387</span>  <span class="keyword">typename</span> ThreadMap_,</div><div class="line"><a name="l00388"></a><span class="lineno"> 388</span>  <span class="keywordtype">bool</span> Transpose_</div><div class="line"><a name="l00389"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html"> 389</a></span> ></div><div class="line"><a name="l00390"></a><span class="lineno"> 390</span> <span class="keyword">class </span><a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile.html">PredicatedTileIterator2dThreadTile</a><Shape_, Element_, layout::ColumnMajor, AdvanceRank, ThreadMap_, Transpose_> {</div><div class="line"><a name="l00391"></a><span class="lineno"> 391</span> <span class="keyword">public</span>:</div><div class="line"><a name="l00392"></a><span class="lineno"> 392</span> </div><div class="line"><a name="l00393"></a><span class="lineno"> 393</span>  <a class="code" href="platform_8h.html#adde4c9ea91b753491851361a4198c009">static_assert</a>(AdvanceRank == 0 || AdvanceRank == 1, </div><div class="line"><a name="l00394"></a><span class="lineno"> 394</span>  <span class="stringliteral">"Specialization for pitch-linear iterator may along advance along the "</span></div><div class="line"><a name="l00395"></a><span class="lineno"> 395</span>  <span class="stringliteral">"contiguous(rank=0) or strided(rank=1) dimension."</span>);</div><div class="line"><a name="l00396"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#affebbfbc62a67c328d2ceceb382361c3"> 396</a></span> </div><div class="line"><a name="l00397"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#ad87e687d0a0e4a228ccbc6b93cec6daa"> 397</a></span>  <span class="keyword">using</span> <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#affebbfbc62a67c328d2ceceb382361c3">Shape</a> = Shape_;</div><div class="line"><a name="l00398"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#ada60020d427455a3e3aaedfe7fbdd7e4"> 398</a></span>  <span class="keyword">using</span> Element = Element_;</div><div class="line"><a name="l00399"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a472dc9ee3f75f91699c794f1eb24dc0b"> 399</a></span>  <span class="keyword">using</span> Layout = <a class="code" href="classcutlass_1_1layout_1_1ColumnMajor.html">layout::ColumnMajor</a>;</div><div class="line"><a name="l00400"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a02ecbfc622abbdc63ed87c26ccead0fa"> 400</a></span>  <span class="keyword">static</span> <span class="keywordtype">int</span> <span class="keyword">const</span> kAdvanceRank = AdvanceRank;</div><div class="line"><a name="l00401"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a5d121ffe56cb69b9b0398416c75f4bd2"> 401</a></span>  <span class="keyword">using</span> ThreadMap = ThreadMap_;</div><div class="line"><a name="l00402"></a><span class="lineno"> 402</span>  <span class="keyword">static</span> <span class="keywordtype">bool</span> <span class="keyword">const</span> Transpose = Transpose_;</div><div class="line"><a name="l00403"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a7912240a9802251763983a1fa4a297f1"> 403</a></span> </div><div class="line"><a name="l00404"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a28c6395acf81c013c8089414a91b66e7"> 404</a></span>  <span class="keyword">using</span> <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a7912240a9802251763983a1fa4a297f1">Index</a> = <span class="keyword">typename</span> <a class="code" href="classcutlass_1_1layout_1_1ColumnMajor.html#a6409273d6b40acb27aff0564eb038336">Layout::Index</a>;</div><div class="line"><a name="l00405"></a><span class="lineno"> 405</span>  <span class="keyword">using</span> <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a28c6395acf81c013c8089414a91b66e7">LongIndex</a> = <span class="keyword">typename</span> <a class="code" href="classcutlass_1_1layout_1_1ColumnMajor.html#a4cc90aa67c4692f0a2cd9f59b8a07997">Layout::LongIndex</a>;</div><div class="line"><a name="l00406"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a0edf4a76d69dee0fb8a260afb888fd90"> 406</a></span> </div><div class="line"><a name="l00407"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a32dcf7af69e884082fcefbfcd3ecf412"> 407</a></span>  <span class="keyword">using</span> <a class="code" href="classcutlass_1_1TensorRef.html">TensorRef</a> = <a class="code" href="classcutlass_1_1TensorRef.html">TensorRef<Element, Layout></a>;</div><div class="line"><a name="l00408"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a1568b1a89e72af084604e3c209b08f11"> 408</a></span>  <span class="keyword">using</span> <a class="code" href="classcutlass_1_1TensorView.html">TensorView</a> = <a class="code" href="classcutlass_1_1TensorView.html">TensorView<Element, Layout></a>;</div><div class="line"><a name="l00409"></a><span class="lineno"> 409</span>  <span class="keyword">using</span> <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a1568b1a89e72af084604e3c209b08f11">TensorCoord</a> = <span class="keyword">typename</span> <a class="code" href="structcutlass_1_1MatrixCoord.html">Layout::TensorCoord</a>;</div><div class="line"><a name="l00410"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#abbc6f561480d48c05001038ab7a0bebb"> 410</a></span> </div><div class="line"><a name="l00411"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a3d483d123a36af834026cf40da581a11"> 411</a></span>  <span class="keyword">using</span> <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#abbc6f561480d48c05001038ab7a0bebb">Pointer</a> = Element *;</div><div class="line"><a name="l00412"></a><span class="lineno"> 412</span>  <span class="keyword">using</span> <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a3d483d123a36af834026cf40da581a11">NonConstPointer</a> = <span class="keyword">typename</span> <a class="code" href="structcutlass_1_1platform_1_1remove__const.html#ac3662947fa50251daf58240a9c798085">platform::remove_const<Element>::type</a> *;</div><div class="line"><a name="l00413"></a><span class="lineno"> 413</span> </div><div class="line"><a name="l00414"></a><span class="lineno"> 414</span>  <span class="keyword">using</span> <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile.html">UnderlyingIterator</a> = <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile.html">PredicatedTileIterator2dThreadTile</a><</div><div class="line"><a name="l00415"></a><span class="lineno"> 415</span>  <a class="code" href="structcutlass_1_1layout_1_1PitchLinearShape.html">layout::PitchLinearShape<Shape::kRow, Shape::kColumn></a>,</div><div class="line"><a name="l00416"></a><span class="lineno"> 416</span>  Element,</div><div class="line"><a name="l00417"></a><span class="lineno"> 417</span>  <a class="code" href="classcutlass_1_1layout_1_1PitchLinear.html">layout::PitchLinear</a>,</div><div class="line"><a name="l00418"></a><span class="lineno"> 418</span>  (kAdvanceRank == 0 ? 0 : 1),</div><div class="line"><a name="l00419"></a><span class="lineno"> 419</span>  ThreadMap,</div><div class="line"><a name="l00420"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a68872ddeed009fde4b7b433b901fb4ee"> 420</a></span>  Transpose</div><div class="line"><a name="l00421"></a><span class="lineno"> 421</span>  >;</div><div class="line"><a name="l00422"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a50525f7a7785ed266f6e09290a6f0634"> 422</a></span> </div><div class="line"><a name="l00423"></a><span class="lineno"> 423</span>  <span class="keyword">using</span> <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a50525f7a7785ed266f6e09290a6f0634">AccessType</a> = <span class="keyword">typename</span> UnderlyingIterator::AccessType;</div><div class="line"><a name="l00424"></a><span class="lineno"> 424</span> </div><div class="line"><a name="l00426"></a><span class="lineno"> 426</span>  <span class="keyword">using</span> <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a9be82cb5e5c9a215b5862dba28dc57f4">Fragment</a> = cutlass::Array<Element, ThreadMap::Iterations::kCount * ThreadMap::ThreadAccessShape::kCount>;</div><div class="line"><a name="l00427"></a><span class="lineno"> 427</span> </div><div class="line"><a name="l00429"></a><span class="lineno"> 429</span>  <span class="keyword">using</span> <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a8275a7f49a107c4122a4d3278aba94ce">Mask</a> = <span class="keyword">typename</span> UnderlyingIterator::Mask;</div><div class="line"><a name="l00430"></a><span class="lineno"> 430</span> </div><div class="line"><a name="l00432"></a><span class="lineno"> 432</span>  <span class="keyword">class </span>Params {</div><div class="line"><a name="l00433"></a><span class="lineno"> 433</span>  <span class="keyword">private</span>:</div><div class="line"><a name="l00434"></a><span class="lineno"> 434</span> </div><div class="line"><a name="l00435"></a><span class="lineno"> 435</span>  <span class="keyword">friend</span> <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile.html">PredicatedTileIterator2dThreadTile</a>;</div><div class="line"><a name="l00436"></a><span class="lineno"> 436</span> </div><div class="line"><a name="l00438"></a><span class="lineno"> 438</span>  <span class="keyword">typename</span> UnderlyingIterator::Params params_;</div><div class="line"><a name="l00439"></a><span class="lineno"> 439</span> </div><div class="line"><a name="l00440"></a><span class="lineno"> 440</span>  <span class="keyword">public</span>:</div><div class="line"><a name="l00441"></a><span class="lineno"> 441</span>  </div><div class="line"><a name="l00442"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_01e11ed7192af5d7ad1bce5641fa13112e.html#af4d9c8e63b88ca4a42c87a86d1cb9abc"> 442</a></span>  <a class="code" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a></div><div class="line"><a name="l00443"></a><span class="lineno"> 443</span>  Params() { }</div><div class="line"><a name="l00444"></a><span class="lineno"> 444</span> </div><div class="line"><a name="l00446"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_01e11ed7192af5d7ad1bce5641fa13112e.html#a776622c626e88ec2a2ecfeec61e08995"> 446</a></span>  <a class="code" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a></div><div class="line"><a name="l00447"></a><span class="lineno"> 447</span>  Params(Layout <span class="keyword">const</span> &layout): params_(layout::PitchLinear(layout.stride(0))) {</div><div class="line"><a name="l00448"></a><span class="lineno"> 448</span> </div><div class="line"><a name="l00449"></a><span class="lineno"> 449</span>  }</div><div class="line"><a name="l00450"></a><span class="lineno"> 450</span>  };</div><div class="line"><a name="l00451"></a><span class="lineno"> 451</span> </div><div class="line"><a name="l00452"></a><span class="lineno"> 452</span> </div><div class="line"><a name="l00453"></a><span class="lineno"> 453</span> <span class="keyword">private</span>:</div><div class="line"><a name="l00454"></a><span class="lineno"> 454</span> </div><div class="line"><a name="l00455"></a><span class="lineno"> 455</span>  <span class="comment">//</span></div><div class="line"><a name="l00456"></a><span class="lineno"> 456</span>  <span class="comment">// Data members</span></div><div class="line"><a name="l00457"></a><span class="lineno"> 457</span>  <span class="comment">//</span></div><div class="line"><a name="l00458"></a><span class="lineno"> 458</span> </div><div class="line"><a name="l00460"></a><span class="lineno"> 460</span>  <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile.html">UnderlyingIterator</a> iterator_;</div><div class="line"><a name="l00461"></a><span class="lineno"> 461</span> </div><div class="line"><a name="l00462"></a><span class="lineno"> 462</span> <span class="keyword">public</span>:</div><div class="line"><a name="l00463"></a><span class="lineno"> 463</span> </div><div class="line"><a name="l00465"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a691ce7629fc7d52146086dffae11c7e4"> 465</a></span>  <a class="code" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a></div><div class="line"><a name="l00466"></a><span class="lineno"> 466</span>  <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile.html">PredicatedTileIterator2dThreadTile</a>(</div><div class="line"><a name="l00467"></a><span class="lineno"> 467</span>  Params <span class="keyword">const</span> &params, </div><div class="line"><a name="l00468"></a><span class="lineno"> 468</span>  <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#abbc6f561480d48c05001038ab7a0bebb">Pointer</a> pointer, </div><div class="line"><a name="l00469"></a><span class="lineno"> 469</span>  <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a1568b1a89e72af084604e3c209b08f11">TensorCoord</a> extent, </div><div class="line"><a name="l00470"></a><span class="lineno"> 470</span>  <span class="keywordtype">int</span> thread_id, </div><div class="line"><a name="l00471"></a><span class="lineno"> 471</span>  <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a1568b1a89e72af084604e3c209b08f11">TensorCoord</a> <span class="keyword">const</span> &threadblock_offset </div><div class="line"><a name="l00472"></a><span class="lineno"> 472</span>  ):</div><div class="line"><a name="l00473"></a><span class="lineno"> 473</span>  iterator_(</div><div class="line"><a name="l00474"></a><span class="lineno"> 474</span>  params.params_,</div><div class="line"><a name="l00475"></a><span class="lineno"> 475</span>  pointer,</div><div class="line"><a name="l00476"></a><span class="lineno"> 476</span>  layout::PitchLinearCoord(extent.row(), extent.column()),</div><div class="line"><a name="l00477"></a><span class="lineno"> 477</span>  thread_id,</div><div class="line"><a name="l00478"></a><span class="lineno"> 478</span>  layout::PitchLinearCoord(threadblock_offset.row(), threadblock_offset.column())</div><div class="line"><a name="l00479"></a><span class="lineno"> 479</span>  ) { }</div><div class="line"><a name="l00480"></a><span class="lineno"> 480</span> </div><div class="line"><a name="l00482"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a26211dc2b48ab669363e9af8738f1e13"> 482</a></span>  <a class="code" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a></div><div class="line"><a name="l00483"></a><span class="lineno"> 483</span>  <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile.html">PredicatedTileIterator2dThreadTile</a>(</div><div class="line"><a name="l00484"></a><span class="lineno"> 484</span>  Params <span class="keyword">const</span> &params, </div><div class="line"><a name="l00485"></a><span class="lineno"> 485</span>  <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#abbc6f561480d48c05001038ab7a0bebb">Pointer</a> pointer, </div><div class="line"><a name="l00486"></a><span class="lineno"> 486</span>  <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a1568b1a89e72af084604e3c209b08f11">TensorCoord</a> extent, </div><div class="line"><a name="l00487"></a><span class="lineno"> 487</span>  <span class="keywordtype">int</span> thread_id </div><div class="line"><a name="l00488"></a><span class="lineno"> 488</span>  ): PredicatedTileIterator2dThreadTile(params, pointer, extent, thread_id, <a class="code" href="namespacecutlass.html#a7419519fa453a121dfa5f26bf87318d9">make_Coord</a>(0, 0)) { }</div><div class="line"><a name="l00489"></a><span class="lineno"> 489</span> </div><div class="line"><a name="l00491"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a4548bdd40a2e608b14ea2f2cf94a6878"> 491</a></span>  <a class="code" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a></div><div class="line"><a name="l00492"></a><span class="lineno"> 492</span>  <span class="keywordtype">void</span> add_pointer_offset(<a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a28c6395acf81c013c8089414a91b66e7">LongIndex</a> pointer_offset) {</div><div class="line"><a name="l00493"></a><span class="lineno"> 493</span>  iterator_.add_pointer_offset(pointer_offset);</div><div class="line"><a name="l00494"></a><span class="lineno"> 494</span>  }</div><div class="line"><a name="l00495"></a><span class="lineno"> 495</span> </div><div class="line"><a name="l00501"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a24603f7fa0795c885eaefd097579ca5e"> 501</a></span>  <a class="code" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a></div><div class="line"><a name="l00502"></a><span class="lineno"> 502</span>  <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile.html">PredicatedTileIterator2dThreadTile</a> &<a class="code" href="namespacecutlass.html#a1c7a9e66ca7b5dc7413ea3b8f349530a">operator++</a>() {</div><div class="line"><a name="l00503"></a><span class="lineno"> 503</span>  ++iterator_;</div><div class="line"><a name="l00504"></a><span class="lineno"> 504</span>  <span class="keywordflow">return</span> *<span class="keyword">this</span>;</div><div class="line"><a name="l00505"></a><span class="lineno"> 505</span>  }</div><div class="line"><a name="l00506"></a><span class="lineno"> 506</span> </div><div class="line"><a name="l00512"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a6ede62f8cbd80ae931bc53e1f6d2b5cb"> 512</a></span>  <a class="code" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a></div><div class="line"><a name="l00513"></a><span class="lineno"> 513</span>  <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile.html">PredicatedTileIterator2dThreadTile</a> <a class="code" href="namespacecutlass.html#a1c7a9e66ca7b5dc7413ea3b8f349530a">operator++</a>(<span class="keywordtype">int</span>) {</div><div class="line"><a name="l00514"></a><span class="lineno"> 514</span>  <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile.html">PredicatedTileIterator2dThreadTile</a> <span class="keyword">self</span>(*this);</div><div class="line"><a name="l00515"></a><span class="lineno"> 515</span>  <a class="code" href="namespacecutlass.html#a1c7a9e66ca7b5dc7413ea3b8f349530a">operator++</a>();</div><div class="line"><a name="l00516"></a><span class="lineno"> 516</span>  <span class="keywordflow">return</span> <span class="keyword">self</span>;</div><div class="line"><a name="l00517"></a><span class="lineno"> 517</span>  }</div><div class="line"><a name="l00518"></a><span class="lineno"> 518</span> </div><div class="line"><a name="l00520"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a16eb418826fc3bead16d7f15dccae29f"> 520</a></span>  <a class="code" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a></div><div class="line"><a name="l00521"></a><span class="lineno"> 521</span>  <span class="keywordtype">void</span> clear_mask() {</div><div class="line"><a name="l00522"></a><span class="lineno"> 522</span>  iterator_.clear_mask();</div><div class="line"><a name="l00523"></a><span class="lineno"> 523</span>  }</div><div class="line"><a name="l00524"></a><span class="lineno"> 524</span> </div><div class="line"><a name="l00526"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a8f61201ee8e582be64e74c09e87ee36b"> 526</a></span>  <a class="code" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a></div><div class="line"><a name="l00527"></a><span class="lineno"> 527</span>  <span class="keywordtype">void</span> enable_mask() {</div><div class="line"><a name="l00528"></a><span class="lineno"> 528</span>  iterator_.enable_mask();</div><div class="line"><a name="l00529"></a><span class="lineno"> 529</span>  }</div><div class="line"><a name="l00530"></a><span class="lineno"> 530</span> </div><div class="line"><a name="l00532"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a3a948e8e7608cd6959ef161717c63310"> 532</a></span>  <a class="code" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a></div><div class="line"><a name="l00533"></a><span class="lineno"> 533</span>  <span class="keywordtype">void</span> set_mask(<a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a8275a7f49a107c4122a4d3278aba94ce">Mask</a> <span class="keyword">const</span> &mask) {</div><div class="line"><a name="l00534"></a><span class="lineno"> 534</span>  iterator_.set_mask(mask);</div><div class="line"><a name="l00535"></a><span class="lineno"> 535</span>  }</div><div class="line"><a name="l00536"></a><span class="lineno"> 536</span> </div><div class="line"><a name="l00538"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a56503db6f71a1327f375bbbdefc69433"> 538</a></span>  <a class="code" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a></div><div class="line"><a name="l00539"></a><span class="lineno"> 539</span>  <span class="keywordtype">void</span> get_mask(<a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a8275a7f49a107c4122a4d3278aba94ce">Mask</a> &mask) {</div><div class="line"><a name="l00540"></a><span class="lineno"> 540</span>  iterator_.get_mask(mask);</div><div class="line"><a name="l00541"></a><span class="lineno"> 541</span>  }</div><div class="line"><a name="l00542"></a><span class="lineno"> 542</span> </div><div class="line"><a name="l00544"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a3b94fd9b0cbdc4a57f1d6e45381e0cd3"> 544</a></span>  CUTLASS_DEVICE</div><div class="line"><a name="l00545"></a><span class="lineno"> 545</span>  <span class="keywordtype">void</span> load_with_pointer_offset(<a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a9be82cb5e5c9a215b5862dba28dc57f4">Fragment</a> &frag, <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a7912240a9802251763983a1fa4a297f1">Index</a> pointer_offset) {</div><div class="line"><a name="l00546"></a><span class="lineno"> 546</span>  iterator_.load_with_pointer_offset(frag, pointer_offset);</div><div class="line"><a name="l00547"></a><span class="lineno"> 547</span>  }</div><div class="line"><a name="l00548"></a><span class="lineno"> 548</span> </div><div class="line"><a name="l00550"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#aaa2ee3db6431b7340e3bb2c347344a9c"> 550</a></span>  CUTLASS_DEVICE</div><div class="line"><a name="l00551"></a><span class="lineno"> 551</span>  <span class="keywordtype">void</span> load(<a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a9be82cb5e5c9a215b5862dba28dc57f4">Fragment</a> &frag) {</div><div class="line"><a name="l00552"></a><span class="lineno"> 552</span>  load_with_pointer_offset(frag, 0);</div><div class="line"><a name="l00553"></a><span class="lineno"> 553</span>  }</div><div class="line"><a name="l00554"></a><span class="lineno"> 554</span> </div><div class="line"><a name="l00556"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a180abcaed15051efb8a577e9afc1f079"> 556</a></span>  CUTLASS_DEVICE</div><div class="line"><a name="l00557"></a><span class="lineno"> 557</span>  <span class="keywordtype">void</span> store_with_pointer_offset(<a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a9be82cb5e5c9a215b5862dba28dc57f4">Fragment</a> <span class="keyword">const</span> &frag, <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a7912240a9802251763983a1fa4a297f1">Index</a> pointer_offset) {</div><div class="line"><a name="l00558"></a><span class="lineno"> 558</span>  iterator_.store_with_pointer_offset(frag, pointer_offset);</div><div class="line"><a name="l00559"></a><span class="lineno"> 559</span>  }</div><div class="line"><a name="l00560"></a><span class="lineno"> 560</span> </div><div class="line"><a name="l00562"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#ab1ecb84f837553153808f2fc827e77c8"> 562</a></span>  CUTLASS_DEVICE</div><div class="line"><a name="l00563"></a><span class="lineno"> 563</span>  <span class="keywordtype">void</span> store(<a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a9be82cb5e5c9a215b5862dba28dc57f4">Fragment</a> <span class="keyword">const</span> &frag) {</div><div class="line"><a name="l00564"></a><span class="lineno"> 564</span>  store_with_pointer_offset(frag, 0);</div><div class="line"><a name="l00565"></a><span class="lineno"> 565</span>  }</div><div class="line"><a name="l00566"></a><span class="lineno"> 566</span> };</div><div class="line"><a name="l00567"></a><span class="lineno"> 567</span> </div><div class="line"><a name="l00569"></a><span class="lineno"> 569</span> </div><div class="line"><a name="l00577"></a><span class="lineno"> 577</span> <span class="keyword">template</span> <</div><div class="line"><a name="l00578"></a><span class="lineno"> 578</span>  <span class="keyword">typename</span> Shape_,</div><div class="line"><a name="l00579"></a><span class="lineno"> 579</span>  <span class="keyword">typename</span> Element_,</div><div class="line"><a name="l00580"></a><span class="lineno"> 580</span>  <span class="keywordtype">int</span> AdvanceRank,</div><div class="line"><a name="l00581"></a><span class="lineno"> 581</span>  <span class="keyword">typename</span> ThreadMap_,</div><div class="line"><a name="l00582"></a><span class="lineno"> 582</span>  <span class="keywordtype">bool</span> Transpose_</div><div class="line"><a name="l00583"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html"> 583</a></span> ></div><div class="line"><a name="l00584"></a><span class="lineno"> 584</span> <span class="keyword">class </span><a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile.html">PredicatedTileIterator2dThreadTile</a><Shape_, Element_, layout::RowMajor, AdvanceRank, ThreadMap_, Transpose_> {</div><div class="line"><a name="l00585"></a><span class="lineno"> 585</span> <span class="keyword">public</span>:</div><div class="line"><a name="l00586"></a><span class="lineno"> 586</span> </div><div class="line"><a name="l00587"></a><span class="lineno"> 587</span>  <a class="code" href="platform_8h.html#adde4c9ea91b753491851361a4198c009">static_assert</a>(AdvanceRank == 0 || AdvanceRank == 1, </div><div class="line"><a name="l00588"></a><span class="lineno"> 588</span>  <span class="stringliteral">"Specialization for pitch-linear iterator may along advance along the "</span></div><div class="line"><a name="l00589"></a><span class="lineno"> 589</span>  <span class="stringliteral">"contiguous(rank=0) or strided(rank=1) dimension."</span>);</div><div class="line"><a name="l00590"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a72e72c4beb8186dbfed16b7dd2008ac6"> 590</a></span> </div><div class="line"><a name="l00591"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a0dd47275aba244cbab8e912a4cb62b91"> 591</a></span>  <span class="keyword">using</span> <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a72e72c4beb8186dbfed16b7dd2008ac6">Shape</a> = Shape_;</div><div class="line"><a name="l00592"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a2f676f0cc3fa9698a1c59ea02f65aca7"> 592</a></span>  <span class="keyword">using</span> Element = Element_;</div><div class="line"><a name="l00593"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#ab9b93715c82a268fd53eb1d55c281bab"> 593</a></span>  <span class="keyword">using</span> Layout = <a class="code" href="classcutlass_1_1layout_1_1RowMajor.html">layout::RowMajor</a>;</div><div class="line"><a name="l00594"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a32ccc629da4624beca65ac36a30555eb"> 594</a></span>  <span class="keyword">static</span> <span class="keywordtype">int</span> <span class="keyword">const</span> kAdvanceRank = AdvanceRank;</div><div class="line"><a name="l00595"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a87728d1f88d8c495779b9040c5be37f3"> 595</a></span>  <span class="keyword">using</span> ThreadMap = ThreadMap_;</div><div class="line"><a name="l00596"></a><span class="lineno"> 596</span>  <span class="keyword">static</span> <span class="keywordtype">bool</span> <span class="keyword">const</span> Transpose = Transpose_;</div><div class="line"><a name="l00597"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a31ddecaefe6832464ccd170834e398ca"> 597</a></span> </div><div class="line"><a name="l00598"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a33d57eb89928e9e38a9be7190452928c"> 598</a></span>  <span class="keyword">using</span> <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a31ddecaefe6832464ccd170834e398ca">Index</a> = <span class="keyword">typename</span> <a class="code" href="classcutlass_1_1layout_1_1RowMajor.html#aa49e242b14b4f482bc6bdd082acfb576">Layout::Index</a>;</div><div class="line"><a name="l00599"></a><span class="lineno"> 599</span>  <span class="keyword">using</span> <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a33d57eb89928e9e38a9be7190452928c">LongIndex</a> = <span class="keyword">typename</span> <a class="code" href="classcutlass_1_1layout_1_1RowMajor.html#a4062a36ab044fdea058504ed52ee60b8">Layout::LongIndex</a>;</div><div class="line"><a name="l00600"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a82213beb5d334a8da859825af3d18df1"> 600</a></span> </div><div class="line"><a name="l00601"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#ac8a765c7c18be892c6e742eb71425843"> 601</a></span>  <span class="keyword">using</span> <a class="code" href="classcutlass_1_1TensorRef.html">TensorRef</a> = <a class="code" href="classcutlass_1_1TensorRef.html">TensorRef<Element, Layout></a>;</div><div class="line"><a name="l00602"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a6f8d0fbaf19bf59d0d76bc3937609443"> 602</a></span>  <span class="keyword">using</span> <a class="code" href="classcutlass_1_1TensorView.html">TensorView</a> = <a class="code" href="classcutlass_1_1TensorView.html">TensorView<Element, Layout></a>;</div><div class="line"><a name="l00603"></a><span class="lineno"> 603</span>  <span class="keyword">using</span> <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a6f8d0fbaf19bf59d0d76bc3937609443">TensorCoord</a> = <span class="keyword">typename</span> <a class="code" href="structcutlass_1_1MatrixCoord.html">Layout::TensorCoord</a>;</div><div class="line"><a name="l00604"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a9ecc09cd70be9f7e56f309465d878f79"> 604</a></span> </div><div class="line"><a name="l00605"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#af81f97366f2524518d48955568d45262"> 605</a></span>  <span class="keyword">using</span> <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a9ecc09cd70be9f7e56f309465d878f79">Pointer</a> = Element *;</div><div class="line"><a name="l00606"></a><span class="lineno"> 606</span>  <span class="keyword">using</span> <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#af81f97366f2524518d48955568d45262">NonConstPointer</a> = <span class="keyword">typename</span> <a class="code" href="structcutlass_1_1platform_1_1remove__const.html#ac3662947fa50251daf58240a9c798085">platform::remove_const<Element>::type</a> *;</div><div class="line"><a name="l00607"></a><span class="lineno"> 607</span> </div><div class="line"><a name="l00608"></a><span class="lineno"> 608</span>  <span class="keyword">using</span> <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile.html">UnderlyingIterator</a> = <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile.html">PredicatedTileIterator2dThreadTile</a><</div><div class="line"><a name="l00609"></a><span class="lineno"> 609</span>  <a class="code" href="structcutlass_1_1layout_1_1PitchLinearShape.html">layout::PitchLinearShape<Shape::kColumn, Shape::kRow></a>,</div><div class="line"><a name="l00610"></a><span class="lineno"> 610</span>  Element,</div><div class="line"><a name="l00611"></a><span class="lineno"> 611</span>  <a class="code" href="classcutlass_1_1layout_1_1PitchLinear.html">layout::PitchLinear</a>,</div><div class="line"><a name="l00612"></a><span class="lineno"> 612</span>  (kAdvanceRank == 0 ? 1 : 0),</div><div class="line"><a name="l00613"></a><span class="lineno"> 613</span>  ThreadMap,</div><div class="line"><a name="l00614"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a5edb8ebf45b0a2cafe34a144007ff22d"> 614</a></span>  Transpose</div><div class="line"><a name="l00615"></a><span class="lineno"> 615</span>  >;</div><div class="line"><a name="l00616"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#aadb4b95b45598045f882a077fb5c587e"> 616</a></span> </div><div class="line"><a name="l00617"></a><span class="lineno"> 617</span>  <span class="keyword">using</span> <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#aadb4b95b45598045f882a077fb5c587e">AccessType</a> = <span class="keyword">typename</span> UnderlyingIterator::AccessType;</div><div class="line"><a name="l00618"></a><span class="lineno"> 618</span> </div><div class="line"><a name="l00620"></a><span class="lineno"> 620</span>  <span class="keyword">using</span> <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#ae342607106778417432c4694b92736b3">Fragment</a> = cutlass::Array<Element, ThreadMap::Iterations::kCount * ThreadMap::ThreadAccessShape::kCount>;</div><div class="line"><a name="l00621"></a><span class="lineno"> 621</span> </div><div class="line"><a name="l00623"></a><span class="lineno"> 623</span>  <span class="keyword">using</span> <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#aef7b4321bef75c2ad969fedfb7e1aeee">Mask</a> = <span class="keyword">typename</span> UnderlyingIterator::Mask;</div><div class="line"><a name="l00624"></a><span class="lineno"> 624</span> </div><div class="line"><a name="l00626"></a><span class="lineno"> 626</span>  <span class="keyword">class </span>Params {</div><div class="line"><a name="l00627"></a><span class="lineno"> 627</span>  <span class="keyword">private</span>:</div><div class="line"><a name="l00628"></a><span class="lineno"> 628</span> </div><div class="line"><a name="l00629"></a><span class="lineno"> 629</span>  <span class="keyword">friend</span> <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile.html">PredicatedTileIterator2dThreadTile</a>;</div><div class="line"><a name="l00630"></a><span class="lineno"> 630</span> </div><div class="line"><a name="l00632"></a><span class="lineno"> 632</span>  <span class="keyword">typename</span> UnderlyingIterator::Params params_;</div><div class="line"><a name="l00633"></a><span class="lineno"> 633</span> </div><div class="line"><a name="l00634"></a><span class="lineno"> 634</span>  <span class="keyword">public</span>:</div><div class="line"><a name="l00635"></a><span class="lineno"> 635</span>  </div><div class="line"><a name="l00636"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0102e766863c6ac9ec2063a02c4803eecb.html#ac9762c9d5012de584dfac064d9928d00"> 636</a></span>  <a class="code" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a></div><div class="line"><a name="l00637"></a><span class="lineno"> 637</span>  Params() { } </div><div class="line"><a name="l00638"></a><span class="lineno"> 638</span> </div><div class="line"><a name="l00640"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0102e766863c6ac9ec2063a02c4803eecb.html#a556a9533f59d90b23426798a3bb12d82"> 640</a></span>  <a class="code" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a></div><div class="line"><a name="l00641"></a><span class="lineno"> 641</span>  Params(Layout <span class="keyword">const</span> &layout): params_(layout::PitchLinear(layout.stride(0))) {</div><div class="line"><a name="l00642"></a><span class="lineno"> 642</span> </div><div class="line"><a name="l00643"></a><span class="lineno"> 643</span>  };</div><div class="line"><a name="l00644"></a><span class="lineno"> 644</span>  };</div><div class="line"><a name="l00645"></a><span class="lineno"> 645</span> </div><div class="line"><a name="l00646"></a><span class="lineno"> 646</span> </div><div class="line"><a name="l00647"></a><span class="lineno"> 647</span> <span class="keyword">private</span>:</div><div class="line"><a name="l00648"></a><span class="lineno"> 648</span> </div><div class="line"><a name="l00649"></a><span class="lineno"> 649</span>  <span class="comment">//</span></div><div class="line"><a name="l00650"></a><span class="lineno"> 650</span>  <span class="comment">// Data members</span></div><div class="line"><a name="l00651"></a><span class="lineno"> 651</span>  <span class="comment">//</span></div><div class="line"><a name="l00652"></a><span class="lineno"> 652</span> </div><div class="line"><a name="l00654"></a><span class="lineno"> 654</span>  <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile.html">UnderlyingIterator</a> iterator_;</div><div class="line"><a name="l00655"></a><span class="lineno"> 655</span> </div><div class="line"><a name="l00656"></a><span class="lineno"> 656</span> <span class="keyword">public</span>:</div><div class="line"><a name="l00657"></a><span class="lineno"> 657</span> </div><div class="line"><a name="l00659"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a206d1ee28d98522f4e6577c41cce5ddb"> 659</a></span>  <a class="code" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a></div><div class="line"><a name="l00660"></a><span class="lineno"> 660</span>  <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile.html">PredicatedTileIterator2dThreadTile</a>(</div><div class="line"><a name="l00661"></a><span class="lineno"> 661</span>  Params <span class="keyword">const</span> &params, </div><div class="line"><a name="l00662"></a><span class="lineno"> 662</span>  <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a9ecc09cd70be9f7e56f309465d878f79">Pointer</a> pointer, </div><div class="line"><a name="l00663"></a><span class="lineno"> 663</span>  <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a6f8d0fbaf19bf59d0d76bc3937609443">TensorCoord</a> extent, </div><div class="line"><a name="l00664"></a><span class="lineno"> 664</span>  <span class="keywordtype">int</span> thread_id, </div><div class="line"><a name="l00665"></a><span class="lineno"> 665</span>  <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a6f8d0fbaf19bf59d0d76bc3937609443">TensorCoord</a> <span class="keyword">const</span> &threadblock_offset </div><div class="line"><a name="l00666"></a><span class="lineno"> 666</span>  ):</div><div class="line"><a name="l00667"></a><span class="lineno"> 667</span>  iterator_(</div><div class="line"><a name="l00668"></a><span class="lineno"> 668</span>  params.params_,</div><div class="line"><a name="l00669"></a><span class="lineno"> 669</span>  pointer,</div><div class="line"><a name="l00670"></a><span class="lineno"> 670</span>  layout::PitchLinearCoord(extent.column(), extent.row()),</div><div class="line"><a name="l00671"></a><span class="lineno"> 671</span>  thread_id,</div><div class="line"><a name="l00672"></a><span class="lineno"> 672</span>  layout::PitchLinearCoord(threadblock_offset.column(), threadblock_offset.row())</div><div class="line"><a name="l00673"></a><span class="lineno"> 673</span>  ) { }</div><div class="line"><a name="l00674"></a><span class="lineno"> 674</span> </div><div class="line"><a name="l00676"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a5b351404fe3d7bd3af79cba4118f04b8"> 676</a></span>  <a class="code" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a></div><div class="line"><a name="l00677"></a><span class="lineno"> 677</span>  <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile.html">PredicatedTileIterator2dThreadTile</a>(</div><div class="line"><a name="l00678"></a><span class="lineno"> 678</span>  Params <span class="keyword">const</span> &params, </div><div class="line"><a name="l00679"></a><span class="lineno"> 679</span>  <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a9ecc09cd70be9f7e56f309465d878f79">Pointer</a> pointer, </div><div class="line"><a name="l00680"></a><span class="lineno"> 680</span>  <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a6f8d0fbaf19bf59d0d76bc3937609443">TensorCoord</a> extent, </div><div class="line"><a name="l00681"></a><span class="lineno"> 681</span>  <span class="keywordtype">int</span> thread_id </div><div class="line"><a name="l00682"></a><span class="lineno"> 682</span>  ): PredicatedTileIterator2dThreadTile(params, pointer, extent, thread_id, <a class="code" href="namespacecutlass.html#a7419519fa453a121dfa5f26bf87318d9">make_Coord</a>(0, 0)) { }</div><div class="line"><a name="l00683"></a><span class="lineno"> 683</span> </div><div class="line"><a name="l00685"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#ad3e86cb09960501091069dea42724224"> 685</a></span>  <a class="code" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a></div><div class="line"><a name="l00686"></a><span class="lineno"> 686</span>  <span class="keywordtype">void</span> add_pointer_offset(<a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a33d57eb89928e9e38a9be7190452928c">LongIndex</a> pointer_offset) {</div><div class="line"><a name="l00687"></a><span class="lineno"> 687</span>  iterator_.add_pointer_offset(pointer_offset);</div><div class="line"><a name="l00688"></a><span class="lineno"> 688</span>  }</div><div class="line"><a name="l00689"></a><span class="lineno"> 689</span> </div><div class="line"><a name="l00695"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a5e72ab609350946f7d0773c347d3f4e4"> 695</a></span>  <a class="code" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a></div><div class="line"><a name="l00696"></a><span class="lineno"> 696</span>  <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile.html">PredicatedTileIterator2dThreadTile</a> &<a class="code" href="namespacecutlass.html#a1c7a9e66ca7b5dc7413ea3b8f349530a">operator++</a>() {</div><div class="line"><a name="l00697"></a><span class="lineno"> 697</span>  ++iterator_;</div><div class="line"><a name="l00698"></a><span class="lineno"> 698</span>  <span class="keywordflow">return</span> *<span class="keyword">this</span>;</div><div class="line"><a name="l00699"></a><span class="lineno"> 699</span>  }</div><div class="line"><a name="l00700"></a><span class="lineno"> 700</span> </div><div class="line"><a name="l00706"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a649a69d5df70c8fa5f400697edeb83f2"> 706</a></span>  <a class="code" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a></div><div class="line"><a name="l00707"></a><span class="lineno"> 707</span>  <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile.html">PredicatedTileIterator2dThreadTile</a> <a class="code" href="namespacecutlass.html#a1c7a9e66ca7b5dc7413ea3b8f349530a">operator++</a>(<span class="keywordtype">int</span>) {</div><div class="line"><a name="l00708"></a><span class="lineno"> 708</span>  <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile.html">PredicatedTileIterator2dThreadTile</a> <span class="keyword">self</span>(*this);</div><div class="line"><a name="l00709"></a><span class="lineno"> 709</span>  <a class="code" href="namespacecutlass.html#a1c7a9e66ca7b5dc7413ea3b8f349530a">operator++</a>();</div><div class="line"><a name="l00710"></a><span class="lineno"> 710</span>  <span class="keywordflow">return</span> <span class="keyword">self</span>;</div><div class="line"><a name="l00711"></a><span class="lineno"> 711</span>  }</div><div class="line"><a name="l00712"></a><span class="lineno"> 712</span> </div><div class="line"><a name="l00714"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a2778dc7d89b47e3fe9b1d3f9e67c9601"> 714</a></span>  <a class="code" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a></div><div class="line"><a name="l00715"></a><span class="lineno"> 715</span>  <span class="keywordtype">void</span> clear_mask() {</div><div class="line"><a name="l00716"></a><span class="lineno"> 716</span>  iterator_.clear_mask();</div><div class="line"><a name="l00717"></a><span class="lineno"> 717</span>  }</div><div class="line"><a name="l00718"></a><span class="lineno"> 718</span> </div><div class="line"><a name="l00720"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a7eaef0c562d8a25f5713fc1cffcc1ead"> 720</a></span>  <a class="code" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a></div><div class="line"><a name="l00721"></a><span class="lineno"> 721</span>  <span class="keywordtype">void</span> enable_mask() {</div><div class="line"><a name="l00722"></a><span class="lineno"> 722</span>  iterator_.enable_mask();</div><div class="line"><a name="l00723"></a><span class="lineno"> 723</span>  }</div><div class="line"><a name="l00724"></a><span class="lineno"> 724</span> </div><div class="line"><a name="l00726"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a9d7b7388e73405a4fddcd87dfeb7f407"> 726</a></span>  <a class="code" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a></div><div class="line"><a name="l00727"></a><span class="lineno"> 727</span>  <span class="keywordtype">void</span> set_mask(<a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#aef7b4321bef75c2ad969fedfb7e1aeee">Mask</a> <span class="keyword">const</span> &mask) {</div><div class="line"><a name="l00728"></a><span class="lineno"> 728</span>  iterator_.set_mask(mask);</div><div class="line"><a name="l00729"></a><span class="lineno"> 729</span>  }</div><div class="line"><a name="l00730"></a><span class="lineno"> 730</span> </div><div class="line"><a name="l00732"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#aea87c618a73e18e373dada2a1d33041c"> 732</a></span>  <a class="code" href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a></div><div class="line"><a name="l00733"></a><span class="lineno"> 733</span>  <span class="keywordtype">void</span> get_mask(<a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#aef7b4321bef75c2ad969fedfb7e1aeee">Mask</a> &mask) {</div><div class="line"><a name="l00734"></a><span class="lineno"> 734</span>  iterator_.get_mask(mask);</div><div class="line"><a name="l00735"></a><span class="lineno"> 735</span>  }</div><div class="line"><a name="l00736"></a><span class="lineno"> 736</span> </div><div class="line"><a name="l00738"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a380b3dc2403001e307a84f1634968f4e"> 738</a></span>  CUTLASS_DEVICE</div><div class="line"><a name="l00739"></a><span class="lineno"> 739</span>  <span class="keywordtype">void</span> load_with_pointer_offset(<a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#ae342607106778417432c4694b92736b3">Fragment</a> &frag, <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a31ddecaefe6832464ccd170834e398ca">Index</a> pointer_offset) {</div><div class="line"><a name="l00740"></a><span class="lineno"> 740</span>  iterator_.load_with_pointer_offset(frag, pointer_offset);</div><div class="line"><a name="l00741"></a><span class="lineno"> 741</span>  }</div><div class="line"><a name="l00742"></a><span class="lineno"> 742</span> </div><div class="line"><a name="l00744"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#ae399724b445824fda856949817f712c7"> 744</a></span>  CUTLASS_DEVICE</div><div class="line"><a name="l00745"></a><span class="lineno"> 745</span>  <span class="keywordtype">void</span> load(<a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#ae342607106778417432c4694b92736b3">Fragment</a> &frag) {</div><div class="line"><a name="l00746"></a><span class="lineno"> 746</span>  load_with_pointer_offset(frag, 0);</div><div class="line"><a name="l00747"></a><span class="lineno"> 747</span>  }</div><div class="line"><a name="l00748"></a><span class="lineno"> 748</span> </div><div class="line"><a name="l00750"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a0188cef96354b0b0b9db716b4a3e836f"> 750</a></span>  CUTLASS_DEVICE</div><div class="line"><a name="l00751"></a><span class="lineno"> 751</span>  <span class="keywordtype">void</span> store_with_pointer_offset(<a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#ae342607106778417432c4694b92736b3">Fragment</a> <span class="keyword">const</span> &frag, <a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a31ddecaefe6832464ccd170834e398ca">Index</a> pointer_offset) {</div><div class="line"><a name="l00752"></a><span class="lineno"> 752</span>  iterator_.store_with_pointer_offset(frag, pointer_offset);</div><div class="line"><a name="l00753"></a><span class="lineno"> 753</span>  }</div><div class="line"><a name="l00754"></a><span class="lineno"> 754</span> </div><div class="line"><a name="l00756"></a><span class="lineno"><a class="line" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a1b231e859447004c280cafe59f5016de"> 756</a></span>  CUTLASS_DEVICE</div><div class="line"><a name="l00757"></a><span class="lineno"> 757</span>  <span class="keywordtype">void</span> store(<a class="code" href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#ae342607106778417432c4694b92736b3">Fragment</a> <span class="keyword">const</span> &frag) {</div><div class="line"><a name="l00758"></a><span class="lineno"> 758</span>  store_with_pointer_offset(frag, 0);</div><div class="line"><a name="l00759"></a><span class="lineno"> 759</span>  }</div><div class="line"><a name="l00760"></a><span class="lineno"> 760</span> };</div><div class="line"><a name="l00761"></a><span class="lineno"> 761</span> </div><div class="line"><a name="l00763"></a><span class="lineno"> 763</span> </div><div class="line"><a name="l00764"></a><span class="lineno"> 764</span> } <span class="comment">// namespace threadblock</span></div><div class="line"><a name="l00765"></a><span class="lineno"> 765</span> } <span class="comment">// namespace transform</span></div><div class="line"><a name="l00766"></a><span class="lineno"> 766</span> } <span class="comment">// namespace cutlass</span></div><div class="line"><a name="l00767"></a><span class="lineno"> 767</span> </div><div class="ttc" id="classcutlass_1_1layout_1_1RowMajor_html_a4062a36ab044fdea058504ed52ee60b8"><div class="ttname"><a href="classcutlass_1_1layout_1_1RowMajor.html#a4062a36ab044fdea058504ed52ee60b8">cutlass::layout::RowMajor::LongIndex</a></div><div class="ttdeci">int64_t LongIndex</div><div class="ttdoc">Long index type used for offsets. </div><div class="ttdef"><b>Definition:</b> layout/matrix.h:62</div></div>
|
|
<div class="ttc" id="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1_html_a587fc2ae48d8793d5e563249242c16fa"><div class="ttname"><a href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a587fc2ae48d8793d5e563249242c16fa">cutlass::transform::threadblock::PredicatedTileIterator2dThreadTile< Shape_, Element_, layout::PitchLinear, AdvanceRank, ThreadMap_, Transpose_ >::TensorCoord</a></div><div class="ttdeci">typename Layout::TensorCoord TensorCoord</div><div class="ttdef"><b>Definition:</b> predicated_tile_iterator_2dthreadtile.h:163</div></div>
|
|
<div class="ttc" id="namespacecutlass_html"><div class="ttname"><a href="namespacecutlass.html">cutlass</a></div><div class="ttdef"><b>Definition:</b> aligned_buffer.h:35</div></div>
|
|
<div class="ttc" id="structcutlass_1_1layout_1_1PitchLinearCoord_html"><div class="ttname"><a href="structcutlass_1_1layout_1_1PitchLinearCoord.html">cutlass::layout::PitchLinearCoord</a></div><div class="ttdoc">Coordinate in pitch-linear space. </div><div class="ttdef"><b>Definition:</b> pitch_linear.h:52</div></div>
|
|
<div class="ttc" id="structcutlass_1_1platform_1_1remove__const_html_ac3662947fa50251daf58240a9c798085"><div class="ttname"><a href="structcutlass_1_1platform_1_1remove__const.html#ac3662947fa50251daf58240a9c798085">cutlass::platform::remove_const::type</a></div><div class="ttdeci">T type</div><div class="ttdef"><b>Definition:</b> platform.h:351</div></div>
|
|
<div class="ttc" id="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11_html_a72e72c4beb8186dbfed16b7dd2008ac6"><div class="ttname"><a href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a72e72c4beb8186dbfed16b7dd2008ac6">cutlass::transform::threadblock::PredicatedTileIterator2dThreadTile< Shape_, Element_, layout::RowMajor, AdvanceRank, ThreadMap_, Transpose_ >::Shape</a></div><div class="ttdeci">Shape_ Shape</div><div class="ttdef"><b>Definition:</b> predicated_tile_iterator_2dthreadtile.h:590</div></div>
|
|
<div class="ttc" id="transpose_8h_html"><div class="ttname"><a href="transpose_8h.html">transpose.h</a></div><div class="ttdoc">Basic copy routines for tensor views. </div></div>
|
|
<div class="ttc" id="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11_html_a33d57eb89928e9e38a9be7190452928c"><div class="ttname"><a href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a33d57eb89928e9e38a9be7190452928c">cutlass::transform::threadblock::PredicatedTileIterator2dThreadTile< Shape_, Element_, layout::RowMajor, AdvanceRank, ThreadMap_, Transpose_ >::LongIndex</a></div><div class="ttdeci">typename Layout::LongIndex LongIndex</div><div class="ttdef"><b>Definition:</b> predicated_tile_iterator_2dthreadtile.h:598</div></div>
|
|
<div class="ttc" id="classcutlass_1_1layout_1_1PitchLinear_html"><div class="ttname"><a href="classcutlass_1_1layout_1_1PitchLinear.html">cutlass::layout::PitchLinear</a></div><div class="ttdoc">Mapping function for pitch-linear memory. </div><div class="ttdef"><b>Definition:</b> pitch_linear.h:163</div></div>
|
|
<div class="ttc" id="namespacecutlass_html_a7419519fa453a121dfa5f26bf87318d9"><div class="ttname"><a href="namespacecutlass.html#a7419519fa453a121dfa5f26bf87318d9">cutlass::make_Coord</a></div><div class="ttdeci">CUTLASS_HOST_DEVICE Coord< 1 > make_Coord(int _0)</div><div class="ttdoc">Helper to make a 2-element coordinate. </div><div class="ttdef"><b>Definition:</b> coord.h:387</div></div>
|
|
<div class="ttc" id="classcutlass_1_1layout_1_1ColumnMajor_html_a4cc90aa67c4692f0a2cd9f59b8a07997"><div class="ttname"><a href="classcutlass_1_1layout_1_1ColumnMajor.html#a4cc90aa67c4692f0a2cd9f59b8a07997">cutlass::layout::ColumnMajor::LongIndex</a></div><div class="ttdeci">int64_t LongIndex</div><div class="ttdoc">Long index type used for offsets. </div><div class="ttdef"><b>Definition:</b> layout/matrix.h:154</div></div>
|
|
<div class="ttc" id="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1_html_a715b78e9d8587c4c59e5337d33eb7025"><div class="ttname"><a href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a715b78e9d8587c4c59e5337d33eb7025">cutlass::transform::threadblock::PredicatedTileIterator2dThreadTile< Shape_, Element_, layout::PitchLinear, AdvanceRank, ThreadMap_, Transpose_ >::Index</a></div><div class="ttdeci">typename Layout::Index Index</div><div class="ttdef"><b>Definition:</b> predicated_tile_iterator_2dthreadtile.h:158</div></div>
|
|
<div class="ttc" id="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9_html_a50525f7a7785ed266f6e09290a6f0634"><div class="ttname"><a href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a50525f7a7785ed266f6e09290a6f0634">cutlass::transform::threadblock::PredicatedTileIterator2dThreadTile< Shape_, Element_, layout::ColumnMajor, AdvanceRank, ThreadMap_, Transpose_ >::AccessType</a></div><div class="ttdeci">typename UnderlyingIterator::AccessType AccessType</div><div class="ttdef"><b>Definition:</b> predicated_tile_iterator_2dthreadtile.h:422</div></div>
|
|
<div class="ttc" id="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1_html_aeae69750c7c478c23653af6e785892af"><div class="ttname"><a href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#aeae69750c7c478c23653af6e785892af">cutlass::transform::threadblock::PredicatedTileIterator2dThreadTile< Shape_, Element_, layout::PitchLinear, AdvanceRank, ThreadMap_, Transpose_ >::Pointer</a></div><div class="ttdeci">Element * Pointer</div><div class="ttdef"><b>Definition:</b> predicated_tile_iterator_2dthreadtile.h:165</div></div>
|
|
<div class="ttc" id="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileAccessIterator2dThreadTile_html"><div class="ttname"><a href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileAccessIterator2dThreadTile.html">cutlass::transform::threadblock::PredicatedTileAccessIterator2dThreadTile< Shape, Element, Layout, kAdvanceRank, ThreadMap, AccessType ></a></div></div>
|
|
<div class="ttc" id="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11_html_a6f8d0fbaf19bf59d0d76bc3937609443"><div class="ttname"><a href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a6f8d0fbaf19bf59d0d76bc3937609443">cutlass::transform::threadblock::PredicatedTileIterator2dThreadTile< Shape_, Element_, layout::RowMajor, AdvanceRank, ThreadMap_, Transpose_ >::TensorCoord</a></div><div class="ttdeci">typename Layout::TensorCoord TensorCoord</div><div class="ttdef"><b>Definition:</b> predicated_tile_iterator_2dthreadtile.h:602</div></div>
|
|
<div class="ttc" id="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11_html_ae342607106778417432c4694b92736b3"><div class="ttname"><a href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#ae342607106778417432c4694b92736b3">cutlass::transform::threadblock::PredicatedTileIterator2dThreadTile< Shape_, Element_, layout::RowMajor, AdvanceRank, ThreadMap_, Transpose_ >::Fragment</a></div><div class="ttdeci">cutlass::Array< Element, ThreadMap::Iterations::kCount *ThreadMap::ThreadAccessShape::kCount > Fragment</div><div class="ttdoc">Fragment object to be loaded or stored. </div><div class="ttdef"><b>Definition:</b> predicated_tile_iterator_2dthreadtile.h:619</div></div>
|
|
<div class="ttc" id="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11_html_a31ddecaefe6832464ccd170834e398ca"><div class="ttname"><a href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a31ddecaefe6832464ccd170834e398ca">cutlass::transform::threadblock::PredicatedTileIterator2dThreadTile< Shape_, Element_, layout::RowMajor, AdvanceRank, ThreadMap_, Transpose_ >::Index</a></div><div class="ttdeci">typename Layout::Index Index</div><div class="ttdef"><b>Definition:</b> predicated_tile_iterator_2dthreadtile.h:597</div></div>
|
|
<div class="ttc" id="classcutlass_1_1layout_1_1ColumnMajor_html"><div class="ttname"><a href="classcutlass_1_1layout_1_1ColumnMajor.html">cutlass::layout::ColumnMajor</a></div><div class="ttdoc">Mapping function for column-major matrices. </div><div class="ttdef"><b>Definition:</b> layout/matrix.h:142</div></div>
|
|
<div class="ttc" id="structcutlass_1_1layout_1_1PitchLinearShape_html"><div class="ttname"><a href="structcutlass_1_1layout_1_1PitchLinearShape.html">cutlass::layout::PitchLinearShape</a></div><div class="ttdoc">Template defining a shape used by pitch-linear operators. </div><div class="ttdef"><b>Definition:</b> pitch_linear.h:43</div></div>
|
|
<div class="ttc" id="cutlass_8h_html_a4b1c9f25ab6eaa25e1f2258dd63e6ce4"><div class="ttname"><a href="cutlass_8h.html#a4b1c9f25ab6eaa25e1f2258dd63e6ce4">CUTLASS_PRAGMA_UNROLL</a></div><div class="ttdeci">#define CUTLASS_PRAGMA_UNROLL</div><div class="ttdef"><b>Definition:</b> cutlass.h:110</div></div>
|
|
<div class="ttc" id="classcutlass_1_1layout_1_1RowMajor_html_aa49e242b14b4f482bc6bdd082acfb576"><div class="ttname"><a href="classcutlass_1_1layout_1_1RowMajor.html#aa49e242b14b4f482bc6bdd082acfb576">cutlass::layout::RowMajor::Index</a></div><div class="ttdeci">int32_t Index</div><div class="ttdoc">Index type used for coordinates. </div><div class="ttdef"><b>Definition:</b> layout/matrix.h:59</div></div>
|
|
<div class="ttc" id="namespacecutlass_html_a1c7a9e66ca7b5dc7413ea3b8f349530a"><div class="ttname"><a href="namespacecutlass.html#a1c7a9e66ca7b5dc7413ea3b8f349530a">cutlass::operator++</a></div><div class="ttdeci">CUTLASS_HOST_DEVICE half_t & operator++(half_t &lhs)</div><div class="ttdef"><b>Definition:</b> half.h:694</div></div>
|
|
<div class="ttc" id="classcutlass_1_1layout_1_1PitchLinear_html_a1a4b31740e77b3c03925f507650978ea"><div class="ttname"><a href="classcutlass_1_1layout_1_1PitchLinear.html#a1a4b31740e77b3c03925f507650978ea">cutlass::layout::PitchLinear::LongIndex</a></div><div class="ttdeci">int64_t LongIndex</div><div class="ttdoc">Long index type used for offsets. </div><div class="ttdef"><b>Definition:</b> pitch_linear.h:175</div></div>
|
|
<div class="ttc" id="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1_html_a560fc3bce354a657f12b423969eb5b37"><div class="ttname"><a href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a560fc3bce354a657f12b423969eb5b37">cutlass::transform::threadblock::PredicatedTileIterator2dThreadTile< Shape_, Element_, layout::PitchLinear, AdvanceRank, ThreadMap_, Transpose_ >::LongIndex</a></div><div class="ttdeci">typename Layout::LongIndex LongIndex</div><div class="ttdef"><b>Definition:</b> predicated_tile_iterator_2dthreadtile.h:159</div></div>
|
|
<div class="ttc" id="classcutlass_1_1TensorView_html"><div class="ttname"><a href="classcutlass_1_1TensorView.html">cutlass::TensorView< Element, Layout ></a></div></div>
|
|
<div class="ttc" id="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9_html_abbc6f561480d48c05001038ab7a0bebb"><div class="ttname"><a href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#abbc6f561480d48c05001038ab7a0bebb">cutlass::transform::threadblock::PredicatedTileIterator2dThreadTile< Shape_, Element_, layout::ColumnMajor, AdvanceRank, ThreadMap_, Transpose_ >::Pointer</a></div><div class="ttdeci">Element * Pointer</div><div class="ttdef"><b>Definition:</b> predicated_tile_iterator_2dthreadtile.h:410</div></div>
|
|
<div class="ttc" id="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1_html_a20a0f1a208b301ecd6f29cced4f7cbb7"><div class="ttname"><a href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a20a0f1a208b301ecd6f29cced4f7cbb7">cutlass::transform::threadblock::PredicatedTileIterator2dThreadTile< Shape_, Element_, layout::PitchLinear, AdvanceRank, ThreadMap_, Transpose_ >::Mask</a></div><div class="ttdeci">typename TileAccessIterator::Mask Mask</div><div class="ttdoc">Predicate vector stores mask to guard accesses. </div><div class="ttdef"><b>Definition:</b> predicated_tile_iterator_2dthreadtile.h:192</div></div>
|
|
<div class="ttc" id="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9_html_a28c6395acf81c013c8089414a91b66e7"><div class="ttname"><a href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a28c6395acf81c013c8089414a91b66e7">cutlass::transform::threadblock::PredicatedTileIterator2dThreadTile< Shape_, Element_, layout::ColumnMajor, AdvanceRank, ThreadMap_, Transpose_ >::LongIndex</a></div><div class="ttdeci">typename Layout::LongIndex LongIndex</div><div class="ttdef"><b>Definition:</b> predicated_tile_iterator_2dthreadtile.h:404</div></div>
|
|
<div class="ttc" id="classcutlass_1_1TensorRef_html"><div class="ttname"><a href="classcutlass_1_1TensorRef.html">cutlass::TensorRef< Element, Layout ></a></div></div>
|
|
<div class="ttc" id="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9_html_a7912240a9802251763983a1fa4a297f1"><div class="ttname"><a href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a7912240a9802251763983a1fa4a297f1">cutlass::transform::threadblock::PredicatedTileIterator2dThreadTile< Shape_, Element_, layout::ColumnMajor, AdvanceRank, ThreadMap_, Transpose_ >::Index</a></div><div class="ttdeci">typename Layout::Index Index</div><div class="ttdef"><b>Definition:</b> predicated_tile_iterator_2dthreadtile.h:403</div></div>
|
|
<div class="ttc" id="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11_html_aadb4b95b45598045f882a077fb5c587e"><div class="ttname"><a href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#aadb4b95b45598045f882a077fb5c587e">cutlass::transform::threadblock::PredicatedTileIterator2dThreadTile< Shape_, Element_, layout::RowMajor, AdvanceRank, ThreadMap_, Transpose_ >::AccessType</a></div><div class="ttdeci">typename UnderlyingIterator::AccessType AccessType</div><div class="ttdef"><b>Definition:</b> predicated_tile_iterator_2dthreadtile.h:616</div></div>
|
|
<div class="ttc" id="cutlass_8h_html_a28c2443a142676d3d71effdae1a986b1"><div class="ttname"><a href="cutlass_8h.html#a28c2443a142676d3d71effdae1a986b1">CUTLASS_HOST_DEVICE</a></div><div class="ttdeci">#define CUTLASS_HOST_DEVICE</div><div class="ttdef"><b>Definition:</b> cutlass.h:89</div></div>
|
|
<div class="ttc" id="classcutlass_1_1transform_1_1thread_1_1Transpose_html"><div class="ttname"><a href="classcutlass_1_1transform_1_1thread_1_1Transpose.html">cutlass::transform::thread::Transpose</a></div><div class="ttdoc">Transforms a fragment by doing a transpose. </div><div class="ttdef"><b>Definition:</b> transpose.h:39</div></div>
|
|
<div class="ttc" id="platform_8h_html_adde4c9ea91b753491851361a4198c009"><div class="ttname"><a href="platform_8h.html#adde4c9ea91b753491851361a4198c009">static_assert</a></div><div class="ttdeci">#define static_assert(__e, __m)</div><div class="ttdef"><b>Definition:</b> platform.h:153</div></div>
|
|
<div class="ttc" id="classcutlass_1_1layout_1_1PitchLinear_html_a9d1dfd7b6d3b2b651009dcba8f5fd5cd"><div class="ttname"><a href="classcutlass_1_1layout_1_1PitchLinear.html#a9d1dfd7b6d3b2b651009dcba8f5fd5cd">cutlass::layout::PitchLinear::Index</a></div><div class="ttdeci">int32_t Index</div><div class="ttdoc">Index type used for coordinates. </div><div class="ttdef"><b>Definition:</b> pitch_linear.h:172</div></div>
|
|
<div class="ttc" id="classcutlass_1_1layout_1_1RowMajor_html"><div class="ttname"><a href="classcutlass_1_1layout_1_1RowMajor.html">cutlass::layout::RowMajor</a></div><div class="ttdoc">Mapping function for row-major matrices. </div><div class="ttdef"><b>Definition:</b> layout/matrix.h:50</div></div>
|
|
<div class="ttc" id="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1_html_a837b46168668a29afce4f36f5c4c598f"><div class="ttname"><a href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a837b46168668a29afce4f36f5c4c598f">cutlass::transform::threadblock::PredicatedTileIterator2dThreadTile< Shape_, Element_, layout::PitchLinear, AdvanceRank, ThreadMap_, Transpose_ >::Shape</a></div><div class="ttdeci">Shape_ Shape</div><div class="ttdef"><b>Definition:</b> predicated_tile_iterator_2dthreadtile.h:152</div></div>
|
|
<div class="ttc" id="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9_html_a8275a7f49a107c4122a4d3278aba94ce"><div class="ttname"><a href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a8275a7f49a107c4122a4d3278aba94ce">cutlass::transform::threadblock::PredicatedTileIterator2dThreadTile< Shape_, Element_, layout::ColumnMajor, AdvanceRank, ThreadMap_, Transpose_ >::Mask</a></div><div class="ttdeci">typename UnderlyingIterator::Mask Mask</div><div class="ttdoc">Predicate vector stores mask to guard accesses. </div><div class="ttdef"><b>Definition:</b> predicated_tile_iterator_2dthreadtile.h:428</div></div>
|
|
<div class="ttc" id="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11_html_af81f97366f2524518d48955568d45262"><div class="ttname"><a href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#af81f97366f2524518d48955568d45262">cutlass::transform::threadblock::PredicatedTileIterator2dThreadTile< Shape_, Element_, layout::RowMajor, AdvanceRank, ThreadMap_, Transpose_ >::NonConstPointer</a></div><div class="ttdeci">typename platform::remove_const< Element >::type * NonConstPointer</div><div class="ttdef"><b>Definition:</b> predicated_tile_iterator_2dthreadtile.h:605</div></div>
|
|
<div class="ttc" id="classcutlass_1_1layout_1_1ColumnMajor_html_a6409273d6b40acb27aff0564eb038336"><div class="ttname"><a href="classcutlass_1_1layout_1_1ColumnMajor.html#a6409273d6b40acb27aff0564eb038336">cutlass::layout::ColumnMajor::Index</a></div><div class="ttdeci">int32_t Index</div><div class="ttdoc">Index type used for coordinates. </div><div class="ttdef"><b>Definition:</b> layout/matrix.h:151</div></div>
|
|
<div class="ttc" id="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11_html_a9ecc09cd70be9f7e56f309465d878f79"><div class="ttname"><a href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#a9ecc09cd70be9f7e56f309465d878f79">cutlass::transform::threadblock::PredicatedTileIterator2dThreadTile< Shape_, Element_, layout::RowMajor, AdvanceRank, ThreadMap_, Transpose_ >::Pointer</a></div><div class="ttdeci">Element * Pointer</div><div class="ttdef"><b>Definition:</b> predicated_tile_iterator_2dthreadtile.h:604</div></div>
|
|
<div class="ttc" id="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11_html_aef7b4321bef75c2ad969fedfb7e1aeee"><div class="ttname"><a href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_013671177d6219bfeb0e1b4dc4c1b5bf11.html#aef7b4321bef75c2ad969fedfb7e1aeee">cutlass::transform::threadblock::PredicatedTileIterator2dThreadTile< Shape_, Element_, layout::RowMajor, AdvanceRank, ThreadMap_, Transpose_ >::Mask</a></div><div class="ttdeci">typename UnderlyingIterator::Mask Mask</div><div class="ttdoc">Predicate vector stores mask to guard accesses. </div><div class="ttdef"><b>Definition:</b> predicated_tile_iterator_2dthreadtile.h:622</div></div>
|
|
<div class="ttc" id="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1_html_a1065b487ba57b47e4d03442a91af8f67"><div class="ttname"><a href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a1065b487ba57b47e4d03442a91af8f67">cutlass::transform::threadblock::PredicatedTileIterator2dThreadTile< Shape_, Element_, layout::PitchLinear, AdvanceRank, ThreadMap_, Transpose_ >::NonConstPointer</a></div><div class="ttdeci">typename platform::remove_const< Element >::type * NonConstPointer</div><div class="ttdef"><b>Definition:</b> predicated_tile_iterator_2dthreadtile.h:166</div></div>
|
|
<div class="ttc" id="predicated__tile__access__iterator__2dthreadtile_8h_html"><div class="ttname"><a href="predicated__tile__access__iterator__2dthreadtile_8h.html">predicated_tile_access_iterator_2dthreadtile.h</a></div><div class="ttdoc">Templates calculating the address and predicates to the load of tiles from pitch-linear rank=2 tensor...</div></div>
|
|
<div class="ttc" id="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_html"><div class="ttname"><a href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile.html">cutlass::transform::threadblock::PredicatedTileIterator2dThreadTile</a></div><div class="ttdef"><b>Definition:</b> predicated_tile_iterator_2dthreadtile.h:133</div></div>
|
|
<div class="ttc" id="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9_html_affebbfbc62a67c328d2ceceb382361c3"><div class="ttname"><a href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#affebbfbc62a67c328d2ceceb382361c3">cutlass::transform::threadblock::PredicatedTileIterator2dThreadTile< Shape_, Element_, layout::ColumnMajor, AdvanceRank, ThreadMap_, Transpose_ >::Shape</a></div><div class="ttdeci">Shape_ Shape</div><div class="ttdef"><b>Definition:</b> predicated_tile_iterator_2dthreadtile.h:396</div></div>
|
|
<div class="ttc" id="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9_html_a3d483d123a36af834026cf40da581a11"><div class="ttname"><a href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a3d483d123a36af834026cf40da581a11">cutlass::transform::threadblock::PredicatedTileIterator2dThreadTile< Shape_, Element_, layout::ColumnMajor, AdvanceRank, ThreadMap_, Transpose_ >::NonConstPointer</a></div><div class="ttdeci">typename platform::remove_const< Element >::type * NonConstPointer</div><div class="ttdef"><b>Definition:</b> predicated_tile_iterator_2dthreadtile.h:411</div></div>
|
|
<div class="ttc" id="structcutlass_1_1MatrixCoord_html"><div class="ttname"><a href="structcutlass_1_1MatrixCoord.html">cutlass::MatrixCoord</a></div><div class="ttdef"><b>Definition:</b> matrix_coord.h:39</div></div>
|
|
<div class="ttc" id="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1_html_a0c7351825135690fa82c4fba5f9447c7"><div class="ttname"><a href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_017a517f3c73efd795ab05059cc9b111e1.html#a0c7351825135690fa82c4fba5f9447c7">cutlass::transform::threadblock::PredicatedTileIterator2dThreadTile< Shape_, Element_, layout::PitchLinear, AdvanceRank, ThreadMap_, Transpose_ >::Fragment</a></div><div class="ttdeci">cutlass::Array< Element, ThreadMap::Iterations::kCount *ThreadMap::ThreadAccessShape::kCount > Fragment</div><div class="ttdoc">Fragment object to be loaded or stored. </div><div class="ttdef"><b>Definition:</b> predicated_tile_iterator_2dthreadtile.h:189</div></div>
|
|
<div class="ttc" id="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9_html_a1568b1a89e72af084604e3c209b08f11"><div class="ttname"><a href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a1568b1a89e72af084604e3c209b08f11">cutlass::transform::threadblock::PredicatedTileIterator2dThreadTile< Shape_, Element_, layout::ColumnMajor, AdvanceRank, ThreadMap_, Transpose_ >::TensorCoord</a></div><div class="ttdeci">typename Layout::TensorCoord TensorCoord</div><div class="ttdef"><b>Definition:</b> predicated_tile_iterator_2dthreadtile.h:408</div></div>
|
|
<div class="ttc" id="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9_html_a9be82cb5e5c9a215b5862dba28dc57f4"><div class="ttname"><a href="classcutlass_1_1transform_1_1threadblock_1_1PredicatedTileIterator2dThreadTile_3_01Shape___00_0165b39a630d10785a3558406f9adb99b9.html#a9be82cb5e5c9a215b5862dba28dc57f4">cutlass::transform::threadblock::PredicatedTileIterator2dThreadTile< Shape_, Element_, layout::ColumnMajor, AdvanceRank, ThreadMap_, Transpose_ >::Fragment</a></div><div class="ttdeci">cutlass::Array< Element, ThreadMap::Iterations::kCount *ThreadMap::ThreadAccessShape::kCount > Fragment</div><div class="ttdoc">Fragment object to be loaded or stored. </div><div class="ttdef"><b>Definition:</b> predicated_tile_iterator_2dthreadtile.h:425</div></div>
|
|
</div><!-- fragment --></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>
|