Skip to content

Commit d010279

Browse files
Update docs
1 parent 5682e96 commit d010279

File tree

5 files changed

+138
-6
lines changed

5 files changed

+138
-6
lines changed

_sources/autoapi/tilelang/language/reduce/index.rst.txt

Lines changed: 48 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,9 @@ Functions
2020
tilelang.language.reduce.reduce_sum
2121
tilelang.language.reduce.reduce_abssum
2222
tilelang.language.reduce.reduce_absmax
23+
tilelang.language.reduce.reduce_bitand
24+
tilelang.language.reduce.reduce_bitor
25+
tilelang.language.reduce.reduce_bitxor
2326
tilelang.language.reduce.cumsum_fragment
2427
tilelang.language.reduce.cumsum
2528
tilelang.language.reduce.finalize_reducer
@@ -138,6 +141,51 @@ Module Contents
138141
:rtype: tir.Call
139142

140143

144+
.. py:function:: reduce_bitand(buffer, out, dim = -1, clear = True)
145+
146+
Perform reduce bitwise-and on input buffer, store the result to output buffer.
147+
148+
:param buffer: The input buffer
149+
:type buffer: tir.Buffer
150+
:param out: The output buffer
151+
:type out: tir.Buffer
152+
:param dim: The dimension to perform reduce on
153+
:type dim: int
154+
155+
:returns: Handle to the reduction operation
156+
:rtype: tir.Call
157+
158+
159+
.. py:function:: reduce_bitor(buffer, out, dim = -1, clear = True)
160+
161+
Perform reduce bitwise-or on input buffer, store the result to output buffer.
162+
163+
:param buffer: The input buffer
164+
:type buffer: tir.Buffer
165+
:param out: The output buffer
166+
:type out: tir.Buffer
167+
:param dim: The dimension to perform reduce on
168+
:type dim: int
169+
170+
:returns: Handle to the reduction operation
171+
:rtype: tir.Call
172+
173+
174+
.. py:function:: reduce_bitxor(buffer, out, dim = -1, clear = True)
175+
176+
Perform reduce bitwise-xor on input buffer, store the result to output buffer.
177+
178+
:param buffer: The input buffer
179+
:type buffer: tir.Buffer
180+
:param out: The output buffer
181+
:type out: tir.Buffer
182+
:param dim: The dimension to perform reduce on
183+
:type dim: int
184+
185+
:returns: Handle to the reduction operation
186+
:rtype: tir.Call
187+
188+
141189
.. py:function:: cumsum_fragment(src, dst, dim, reverse)
142190
143191
.. py:function:: cumsum(src, dst = None, dim = 0, reverse = False)

autoapi/tilelang/language/reduce/index.html

Lines changed: 81 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -502,13 +502,22 @@ <h2>Functions<a class="headerlink" href="#functions" title="Link to this heading
502502
<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.language.reduce.reduce_absmax" title="tilelang.language.reduce.reduce_absmax"><code class="xref py py-obj docutils literal notranslate"><span class="pre">reduce_absmax</span></code></a>(buffer, out[, dim, clear])</p></td>
503503
<td><p>Perform reduce absolute max on input buffer, store the result to output buffer.</p></td>
504504
</tr>
505-
<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.language.reduce.cumsum_fragment" title="tilelang.language.reduce.cumsum_fragment"><code class="xref py py-obj docutils literal notranslate"><span class="pre">cumsum_fragment</span></code></a>(src, dst, dim, reverse)</p></td>
505+
<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.language.reduce.reduce_bitand" title="tilelang.language.reduce.reduce_bitand"><code class="xref py py-obj docutils literal notranslate"><span class="pre">reduce_bitand</span></code></a>(buffer, out[, dim, clear])</p></td>
506+
<td><p>Perform reduce bitwise-and on input buffer, store the result to output buffer.</p></td>
507+
</tr>
508+
<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.language.reduce.reduce_bitor" title="tilelang.language.reduce.reduce_bitor"><code class="xref py py-obj docutils literal notranslate"><span class="pre">reduce_bitor</span></code></a>(buffer, out[, dim, clear])</p></td>
509+
<td><p>Perform reduce bitwise-or on input buffer, store the result to output buffer.</p></td>
510+
</tr>
511+
<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.language.reduce.reduce_bitxor" title="tilelang.language.reduce.reduce_bitxor"><code class="xref py py-obj docutils literal notranslate"><span class="pre">reduce_bitxor</span></code></a>(buffer, out[, dim, clear])</p></td>
512+
<td><p>Perform reduce bitwise-xor on input buffer, store the result to output buffer.</p></td>
513+
</tr>
514+
<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.language.reduce.cumsum_fragment" title="tilelang.language.reduce.cumsum_fragment"><code class="xref py py-obj docutils literal notranslate"><span class="pre">cumsum_fragment</span></code></a>(src, dst, dim, reverse)</p></td>
506515
<td><p></p></td>
507516
</tr>
508-
<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.language.reduce.cumsum" title="tilelang.language.reduce.cumsum"><code class="xref py py-obj docutils literal notranslate"><span class="pre">cumsum</span></code></a>(src[, dst, dim, reverse])</p></td>
517+
<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.language.reduce.cumsum" title="tilelang.language.reduce.cumsum"><code class="xref py py-obj docutils literal notranslate"><span class="pre">cumsum</span></code></a>(src[, dst, dim, reverse])</p></td>
509518
<td><p>Compute the cumulative sum of <cite>src</cite> along <cite>dim</cite>, writing results to <cite>dst</cite>.</p></td>
510519
</tr>
511-
<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.language.reduce.finalize_reducer" title="tilelang.language.reduce.finalize_reducer"><code class="xref py py-obj docutils literal notranslate"><span class="pre">finalize_reducer</span></code></a>(reducer)</p></td>
520+
<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.language.reduce.finalize_reducer" title="tilelang.language.reduce.finalize_reducer"><code class="xref py py-obj docutils literal notranslate"><span class="pre">finalize_reducer</span></code></a>(reducer)</p></td>
512521
<td><p>Finalize a reducer buffer by emitting the <cite>tl.finalize_reducer</cite> intrinsic.</p></td>
513522
</tr>
514523
</tbody>
@@ -674,6 +683,72 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
674683
</dl>
675684
</dd></dl>
676685

686+
<dl class="py function">
687+
<dt class="sig sig-object py" id="tilelang.language.reduce.reduce_bitand">
688+
<span class="sig-prename descclassname"><span class="pre">tilelang.language.reduce.</span></span><span class="sig-name descname"><span class="pre">reduce_bitand</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">buffer</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">out</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">dim</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">-1</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">clear</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">True</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.language.reduce.reduce_bitand" title="Link to this definition"></a></dt>
689+
<dd><p>Perform reduce bitwise-and on input buffer, store the result to output buffer.</p>
690+
<dl class="field-list simple">
691+
<dt class="field-odd">Parameters<span class="colon">:</span></dt>
692+
<dd class="field-odd"><ul class="simple">
693+
<li><p><strong>buffer</strong> (<em>tir.Buffer</em>) – The input buffer</p></li>
694+
<li><p><strong>out</strong> (<em>tir.Buffer</em>) – The output buffer</p></li>
695+
<li><p><strong>dim</strong> (<em>int</em>) – The dimension to perform reduce on</p></li>
696+
<li><p><strong>clear</strong> (<em>bool</em>)</p></li>
697+
</ul>
698+
</dd>
699+
<dt class="field-even">Returns<span class="colon">:</span></dt>
700+
<dd class="field-even"><p>Handle to the reduction operation</p>
701+
</dd>
702+
<dt class="field-odd">Return type<span class="colon">:</span></dt>
703+
<dd class="field-odd"><p>tir.Call</p>
704+
</dd>
705+
</dl>
706+
</dd></dl>
707+
708+
<dl class="py function">
709+
<dt class="sig sig-object py" id="tilelang.language.reduce.reduce_bitor">
710+
<span class="sig-prename descclassname"><span class="pre">tilelang.language.reduce.</span></span><span class="sig-name descname"><span class="pre">reduce_bitor</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">buffer</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">out</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">dim</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">-1</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">clear</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">True</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.language.reduce.reduce_bitor" title="Link to this definition"></a></dt>
711+
<dd><p>Perform reduce bitwise-or on input buffer, store the result to output buffer.</p>
712+
<dl class="field-list simple">
713+
<dt class="field-odd">Parameters<span class="colon">:</span></dt>
714+
<dd class="field-odd"><ul class="simple">
715+
<li><p><strong>buffer</strong> (<em>tir.Buffer</em>) – The input buffer</p></li>
716+
<li><p><strong>out</strong> (<em>tir.Buffer</em>) – The output buffer</p></li>
717+
<li><p><strong>dim</strong> (<em>int</em>) – The dimension to perform reduce on</p></li>
718+
<li><p><strong>clear</strong> (<em>bool</em>)</p></li>
719+
</ul>
720+
</dd>
721+
<dt class="field-even">Returns<span class="colon">:</span></dt>
722+
<dd class="field-even"><p>Handle to the reduction operation</p>
723+
</dd>
724+
<dt class="field-odd">Return type<span class="colon">:</span></dt>
725+
<dd class="field-odd"><p>tir.Call</p>
726+
</dd>
727+
</dl>
728+
</dd></dl>
729+
730+
<dl class="py function">
731+
<dt class="sig sig-object py" id="tilelang.language.reduce.reduce_bitxor">
732+
<span class="sig-prename descclassname"><span class="pre">tilelang.language.reduce.</span></span><span class="sig-name descname"><span class="pre">reduce_bitxor</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">buffer</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">out</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">dim</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">-1</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">clear</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">True</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.language.reduce.reduce_bitxor" title="Link to this definition"></a></dt>
733+
<dd><p>Perform reduce bitwise-xor on input buffer, store the result to output buffer.</p>
734+
<dl class="field-list simple">
735+
<dt class="field-odd">Parameters<span class="colon">:</span></dt>
736+
<dd class="field-odd"><ul class="simple">
737+
<li><p><strong>buffer</strong> (<em>tir.Buffer</em>) – The input buffer</p></li>
738+
<li><p><strong>out</strong> (<em>tir.Buffer</em>) – The output buffer</p></li>
739+
<li><p><strong>dim</strong> (<em>int</em>) – The dimension to perform reduce on</p></li>
740+
<li><p><strong>clear</strong> (<em>bool</em>)</p></li>
741+
</ul>
742+
</dd>
743+
<dt class="field-even">Returns<span class="colon">:</span></dt>
744+
<dd class="field-even"><p>Handle to the reduction operation</p>
745+
</dd>
746+
<dt class="field-odd">Return type<span class="colon">:</span></dt>
747+
<dd class="field-odd"><p>tir.Call</p>
748+
</dd>
749+
</dl>
750+
</dd></dl>
751+
677752
<dl class="py function">
678753
<dt class="sig sig-object py" id="tilelang.language.reduce.cumsum_fragment">
679754
<span class="sig-prename descclassname"><span class="pre">tilelang.language.reduce.</span></span><span class="sig-name descname"><span class="pre">cumsum_fragment</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">src</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">dst</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">dim</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">reverse</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.language.reduce.cumsum_fragment" title="Link to this definition"></a></dt>
@@ -824,6 +899,9 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
824899
<li><a class="reference internal" href="#tilelang.language.reduce.reduce_sum"><code class="docutils literal notranslate"><span class="pre">reduce_sum()</span></code></a></li>
825900
<li><a class="reference internal" href="#tilelang.language.reduce.reduce_abssum"><code class="docutils literal notranslate"><span class="pre">reduce_abssum()</span></code></a></li>
826901
<li><a class="reference internal" href="#tilelang.language.reduce.reduce_absmax"><code class="docutils literal notranslate"><span class="pre">reduce_absmax()</span></code></a></li>
902+
<li><a class="reference internal" href="#tilelang.language.reduce.reduce_bitand"><code class="docutils literal notranslate"><span class="pre">reduce_bitand()</span></code></a></li>
903+
<li><a class="reference internal" href="#tilelang.language.reduce.reduce_bitor"><code class="docutils literal notranslate"><span class="pre">reduce_bitor()</span></code></a></li>
904+
<li><a class="reference internal" href="#tilelang.language.reduce.reduce_bitxor"><code class="docutils literal notranslate"><span class="pre">reduce_bitxor()</span></code></a></li>
827905
<li><a class="reference internal" href="#tilelang.language.reduce.cumsum_fragment"><code class="docutils literal notranslate"><span class="pre">cumsum_fragment()</span></code></a></li>
828906
<li><a class="reference internal" href="#tilelang.language.reduce.cumsum"><code class="docutils literal notranslate"><span class="pre">cumsum()</span></code></a></li>
829907
<li><a class="reference internal" href="#tilelang.language.reduce.finalize_reducer"><code class="docutils literal notranslate"><span class="pre">finalize_reducer()</span></code></a></li>

genindex.html

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4195,6 +4195,12 @@ <h2>R</h2>
41954195
<li><a href="autoapi/tilelang/language/reduce/index.html#tilelang.language.reduce.reduce_abssum">reduce_abssum() (in module tilelang.language.reduce)</a>
41964196
</li>
41974197
<li><a href="autoapi/tilelang/carver/roller/shape_inference/tir/index.html#tilelang.carver.roller.shape_inference.tir.InputShapeInference.reduce_axes">reduce_axes (tilelang.carver.roller.shape_inference.tir.InputShapeInference attribute)</a>
4198+
</li>
4199+
<li><a href="autoapi/tilelang/language/reduce/index.html#tilelang.language.reduce.reduce_bitand">reduce_bitand() (in module tilelang.language.reduce)</a>
4200+
</li>
4201+
<li><a href="autoapi/tilelang/language/reduce/index.html#tilelang.language.reduce.reduce_bitor">reduce_bitor() (in module tilelang.language.reduce)</a>
4202+
</li>
4203+
<li><a href="autoapi/tilelang/language/reduce/index.html#tilelang.language.reduce.reduce_bitxor">reduce_bitxor() (in module tilelang.language.reduce)</a>
41984204
</li>
41994205
<li><a href="autoapi/tilelang/intrinsics/mfma_macro_generator/index.html#tilelang.intrinsics.mfma_macro_generator.MatrixCoreIntrinEmitter.reduce_k">reduce_k (tilelang.intrinsics.mfma_macro_generator.MatrixCoreIntrinEmitter attribute)</a>
42004206

@@ -4246,12 +4252,12 @@ <h2>R</h2>
42464252
</li>
42474253
<li><a href="autoapi/tilelang/engine/callback/index.html#tilelang.engine.callback.register_cuda_postproc">register_cuda_postproc() (in module tilelang.engine.callback)</a>
42484254
</li>
4255+
</ul></td>
4256+
<td style="width: 33%; vertical-align: top;"><ul>
42494257
<li><a href="autoapi/tilelang/engine/callback/index.html#tilelang.engine.callback.register_cuda_postproc_callback">register_cuda_postproc_callback() (in module tilelang.engine.callback)</a>
42504258
</li>
42514259
<li><a href="autoapi/tilelang/engine/callback/index.html#tilelang.engine.callback.register_hip_postproc">register_hip_postproc() (in module tilelang.engine.callback)</a>
42524260
</li>
4253-
</ul></td>
4254-
<td style="width: 33%; vertical-align: top;"><ul>
42554261
<li><a href="autoapi/tilelang/engine/callback/index.html#tilelang.engine.callback.register_hip_postproc_callback">register_hip_postproc_callback() (in module tilelang.engine.callback)</a>
42564262
</li>
42574263
<li><a href="autoapi/tilelang/language/tir/ir/index.html#tilelang.language.tir.ir.reinterpret">reinterpret (in module tilelang.language.tir.ir)</a>

objects.inv

14 Bytes
Binary file not shown.

searchindex.js

Lines changed: 1 addition & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)