Skip to content

Commit 1a76e9d

Browse files
Update docs
1 parent dfaac12 commit 1a76e9d

File tree

3 files changed

+23
-3
lines changed

3 files changed

+23
-3
lines changed

_sources/autoapi/tilelang/transform/add_bufstore_wrapper/index.rst.txt

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,3 +17,13 @@ Module Contents
1717

1818
.. py:function:: AddWrapperForSingleBufStore()
1919
20+
Creates a TVM pass that wraps single buffer stores with parallel loops.
21+
22+
This transformation adds T.Parallel wrappers around buffer stores that:
23+
1. Access fragment buffers with index 0
24+
2. Are not inside existing tile operations or thread bindings
25+
3. Don't access fragment buffers with non-zero indices
26+
27+
:returns: A prim_func_pass that applies the transformation
28+
29+

autoapi/tilelang/transform/add_bufstore_wrapper/index.html

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -476,7 +476,7 @@ <h2>Functions<a class="headerlink" href="#functions" title="Link to this heading
476476
<table class="autosummary longtable docutils align-default">
477477
<tbody>
478478
<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.transform.add_bufstore_wrapper.AddWrapperForSingleBufStore" title="tilelang.transform.add_bufstore_wrapper.AddWrapperForSingleBufStore"><code class="xref py py-obj docutils literal notranslate"><span class="pre">AddWrapperForSingleBufStore</span></code></a>()</p></td>
479-
<td><p></p></td>
479+
<td><p>Creates a TVM pass that wraps single buffer stores with parallel loops.</p></td>
480480
</tr>
481481
</tbody>
482482
</table>
@@ -487,7 +487,17 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
487487
<dl class="py function">
488488
<dt class="sig sig-object py" id="tilelang.transform.add_bufstore_wrapper.AddWrapperForSingleBufStore">
489489
<span class="sig-prename descclassname"><span class="pre">tilelang.transform.add_bufstore_wrapper.</span></span><span class="sig-name descname"><span class="pre">AddWrapperForSingleBufStore</span></span><span class="sig-paren">(</span><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.transform.add_bufstore_wrapper.AddWrapperForSingleBufStore" title="Link to this definition"></a></dt>
490-
<dd></dd></dl>
490+
<dd><p>Creates a TVM pass that wraps single buffer stores with parallel loops.</p>
491+
<p>This transformation adds T.Parallel wrappers around buffer stores that:
492+
1. Access fragment buffers with index 0
493+
2. Are not inside existing tile operations or thread bindings
494+
3. Don’t access fragment buffers with non-zero indices</p>
495+
<dl class="field-list simple">
496+
<dt class="field-odd">Returns<span class="colon">:</span></dt>
497+
<dd class="field-odd"><p>A prim_func_pass that applies the transformation</p>
498+
</dd>
499+
</dl>
500+
</dd></dl>
491501

492502
</section>
493503
</section>

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)