Skip to content

Commit 64db56c

Browse files
github-actions[bot]github-actions[bot]
committed
Update docs
1 parent f8482b1 commit 64db56c

File tree

5 files changed

+39
-10
lines changed

5 files changed

+39
-10
lines changed

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

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,7 @@ Functions
6262
tilelang.transform.PersistThreadblock
6363
tilelang.transform.AlignDynamicSharedMemoryAllocations
6464
tilelang.transform.LowerSharedBarrier
65+
tilelang.transform.PlanAndUpdateBufferAllocationLocation
6566
tilelang.transform.StorageRewrite
6667
tilelang.transform.LowerOpaqueBlock
6768
tilelang.transform.LowerThreadAllreduce
@@ -373,6 +374,14 @@ Package Contents
373374

374375

375376

377+
.. py:function:: PlanAndUpdateBufferAllocationLocation()
378+
379+
Plan and update buffer allocation locations within PrimFuncs.
380+
381+
:returns: **fpass** -- The result pass
382+
:rtype: tvm.transform.Pass
383+
384+
376385
.. py:function:: StorageRewrite()
377386
378387
StorageRewrite

‎autoapi/tilelang/transform/index.html‎

Lines changed: 25 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -637,25 +637,28 @@ <h2>Functions<a class="headerlink" href="#functions" title="Link to this heading
637637
<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.transform.LowerSharedBarrier" title="tilelang.transform.LowerSharedBarrier"><code class="xref py py-obj docutils literal notranslate"><span class="pre">LowerSharedBarrier</span></code></a>()</p></td>
638638
<td><p>LowerSharedBarrier</p></td>
639639
</tr>
640-
<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.transform.StorageRewrite" title="tilelang.transform.StorageRewrite"><code class="xref py py-obj docutils literal notranslate"><span class="pre">StorageRewrite</span></code></a>()</p></td>
640+
<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.transform.PlanAndUpdateBufferAllocationLocation" title="tilelang.transform.PlanAndUpdateBufferAllocationLocation"><code class="xref py py-obj docutils literal notranslate"><span class="pre">PlanAndUpdateBufferAllocationLocation</span></code></a>()</p></td>
641+
<td><p>Plan and update buffer allocation locations within PrimFuncs.</p></td>
642+
</tr>
643+
<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.transform.StorageRewrite" title="tilelang.transform.StorageRewrite"><code class="xref py py-obj docutils literal notranslate"><span class="pre">StorageRewrite</span></code></a>()</p></td>
641644
<td><p>StorageRewrite</p></td>
642645
</tr>
643-
<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.transform.LowerOpaqueBlock" title="tilelang.transform.LowerOpaqueBlock"><code class="xref py py-obj docutils literal notranslate"><span class="pre">LowerOpaqueBlock</span></code></a>()</p></td>
646+
<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.transform.LowerOpaqueBlock" title="tilelang.transform.LowerOpaqueBlock"><code class="xref py py-obj docutils literal notranslate"><span class="pre">LowerOpaqueBlock</span></code></a>()</p></td>
644647
<td><p>LowerOpaqueBlock</p></td>
645648
</tr>
646-
<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.transform.LowerThreadAllreduce" title="tilelang.transform.LowerThreadAllreduce"><code class="xref py py-obj docutils literal notranslate"><span class="pre">LowerThreadAllreduce</span></code></a>()</p></td>
649+
<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.transform.LowerThreadAllreduce" title="tilelang.transform.LowerThreadAllreduce"><code class="xref py py-obj docutils literal notranslate"><span class="pre">LowerThreadAllreduce</span></code></a>()</p></td>
647650
<td><p>LowerThreadAllreduce</p></td>
648651
</tr>
649-
<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.transform.LowerIntrin" title="tilelang.transform.LowerIntrin"><code class="xref py py-obj docutils literal notranslate"><span class="pre">LowerIntrin</span></code></a>()</p></td>
652+
<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.transform.LowerIntrin" title="tilelang.transform.LowerIntrin"><code class="xref py py-obj docutils literal notranslate"><span class="pre">LowerIntrin</span></code></a>()</p></td>
650653
<td><p>LowerIntrin</p></td>
651654
</tr>
652-
<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.transform.LowerDeviceKernelLaunch" title="tilelang.transform.LowerDeviceKernelLaunch"><code class="xref py py-obj docutils literal notranslate"><span class="pre">LowerDeviceKernelLaunch</span></code></a>()</p></td>
655+
<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.transform.LowerDeviceKernelLaunch" title="tilelang.transform.LowerDeviceKernelLaunch"><code class="xref py py-obj docutils literal notranslate"><span class="pre">LowerDeviceKernelLaunch</span></code></a>()</p></td>
653656
<td><p>Create and return a transform pass that lowers device kernel launch constructs to target-specific IR.</p></td>
654657
</tr>
655-
<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.transform.LowerSharedTmem" title="tilelang.transform.LowerSharedTmem"><code class="xref py py-obj docutils literal notranslate"><span class="pre">LowerSharedTmem</span></code></a>()</p></td>
658+
<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.transform.LowerSharedTmem" title="tilelang.transform.LowerSharedTmem"><code class="xref py py-obj docutils literal notranslate"><span class="pre">LowerSharedTmem</span></code></a>()</p></td>
656659
<td><p>LowerSharedTmem</p></td>
657660
</tr>
658-
<tr class="row-even"><td><p><a class="reference internal" href="#tilelang.transform.LayoutReducer" title="tilelang.transform.LayoutReducer"><code class="xref py py-obj docutils literal notranslate"><span class="pre">LayoutReducer</span></code></a>()</p></td>
661+
<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.transform.LayoutReducer" title="tilelang.transform.LayoutReducer"><code class="xref py py-obj docutils literal notranslate"><span class="pre">LayoutReducer</span></code></a>()</p></td>
659662
<td><p>Return a TVM transform pass that performs layout reduction/normalization.</p></td>
660663
</tr>
661664
</tbody>
@@ -1161,6 +1164,20 @@ <h3>Returns:<a class="headerlink" href="#returns" title="Link to this heading">
11611164
<dd><p>LowerSharedBarrier</p>
11621165
</dd></dl>
11631166

1167+
<dl class="py function">
1168+
<dt class="sig sig-object py" id="tilelang.transform.PlanAndUpdateBufferAllocationLocation">
1169+
<span class="sig-prename descclassname"><span class="pre">tilelang.transform.</span></span><span class="sig-name descname"><span class="pre">PlanAndUpdateBufferAllocationLocation</span></span><span class="sig-paren">(</span><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.transform.PlanAndUpdateBufferAllocationLocation" title="Link to this definition"></a></dt>
1170+
<dd><p>Plan and update buffer allocation locations within PrimFuncs.</p>
1171+
<dl class="field-list simple">
1172+
<dt class="field-odd">Returns<span class="colon">:</span></dt>
1173+
<dd class="field-odd"><p><strong>fpass</strong> – The result pass</p>
1174+
</dd>
1175+
<dt class="field-even">Return type<span class="colon">:</span></dt>
1176+
<dd class="field-even"><p>tvm.transform.Pass</p>
1177+
</dd>
1178+
</dl>
1179+
</dd></dl>
1180+
11641181
<dl class="py function">
11651182
<dt class="sig sig-object py" id="tilelang.transform.StorageRewrite">
11661183
<span class="sig-prename descclassname"><span class="pre">tilelang.transform.</span></span><span class="sig-name descname"><span class="pre">StorageRewrite</span></span><span class="sig-paren">(</span><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.transform.StorageRewrite" title="Link to this definition"></a></dt>
@@ -1326,6 +1343,7 @@ <h3>Returns:<a class="headerlink" href="#returns" title="Link to this heading">
13261343
<li><a class="reference internal" href="#tilelang.transform.PersistThreadblock"><code class="docutils literal notranslate"><span class="pre">PersistThreadblock()</span></code></a></li>
13271344
<li><a class="reference internal" href="#tilelang.transform.AlignDynamicSharedMemoryAllocations"><code class="docutils literal notranslate"><span class="pre">AlignDynamicSharedMemoryAllocations()</span></code></a></li>
13281345
<li><a class="reference internal" href="#tilelang.transform.LowerSharedBarrier"><code class="docutils literal notranslate"><span class="pre">LowerSharedBarrier()</span></code></a></li>
1346+
<li><a class="reference internal" href="#tilelang.transform.PlanAndUpdateBufferAllocationLocation"><code class="docutils literal notranslate"><span class="pre">PlanAndUpdateBufferAllocationLocation()</span></code></a></li>
13291347
<li><a class="reference internal" href="#tilelang.transform.StorageRewrite"><code class="docutils literal notranslate"><span class="pre">StorageRewrite()</span></code></a></li>
13301348
<li><a class="reference internal" href="#tilelang.transform.LowerOpaqueBlock"><code class="docutils literal notranslate"><span class="pre">LowerOpaqueBlock()</span></code></a></li>
13311349
<li><a class="reference internal" href="#tilelang.transform.LowerThreadAllreduce"><code class="docutils literal notranslate"><span class="pre">LowerThreadAllreduce()</span></code></a></li>

‎genindex.html‎

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4900,6 +4900,8 @@ <h2>P</h2>
49004900
<li><a href="autoapi/tilelang/carver/roller/policy/tensorcore/index.html#tilelang.carver.roller.policy.tensorcore.TensorCorePolicy.plan_rasterization">(tilelang.carver.roller.policy.tensorcore.TensorCorePolicy method)</a>
49014901
</li>
49024902
</ul></li>
4903+
<li><a href="autoapi/tilelang/transform/index.html#tilelang.transform.PlanAndUpdateBufferAllocationLocation">PlanAndUpdateBufferAllocationLocation() (in module tilelang.transform)</a>
4904+
</li>
49034905
<li><a href="autoapi/tilelang/carver/arch/arch_base/index.html#tilelang.carver.arch.arch_base.TileDevice.platform">platform (tilelang.carver.arch.arch_base.TileDevice attribute)</a>
49044906

49054907
<ul>
@@ -4947,11 +4949,11 @@ <h2>P</h2>
49474949
<li><a href="autoapi/tilelang/language/tir/op/index.html#tilelang.language.tir.op.pow_of_int">pow_of_int() (in module tilelang.language.tir.op)</a>
49484950
</li>
49494951
<li><a href="autoapi/tilelang/language/tir/op/index.html#tilelang.language.tir.op.power">power() (in module tilelang.language.tir.op)</a>
4950-
</li>
4951-
<li><a href="autoapi/tilelang/carver/roller/node/index.html#tilelang.carver.roller.node.pre_order_traverse">pre_order_traverse() (in module tilelang.carver.roller.node)</a>
49524952
</li>
49534953
</ul></td>
49544954
<td style="width: 33%; vertical-align: top;"><ul>
4955+
<li><a href="autoapi/tilelang/carver/roller/node/index.html#tilelang.carver.roller.node.pre_order_traverse">pre_order_traverse() (in module tilelang.carver.roller.node)</a>
4956+
</li>
49554957
<li><a href="autoapi/tilelang/jit/adapter/wrapper/index.html#tilelang.jit.adapter.wrapper.PREDEF_ATTRIBUTE_SET_DYNAMIC_MEMORY">PREDEF_ATTRIBUTE_SET_DYNAMIC_MEMORY (in module tilelang.jit.adapter.wrapper)</a>
49564958
</li>
49574959
<li><a href="autoapi/tilelang/jit/adapter/wrapper/index.html#tilelang.jit.adapter.wrapper.PREDEF_ATTRIBUTE_SET_DYNAMIC_MEMORY_HIP">PREDEF_ATTRIBUTE_SET_DYNAMIC_MEMORY_HIP (in module tilelang.jit.adapter.wrapper)</a>

‎objects.inv‎

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