Skip to content

Commit e96024a

Browse files
github-actions[bot]github-actions[bot]
committed
Update docs
1 parent 10509af commit e96024a

File tree

7 files changed

+129
-9
lines changed

7 files changed

+129
-9
lines changed

‎_sources/autoapi/tilelang/language/loop/index.rst.txt‎

Lines changed: 35 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,15 @@ tilelang.language.loop
99

1010

1111

12+
Attributes
13+
----------
14+
15+
.. autoapisummary::
16+
17+
tilelang.language.loop.Serial
18+
tilelang.language.loop.Unroll
19+
20+
1221
Functions
1322
---------
1423

@@ -18,6 +27,7 @@ Functions
1827
tilelang.language.loop.Persistent
1928
tilelang.language.loop.Pipelined
2029
tilelang.language.loop.serial
30+
tilelang.language.loop.unroll
2131

2232

2333
Module Contents
@@ -69,3 +79,28 @@ Module Contents
6979

7080
.. py:function:: serial(start, stop = None, step = None, *, annotations = None)
7181
82+
.. py:function:: unroll(start, stop = None, step = None, *, explicit = False, unroll_factor = None, annotations = None)
83+
84+
The unrolled For statement.
85+
86+
:param start: The minimum value of iteration.
87+
:type start: PrimExpr
88+
:param stop: The maximum value of iteration.
89+
:type stop: PrimExpr
90+
:param step: The step size of the iteration.
91+
:type step: PrimExpr
92+
:param explicit: Whether to explicitly unroll the loop.
93+
:type explicit: bool
94+
:param unroll_factor: The unroll factor of the loop.
95+
:type unroll_factor: int
96+
:param annotations: The optional annotations of the For statement.
97+
:type annotations: Dict[str, Any]
98+
99+
:returns: **res** -- The ForFrame.
100+
:rtype: frame.ForFrame
101+
102+
103+
.. py:data:: Serial
104+
105+
.. py:data:: Unroll
106+

‎_sources/autoapi/tilelang/language/v2/builder/index.rst.txt‎

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,7 @@ Classes
3131
tilelang.language.v2.builder.ContinueFrame
3232
tilelang.language.v2.builder.BreakFrame
3333
tilelang.language.v2.builder.SerialForWithStep
34+
tilelang.language.v2.builder.UnrollForWithStep
3435
tilelang.language.v2.builder.Builder
3536
tilelang.language.v2.builder.PrimFunc
3637
tilelang.language.v2.builder.Macro
@@ -161,6 +162,11 @@ Module Contents
161162

162163

163164

165+
.. py:class:: UnrollForWithStep
166+
167+
Bases: :py:obj:`SerialForWithStep`
168+
169+
164170
.. py:data:: ContinueOrBreak
165171
166172
.. py:data:: AnyFrame

‎autoapi/tilelang/language/loop/index.html‎

Lines changed: 59 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -511,6 +511,21 @@
511511
<section id="module-tilelang.language.loop">
512512
<span id="tilelang-language-loop"></span><h1>tilelang.language.loop<a class="headerlink" href="#module-tilelang.language.loop" title="Link to this heading"></a></h1>
513513
<p>The language interface for tl programs.</p>
514+
<section id="attributes">
515+
<h2>Attributes<a class="headerlink" href="#attributes" title="Link to this heading"></a></h2>
516+
<div class="table-wrapper autosummary longtable docutils container">
517+
<table class="autosummary longtable docutils align-default">
518+
<tbody>
519+
<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.language.loop.Serial" title="tilelang.language.loop.Serial"><code class="xref py py-obj docutils literal notranslate"><span class="pre">Serial</span></code></a></p></td>
520+
<td><p></p></td>
521+
</tr>
522+
<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.language.loop.Unroll" title="tilelang.language.loop.Unroll"><code class="xref py py-obj docutils literal notranslate"><span class="pre">Unroll</span></code></a></p></td>
523+
<td><p></p></td>
524+
</tr>
525+
</tbody>
526+
</table>
527+
</div>
528+
</section>
514529
<section id="functions">
515530
<h2>Functions<a class="headerlink" href="#functions" title="Link to this heading"></a></h2>
516531
<div class="table-wrapper autosummary longtable docutils container">
@@ -528,6 +543,9 @@ <h2>Functions<a class="headerlink" href="#functions" title="Link to this heading
528543
<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.language.loop.serial" title="tilelang.language.loop.serial"><code class="xref py py-obj docutils literal notranslate"><span class="pre">serial</span></code></a>(start[, stop, step, annotations])</p></td>
529544
<td><p></p></td>
530545
</tr>
546+
<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.language.loop.unroll" title="tilelang.language.loop.unroll"><code class="xref py py-obj docutils literal notranslate"><span class="pre">unroll</span></code></a>(start[, stop, step, explicit, unroll_factor, ...])</p></td>
547+
<td><p>The unrolled For statement.</p></td>
548+
</tr>
531549
</tbody>
532550
</table>
533551
</div>
@@ -611,9 +629,46 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
611629
<li><p><strong>annotations</strong> (<em>dict</em><em>[</em><em>str</em><em>, </em><em>Any</em><em>] </em><em>| </em><em>None</em>)</p></li>
612630
</ul>
613631
</dd>
632+
<dt class="field-even">Return type<span class="colon">:</span></dt>
633+
<dd class="field-even"><p>tvm.script.ir_builder.tir.frame.ForFrame</p>
634+
</dd>
635+
</dl>
636+
</dd></dl>
637+
638+
<dl class="py function">
639+
<dt class="sig sig-object py" id="tilelang.language.loop.unroll">
640+
<span class="sig-prename descclassname"><span class="pre">tilelang.language.loop.</span></span><span class="sig-name descname"><span class="pre">unroll</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">start</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">stop</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">step</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="o"><span class="pre">*</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">explicit</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">False</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">unroll_factor</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">annotations</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.language.loop.unroll" title="Link to this definition"></a></dt>
641+
<dd><p>The unrolled For statement.</p>
642+
<dl class="field-list simple">
643+
<dt class="field-odd">Parameters<span class="colon">:</span></dt>
644+
<dd class="field-odd"><ul class="simple">
645+
<li><p><strong>start</strong> (<em>PrimExpr</em>) – The minimum value of iteration.</p></li>
646+
<li><p><strong>stop</strong> (<em>PrimExpr</em>) – The maximum value of iteration.</p></li>
647+
<li><p><strong>step</strong> (<em>PrimExpr</em>) – The step size of the iteration.</p></li>
648+
<li><p><strong>explicit</strong> (<a class="reference internal" href="../v2/dtypes/index.html#tilelang.language.v2.dtypes.bool" title="tilelang.language.v2.dtypes.bool"><em>bool</em></a>) – Whether to explicitly unroll the loop.</p></li>
649+
<li><p><strong>unroll_factor</strong> (<em>int</em>) – The unroll factor of the loop.</p></li>
650+
<li><p><strong>annotations</strong> (<em>Dict</em><em>[</em><em>str</em><em>, </em><em>Any</em><em>]</em>) – The optional annotations of the For statement.</p></li>
651+
</ul>
652+
</dd>
653+
<dt class="field-even">Returns<span class="colon">:</span></dt>
654+
<dd class="field-even"><p><strong>res</strong> – The ForFrame.</p>
655+
</dd>
656+
<dt class="field-odd">Return type<span class="colon">:</span></dt>
657+
<dd class="field-odd"><p>frame.ForFrame</p>
658+
</dd>
614659
</dl>
615660
</dd></dl>
616661

662+
<dl class="py data">
663+
<dt class="sig sig-object py" id="tilelang.language.loop.Serial">
664+
<span class="sig-prename descclassname"><span class="pre">tilelang.language.loop.</span></span><span class="sig-name descname"><span class="pre">Serial</span></span><a class="headerlink" href="#tilelang.language.loop.Serial" title="Link to this definition"></a></dt>
665+
<dd></dd></dl>
666+
667+
<dl class="py data">
668+
<dt class="sig sig-object py" id="tilelang.language.loop.Unroll">
669+
<span class="sig-prename descclassname"><span class="pre">tilelang.language.loop.</span></span><span class="sig-name descname"><span class="pre">Unroll</span></span><a class="headerlink" href="#tilelang.language.loop.Unroll" title="Link to this definition"></a></dt>
670+
<dd></dd></dl>
671+
617672
</section>
618673
</section>
619674

@@ -673,12 +728,16 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
673728
<div class="toc-tree">
674729
<ul>
675730
<li><a class="reference internal" href="#">tilelang.language.loop</a><ul>
731+
<li><a class="reference internal" href="#attributes">Attributes</a></li>
676732
<li><a class="reference internal" href="#functions">Functions</a></li>
677733
<li><a class="reference internal" href="#module-contents">Module Contents</a><ul>
678734
<li><a class="reference internal" href="#tilelang.language.loop.Parallel"><code class="docutils literal notranslate"><span class="pre">Parallel()</span></code></a></li>
679735
<li><a class="reference internal" href="#tilelang.language.loop.Persistent"><code class="docutils literal notranslate"><span class="pre">Persistent()</span></code></a></li>
680736
<li><a class="reference internal" href="#tilelang.language.loop.Pipelined"><code class="docutils literal notranslate"><span class="pre">Pipelined()</span></code></a></li>
681737
<li><a class="reference internal" href="#tilelang.language.loop.serial"><code class="docutils literal notranslate"><span class="pre">serial()</span></code></a></li>
738+
<li><a class="reference internal" href="#tilelang.language.loop.unroll"><code class="docutils literal notranslate"><span class="pre">unroll()</span></code></a></li>
739+
<li><a class="reference internal" href="#tilelang.language.loop.Serial"><code class="docutils literal notranslate"><span class="pre">Serial</span></code></a></li>
740+
<li><a class="reference internal" href="#tilelang.language.loop.Unroll"><code class="docutils literal notranslate"><span class="pre">Unroll</span></code></a></li>
682741
</ul>
683742
</li>
684743
</ul>

‎autoapi/tilelang/language/v2/builder/index.html‎

Lines changed: 13 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -569,13 +569,16 @@ <h2>Classes<a class="headerlink" href="#classes" title="Link to this heading">¶
569569
<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.language.v2.builder.SerialForWithStep" title="tilelang.language.v2.builder.SerialForWithStep"><code class="xref py py-obj docutils literal notranslate"><span class="pre">SerialForWithStep</span></code></a></p></td>
570570
<td><p></p></td>
571571
</tr>
572-
<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.language.v2.builder.Builder" title="tilelang.language.v2.builder.Builder"><code class="xref py py-obj docutils literal notranslate"><span class="pre">Builder</span></code></a></p></td>
572+
<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.language.v2.builder.UnrollForWithStep" title="tilelang.language.v2.builder.UnrollForWithStep"><code class="xref py py-obj docutils literal notranslate"><span class="pre">UnrollForWithStep</span></code></a></p></td>
573573
<td><p></p></td>
574574
</tr>
575-
<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.language.v2.builder.PrimFunc" title="tilelang.language.v2.builder.PrimFunc"><code class="xref py py-obj docutils literal notranslate"><span class="pre">PrimFunc</span></code></a></p></td>
575+
<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.language.v2.builder.Builder" title="tilelang.language.v2.builder.Builder"><code class="xref py py-obj docutils literal notranslate"><span class="pre">Builder</span></code></a></p></td>
576+
<td><p></p></td>
577+
</tr>
578+
<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.language.v2.builder.PrimFunc" title="tilelang.language.v2.builder.PrimFunc"><code class="xref py py-obj docutils literal notranslate"><span class="pre">PrimFunc</span></code></a></p></td>
576579
<td><p>Abstract base class for generic types.</p></td>
577580
</tr>
578-
<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.language.v2.builder.Macro" title="tilelang.language.v2.builder.Macro"><code class="xref py py-obj docutils literal notranslate"><span class="pre">Macro</span></code></a></p></td>
581+
<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.language.v2.builder.Macro" title="tilelang.language.v2.builder.Macro"><code class="xref py py-obj docutils literal notranslate"><span class="pre">Macro</span></code></a></p></td>
579582
<td><p>Abstract base class for generic types.</p></td>
580583
</tr>
581584
</tbody>
@@ -736,6 +739,12 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
736739

737740
</dd></dl>
738741

742+
<dl class="py class">
743+
<dt class="sig sig-object py" id="tilelang.language.v2.builder.UnrollForWithStep">
744+
<em class="property"><span class="pre">class</span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">tilelang.language.v2.builder.</span></span><span class="sig-name descname"><span class="pre">UnrollForWithStep</span></span><a class="headerlink" href="#tilelang.language.v2.builder.UnrollForWithStep" title="Link to this definition"></a></dt>
745+
<dd><p>Bases: <a class="reference internal" href="#tilelang.language.v2.builder.SerialForWithStep" title="tilelang.language.v2.builder.SerialForWithStep"><code class="xref py py-obj docutils literal notranslate"><span class="pre">SerialForWithStep</span></code></a></p>
746+
</dd></dl>
747+
739748
<dl class="py data">
740749
<dt class="sig sig-object py" id="tilelang.language.v2.builder.ContinueOrBreak">
741750
<span class="sig-prename descclassname"><span class="pre">tilelang.language.v2.builder.</span></span><span class="sig-name descname"><span class="pre">ContinueOrBreak</span></span><a class="headerlink" href="#tilelang.language.v2.builder.ContinueOrBreak" title="Link to this definition"></a></dt>
@@ -1347,6 +1356,7 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
13471356
<li><a class="reference internal" href="#tilelang.language.v2.builder.SerialForWithStep.annotations"><code class="docutils literal notranslate"><span class="pre">SerialForWithStep.annotations</span></code></a></li>
13481357
</ul>
13491358
</li>
1359+
<li><a class="reference internal" href="#tilelang.language.v2.builder.UnrollForWithStep"><code class="docutils literal notranslate"><span class="pre">UnrollForWithStep</span></code></a></li>
13501360
<li><a class="reference internal" href="#tilelang.language.v2.builder.ContinueOrBreak"><code class="docutils literal notranslate"><span class="pre">ContinueOrBreak</span></code></a></li>
13511361
<li><a class="reference internal" href="#tilelang.language.v2.builder.AnyFrame"><code class="docutils literal notranslate"><span class="pre">AnyFrame</span></code></a></li>
13521362
<li><a class="reference internal" href="#tilelang.language.v2.builder.TIR_CONTROL_FRAME"><code class="docutils literal notranslate"><span class="pre">TIR_CONTROL_FRAME</span></code></a></li>

‎genindex.html‎

Lines changed: 15 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -5415,6 +5415,8 @@ <h2>S</h2>
54155415
<li><a href="autoapi/tilelang/carver/template/flashattention/index.html#tilelang.carver.template.flashattention.FlashAttentionTemplate.seq_kv_length">seq_kv_length (tilelang.carver.template.flashattention.FlashAttentionTemplate attribute)</a>
54165416
</li>
54175417
<li><a href="autoapi/tilelang/carver/template/flashattention/index.html#tilelang.carver.template.flashattention.FlashAttentionTemplate.seq_length">seq_length (tilelang.carver.template.flashattention.FlashAttentionTemplate attribute)</a>
5418+
</li>
5419+
<li><a href="autoapi/tilelang/language/loop/index.html#tilelang.language.loop.Serial">Serial (in module tilelang.language.loop)</a>
54185420
</li>
54195421
<li><a href="autoapi/tilelang/language/loop/index.html#tilelang.language.loop.serial">serial() (in module tilelang.language.loop)</a>
54205422

@@ -5605,11 +5607,11 @@ <h2>S</h2>
56055607
<li><a href="autoapi/tilelang/language/builtin/index.html#tilelang.language.builtin.shuffle_elect">shuffle_elect() (in module tilelang.language.builtin)</a>
56065608
</li>
56075609
<li><a href="autoapi/tilelang/language/tir/ir/index.html#tilelang.language.tir.ir.sigmoid">sigmoid (in module tilelang.language.tir.ir)</a>
5608-
</li>
5609-
<li><a href="autoapi/tilelang/language/tir/op/index.html#tilelang.language.tir.op.sigmoid">sigmoid() (in module tilelang.language.tir.op)</a>
56105610
</li>
56115611
</ul></td>
56125612
<td style="width: 33%; vertical-align: top;"><ul>
5613+
<li><a href="autoapi/tilelang/language/tir/op/index.html#tilelang.language.tir.op.sigmoid">sigmoid() (in module tilelang.language.tir.op)</a>
5614+
</li>
56135615
<li><a href="autoapi/tilelang/jit/index.html#tilelang.jit.JITImpl.signature">signature (tilelang.jit.JITImpl attribute)</a>
56145616
</li>
56155617
<li><a href="autoapi/tilelang/transform/simplify/index.html#tilelang.transform.simplify.Simplify">Simplify() (in module tilelang.transform.simplify)</a>
@@ -7627,7 +7629,15 @@ <h2>U</h2>
76277629
</li>
76287630
<li><a href="autoapi/tilelang/utils/tensor/index.html#tilelang.utils.tensor.TensorSupplyType.Uniform">Uniform (tilelang.utils.tensor.TensorSupplyType attribute)</a>
76297631
</li>
7630-
<li><a href="autoapi/tilelang/language/tir/ir/index.html#tilelang.language.tir.ir.unroll">unroll() (in module tilelang.language.tir.ir)</a>
7632+
<li><a href="autoapi/tilelang/language/loop/index.html#tilelang.language.loop.Unroll">Unroll (in module tilelang.language.loop)</a>
7633+
</li>
7634+
<li><a href="autoapi/tilelang/language/loop/index.html#tilelang.language.loop.unroll">unroll() (in module tilelang.language.loop)</a>
7635+
7636+
<ul>
7637+
<li><a href="autoapi/tilelang/language/tir/ir/index.html#tilelang.language.tir.ir.unroll">(in module tilelang.language.tir.ir)</a>
7638+
</li>
7639+
</ul></li>
7640+
<li><a href="autoapi/tilelang/language/v2/builder/index.html#tilelang.language.v2.builder.UnrollForWithStep">UnrollForWithStep (class in tilelang.language.v2.builder)</a>
76317641
</li>
76327642
<li><a href="autoapi/tilelang/language/v2/builder/index.html#tilelang.language.v2.builder.unwrap_cond">unwrap_cond() (in module tilelang.language.v2.builder)</a>
76337643
</li>
@@ -7641,6 +7651,8 @@ <h2>U</h2>
76417651
</ul></li>
76427652
<li><a href="autoapi/tilelang/jit/adapter/nvrtc/libgen/index.html#tilelang.jit.adapter.nvrtc.libgen.NVRTCLibraryGenerator.update_host_func">update_host_func() (tilelang.jit.adapter.nvrtc.libgen.NVRTCLibraryGenerator method)</a>
76437653
</li>
7654+
</ul></td>
7655+
<td style="width: 33%; vertical-align: top;"><ul>
76447656
<li><a href="autoapi/tilelang/jit/adapter/libgen/index.html#tilelang.jit.adapter.libgen.LibraryGenerator.update_lib_code">update_lib_code() (tilelang.jit.adapter.libgen.LibraryGenerator method)</a>
76457657

76467658
<ul>
@@ -7653,8 +7665,6 @@ <h2>U</h2>
76537665
<li><a href="autoapi/tilelang/jit/adapter/wrapper/index.html#tilelang.jit.adapter.wrapper.TLMetalSourceWrapper.update_lib_code">(tilelang.jit.adapter.wrapper.TLMetalSourceWrapper method)</a>
76547666
</li>
76557667
</ul></li>
7656-
</ul></td>
7657-
<td style="width: 33%; vertical-align: top;"><ul>
76587668
<li><a href="autoapi/tilelang/carver/roller/node/index.html#tilelang.carver.roller.node.Node.update_tags">update_tags() (tilelang.carver.roller.node.Node method)</a>
76597669
</li>
76607670
<li><a href="autoapi/tilelang/jit/kernel/index.html#tilelang.jit.kernel.JITKernel.update_tuner_result">update_tuner_result() (tilelang.jit.kernel.JITKernel method)</a>

‎objects.inv‎

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