Skip to content

Commit fe9ba15

Browse files
Update docs
1 parent 3334944 commit fe9ba15

File tree

7 files changed

+31
-5
lines changed

7 files changed

+31
-5
lines changed

_sources/autoapi/tilelang/analysis/nested_loop_checker/index.rst.txt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@ Functions
1010
.. autoapisummary::
1111

1212
tilelang.analysis.nested_loop_checker.is_pipelined_for
13+
tilelang.analysis.nested_loop_checker.is_tile_op
1314
tilelang.analysis.nested_loop_checker.NestedLoopChecker
1415

1516

@@ -21,6 +22,11 @@ Module Contents
2122
Check if a for loop is pipelined.
2223

2324

25+
.. py:function:: is_tile_op(op)
26+
27+
Check if a call is a tile-op
28+
29+
2430
.. py:function:: NestedLoopChecker()
2531
2632
User-friendly pass which identifies any invalid any nested-loop pattern.

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -229,7 +229,7 @@ Module Contents
229229

230230
.. py:function:: finalize_reducer(reducer)
231231
232-
Finalize a reducer buffer by emitting the `tl.finalize_reducer` intrinsic.
232+
Finalize a reducer buffer by emitting the `tl.tileop.finalize_reducer` intrinsic.
233233

234234
This returns a TVM `tir.Call` handle that finalizes the given reducer using its writable pointer.
235235
The call does not modify Python objects directly; it produces the low-level intrinsic call used by the IR.

autoapi/tilelang/analysis/nested_loop_checker/index.html

Lines changed: 19 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -510,7 +510,10 @@ <h2>Functions<a class="headerlink" href="#functions" title="Link to this heading
510510
<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.analysis.nested_loop_checker.is_pipelined_for" title="tilelang.analysis.nested_loop_checker.is_pipelined_for"><code class="xref py py-obj docutils literal notranslate"><span class="pre">is_pipelined_for</span></code></a>(op)</p></td>
511511
<td><p>Check if a for loop is pipelined.</p></td>
512512
</tr>
513-
<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.analysis.nested_loop_checker.NestedLoopChecker" title="tilelang.analysis.nested_loop_checker.NestedLoopChecker"><code class="xref py py-obj docutils literal notranslate"><span class="pre">NestedLoopChecker</span></code></a>()</p></td>
513+
<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.analysis.nested_loop_checker.is_tile_op" title="tilelang.analysis.nested_loop_checker.is_tile_op"><code class="xref py py-obj docutils literal notranslate"><span class="pre">is_tile_op</span></code></a>(op)</p></td>
514+
<td><p>Check if a call is a tile-op</p></td>
515+
</tr>
516+
<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.analysis.nested_loop_checker.NestedLoopChecker" title="tilelang.analysis.nested_loop_checker.NestedLoopChecker"><code class="xref py py-obj docutils literal notranslate"><span class="pre">NestedLoopChecker</span></code></a>()</p></td>
514517
<td><p>User-friendly pass which identifies any invalid any nested-loop pattern.</p></td>
515518
</tr>
516519
</tbody>
@@ -533,6 +536,20 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
533536
</dl>
534537
</dd></dl>
535538

539+
<dl class="py function">
540+
<dt class="sig sig-object py" id="tilelang.analysis.nested_loop_checker.is_tile_op">
541+
<span class="sig-prename descclassname"><span class="pre">tilelang.analysis.nested_loop_checker.</span></span><span class="sig-name descname"><span class="pre">is_tile_op</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">op</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.analysis.nested_loop_checker.is_tile_op" title="Link to this definition"></a></dt>
542+
<dd><p>Check if a call is a tile-op</p>
543+
<dl class="field-list simple">
544+
<dt class="field-odd">Parameters<span class="colon">:</span></dt>
545+
<dd class="field-odd"><p><strong>op</strong> (<em>tvm.tir.Call</em>)</p>
546+
</dd>
547+
<dt class="field-even">Return type<span class="colon">:</span></dt>
548+
<dd class="field-even"><p><a class="reference internal" href="../../language/v2/dtypes/index.html#tilelang.language.v2.dtypes.bool" title="tilelang.language.v2.dtypes.bool">bool</a></p>
549+
</dd>
550+
</dl>
551+
</dd></dl>
552+
536553
<dl class="py function">
537554
<dt class="sig sig-object py" id="tilelang.analysis.nested_loop_checker.NestedLoopChecker">
538555
<span class="sig-prename descclassname"><span class="pre">tilelang.analysis.nested_loop_checker.</span></span><span class="sig-name descname"><span class="pre">NestedLoopChecker</span></span><span class="sig-paren">(</span><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.analysis.nested_loop_checker.NestedLoopChecker" title="Link to this definition"></a></dt>
@@ -668,6 +685,7 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
668685
<li><a class="reference internal" href="#functions">Functions</a></li>
669686
<li><a class="reference internal" href="#module-contents">Module Contents</a><ul>
670687
<li><a class="reference internal" href="#tilelang.analysis.nested_loop_checker.is_pipelined_for"><code class="docutils literal notranslate"><span class="pre">is_pipelined_for()</span></code></a></li>
688+
<li><a class="reference internal" href="#tilelang.analysis.nested_loop_checker.is_tile_op"><code class="docutils literal notranslate"><span class="pre">is_tile_op()</span></code></a></li>
671689
<li><a class="reference internal" href="#tilelang.analysis.nested_loop_checker.NestedLoopChecker"><code class="docutils literal notranslate"><span class="pre">NestedLoopChecker()</span></code></a></li>
672690
</ul>
673691
</li>

autoapi/tilelang/language/reduce/index.html

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -542,7 +542,7 @@ <h2>Functions<a class="headerlink" href="#functions" title="Link to this heading
542542
<td><p>Compute the cumulative sum of <cite>src</cite> along <cite>dim</cite>, writing results to <cite>dst</cite>.</p></td>
543543
</tr>
544544
<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>
545-
<td><p>Finalize a reducer buffer by emitting the <cite>tl.finalize_reducer</cite> intrinsic.</p></td>
545+
<td><p>Finalize a reducer buffer by emitting the <cite>tl.tileop.finalize_reducer</cite> intrinsic.</p></td>
546546
</tr>
547547
<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.language.reduce.warp_reduce_sum" title="tilelang.language.reduce.warp_reduce_sum"><code class="xref py py-obj docutils literal notranslate"><span class="pre">warp_reduce_sum</span></code></a>(value)</p></td>
548548
<td><p>Perform warp reduction sum on a register value.</p></td>
@@ -855,7 +855,7 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
855855
<dl class="py function">
856856
<dt class="sig sig-object py" id="tilelang.language.reduce.finalize_reducer">
857857
<span class="sig-prename descclassname"><span class="pre">tilelang.language.reduce.</span></span><span class="sig-name descname"><span class="pre">finalize_reducer</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">reducer</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.language.reduce.finalize_reducer" title="Link to this definition"></a></dt>
858-
<dd><p>Finalize a reducer buffer by emitting the <cite>tl.finalize_reducer</cite> intrinsic.</p>
858+
<dd><p>Finalize a reducer buffer by emitting the <cite>tl.tileop.finalize_reducer</cite> intrinsic.</p>
859859
<p>This returns a TVM <cite>tir.Call</cite> handle that finalizes the given reducer using its writable pointer.
860860
The call does not modify Python objects directly; it produces the low-level intrinsic call used by the IR.</p>
861861
<dl class="field-list simple">

genindex.html

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3258,6 +3258,8 @@ <h2>I</h2>
32583258
<li><a href="autoapi/tilelang/utils/language/index.html#tilelang.utils.language.is_tensor_memory">is_tensor_memory() (in module tilelang.utils.language)</a>
32593259
</li>
32603260
<li><a href="autoapi/tilelang/carver/arch/cuda/index.html#tilelang.carver.arch.cuda.is_tensorcore_supported_precision">is_tensorcore_supported_precision() (in module tilelang.carver.arch.cuda)</a>
3261+
</li>
3262+
<li><a href="autoapi/tilelang/analysis/nested_loop_checker/index.html#tilelang.analysis.nested_loop_checker.is_tile_op">is_tile_op() (in module tilelang.analysis.nested_loop_checker)</a>
32613263
</li>
32623264
<li><a href="autoapi/tilelang/jit/adapter/wrapper/index.html#tilelang.jit.adapter.wrapper.TLCUDASourceWrapper.is_tma_descriptor_arg">is_tma_descriptor_arg() (tilelang.jit.adapter.wrapper.TLCUDASourceWrapper method)</a>
32633265
</li>

objects.inv

8 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)