Update docs

Author: LeiWang1999

api/tilelang.engine.html

@@ -452,6 +452,7 @@ <h2>Submodules<a class="headerlink" href="#submodules" title="Permalink to this
 <li class="toctree-l1"><a class="reference internal" href="tilelang.engine.phase.html">tilelang.engine.phase module</a><ul>
 <li class="toctree-l2"><a class="reference internal" href="tilelang.engine.phase.html#tilelang.engine.phase.LowerAndLegalize"><code class="docutils literal notranslate"><span class="pre">LowerAndLegalize()</span></code></a></li>
 <li class="toctree-l2"><a class="reference internal" href="tilelang.engine.phase.html#tilelang.engine.phase.OptimizeForTarget"><code class="docutils literal notranslate"><span class="pre">OptimizeForTarget()</span></code></a></li>
+<li class="toctree-l2"><a class="reference internal" href="tilelang.engine.phase.html#tilelang.engine.phase.allow_tma_and_warp_specialized"><code class="docutils literal notranslate"><span class="pre">allow_tma_and_warp_specialized()</span></code></a></li>
 </ul>
 </li>
 </ul>

api/tilelang.engine.phase.html

@@ -411,6 +411,11 @@
 <span class="sig-prename descclassname"><span class="pre">tilelang.engine.phase.</span></span><span class="sig-name descname"><span class="pre">OptimizeForTarget</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">mod</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">IRModule</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">target</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">Target</span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><span class="pre">IRModule</span></span></span><a class="headerlink" href="#tilelang.engine.phase.OptimizeForTarget" title="Permalink to this definition">#</a></dt>
 <dd></dd></dl>
 
+<dl class="py function">
+<dt class="sig sig-object py" id="tilelang.engine.phase.allow_tma_and_warp_specialized">
+<span class="sig-prename descclassname"><span class="pre">tilelang.engine.phase.</span></span><span class="sig-name descname"><span class="pre">allow_tma_and_warp_specialized</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">target</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">Target</span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><span class="pre">bool</span></span></span><a class="headerlink" href="#tilelang.engine.phase.allow_tma_and_warp_specialized" title="Permalink to this definition">#</a></dt>
+<dd></dd></dl>
+
 </section>
 
         </article>
@@ -471,6 +476,7 @@
 <li><a class="reference internal" href="#">tilelang.engine.phase module</a><ul>
 <li><a class="reference internal" href="#tilelang.engine.phase.LowerAndLegalize"><code class="docutils literal notranslate"><span class="pre">LowerAndLegalize()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.engine.phase.OptimizeForTarget"><code class="docutils literal notranslate"><span class="pre">OptimizeForTarget()</span></code></a></li>
+<li><a class="reference internal" href="#tilelang.engine.phase.allow_tma_and_warp_specialized"><code class="docutils literal notranslate"><span class="pre">allow_tma_and_warp_specialized()</span></code></a></li>
 </ul>
 </li>
 </ul>

api/tilelang.html

@@ -679,6 +679,7 @@ <h2>Subpackages<a class="headerlink" href="#subpackages" title="Permalink to thi
 <li class="toctree-l3"><a class="reference internal" href="tilelang.engine.phase.html">tilelang.engine.phase module</a><ul>
 <li class="toctree-l4"><a class="reference internal" href="tilelang.engine.phase.html#tilelang.engine.phase.LowerAndLegalize"><code class="docutils literal notranslate"><span class="pre">LowerAndLegalize()</span></code></a></li>
 <li class="toctree-l4"><a class="reference internal" href="tilelang.engine.phase.html#tilelang.engine.phase.OptimizeForTarget"><code class="docutils literal notranslate"><span class="pre">OptimizeForTarget()</span></code></a></li>
+<li class="toctree-l4"><a class="reference internal" href="tilelang.engine.phase.html#tilelang.engine.phase.allow_tma_and_warp_specialized"><code class="docutils literal notranslate"><span class="pre">allow_tma_and_warp_specialized()</span></code></a></li>
 </ul>
 </li>
 </ul>
@@ -1039,6 +1040,7 @@ <h2>Subpackages<a class="headerlink" href="#subpackages" title="Permalink to thi
 <li class="toctree-l3"><a class="reference internal" href="tilelang.transform.html#tilelang.transform.VectorizeLoop"><code class="docutils literal notranslate"><span class="pre">VectorizeLoop()</span></code></a></li>
 <li class="toctree-l3"><a class="reference internal" href="tilelang.transform.html#tilelang.transform.WarpSpecialized"><code class="docutils literal notranslate"><span class="pre">WarpSpecialized()</span></code></a></li>
 <li class="toctree-l3"><a class="reference internal" href="tilelang.transform.html#tilelang.transform.WarpSpecializedPipeline"><code class="docutils literal notranslate"><span class="pre">WarpSpecializedPipeline()</span></code></a></li>
+<li class="toctree-l3"><a class="reference internal" href="tilelang.transform.html#tilelang.transform.get_pass_context"><code class="docutils literal notranslate"><span class="pre">get_pass_context()</span></code></a></li>
 </ul>
 </li>
 </ul>

api/tilelang.language.logical.html

@@ -404,7 +404,7 @@
 <p>The language interface for tl programs.</p>
 <dl class="py function">
 <dt class="sig sig-object py" id="tilelang.language.logical.all_of">
-<span class="sig-prename descclassname"><span class="pre">tilelang.language.logical.</span></span><span class="sig-name descname"><span class="pre">all_of</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="pre">buffer:</span> <span class="pre">~typing.Union[&lt;tilelang.language.proxy.TensorProxy</span> <span class="pre">object</span> <span class="pre">at</span> <span class="pre">0x7ff24cc94c40&gt;,</span> <span class="pre">~tvm.tir.stmt.BufferRegion]</span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.language.logical.all_of" title="Permalink to this definition">#</a></dt>
+<span class="sig-prename descclassname"><span class="pre">tilelang.language.logical.</span></span><span class="sig-name descname"><span class="pre">all_of</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="pre">buffer:</span> <span class="pre">~typing.Union[&lt;tilelang.language.proxy.TensorProxy</span> <span class="pre">object</span> <span class="pre">at</span> <span class="pre">0x7fcd3e69de40&gt;,</span> <span class="pre">~tvm.tir.stmt.BufferRegion]</span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.language.logical.all_of" title="Permalink to this definition">#</a></dt>
 <dd><p>Check if all elements in the buffer are true.</p>
 <dl class="field-list simple">
 <dt class="field-odd">Parameters<span class="colon">:</span></dt>
@@ -418,7 +418,7 @@
 
 <dl class="py function">
 <dt class="sig sig-object py" id="tilelang.language.logical.any_of">
-<span class="sig-prename descclassname"><span class="pre">tilelang.language.logical.</span></span><span class="sig-name descname"><span class="pre">any_of</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="pre">buffer:</span> <span class="pre">~typing.Union[&lt;tilelang.language.proxy.TensorProxy</span> <span class="pre">object</span> <span class="pre">at</span> <span class="pre">0x7ff24cc94c40&gt;,</span> <span class="pre">~tvm.tir.stmt.BufferRegion]</span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.language.logical.any_of" title="Permalink to this definition">#</a></dt>
+<span class="sig-prename descclassname"><span class="pre">tilelang.language.logical.</span></span><span class="sig-name descname"><span class="pre">any_of</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="pre">buffer:</span> <span class="pre">~typing.Union[&lt;tilelang.language.proxy.TensorProxy</span> <span class="pre">object</span> <span class="pre">at</span> <span class="pre">0x7fcd3e69de40&gt;,</span> <span class="pre">~tvm.tir.stmt.BufferRegion]</span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.language.logical.any_of" title="Permalink to this definition">#</a></dt>
 <dd><p>Check if any element in the buffer is true.</p>
 <dl class="field-list simple">
 <dt class="field-odd">Parameters<span class="colon">:</span></dt>

api/tilelang.language.proxy.html

@@ -452,7 +452,7 @@
 <p>Buffer proxy class for constructing tir buffer.</p>
 <dl class="py method">
 <dt class="sig sig-object py" id="tilelang.language.proxy.BufferProxy.from_ptr">
-<span class="sig-name descname"><span class="pre">from_ptr</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">pointer_var</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">Var</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">shape</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">tuple</span><span class="p"><span class="pre">[</span></span><span class="pre">tvm.ir.expr.PrimExpr</span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="p"><span class="pre">...</span></span><span class="p"><span class="pre">]</span></span></span></em>, <em class="sig-param"><span class="n"><span class="pre">dtype</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">str</span></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">'float32'</span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><span class="pre">&lt;tilelang.language.proxy.BufferProxy</span> <span class="pre">object</span> <span class="pre">at</span> <span class="pre">0x7ff24cc958d0&gt;</span></span></span><a class="headerlink" href="#tilelang.language.proxy.BufferProxy.from_ptr" title="Permalink to this definition">#</a></dt>
+<span class="sig-name descname"><span class="pre">from_ptr</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">pointer_var</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">Var</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">shape</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">tuple</span><span class="p"><span class="pre">[</span></span><span class="pre">tvm.ir.expr.PrimExpr</span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="p"><span class="pre">...</span></span><span class="p"><span class="pre">]</span></span></span></em>, <em class="sig-param"><span class="n"><span class="pre">dtype</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">str</span></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">'float32'</span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><span class="pre">&lt;tilelang.language.proxy.BufferProxy</span> <span class="pre">object</span> <span class="pre">at</span> <span class="pre">0x7fcd3e69de10&gt;</span></span></span><a class="headerlink" href="#tilelang.language.proxy.BufferProxy.from_ptr" title="Permalink to this definition">#</a></dt>
 <dd><p>Create a buffer from a pointer, shape, and data type.</p>
 <dl class="field-list simple">
 <dt class="field-odd">Parameters<span class="colon">:</span></dt>

api/tilelang.transform.html

@@ -778,6 +778,12 @@ <h2>Submodules<a class="headerlink" href="#submodules" title="Permalink to this
 </dl>
 </dd></dl>
 
+<dl class="py function">
+<dt class="sig sig-object py" id="tilelang.transform.get_pass_context">
+<span class="sig-prename descclassname"><span class="pre">tilelang.transform.</span></span><span class="sig-name descname"><span class="pre">get_pass_context</span></span><span class="sig-paren">(</span><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.transform.get_pass_context" title="Permalink to this definition">#</a></dt>
+<dd><p>Get the current pass context</p>
+</dd></dl>
+
 </section>
 </section>
 
@@ -865,6 +871,7 @@ <h2>Submodules<a class="headerlink" href="#submodules" title="Permalink to this
 <li><a class="reference internal" href="#tilelang.transform.VectorizeLoop"><code class="docutils literal notranslate"><span class="pre">VectorizeLoop()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.transform.WarpSpecialized"><code class="docutils literal notranslate"><span class="pre">WarpSpecialized()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.transform.WarpSpecializedPipeline"><code class="docutils literal notranslate"><span class="pre">WarpSpecializedPipeline()</span></code></a></li>
+<li><a class="reference internal" href="#tilelang.transform.get_pass_context"><code class="docutils literal notranslate"><span class="pre">get_pass_context()</span></code></a></li>
 </ul>
 </li>
 </ul>

genindex.html

@@ -452,6 +452,8 @@ <h2>A</h2>
         <li><a href="api/tilelang.language.allocate.html#tilelang.language.allocate.alloc_shared">alloc_shared() (in module tilelang.language.allocate)</a>
 </li>
         <li><a href="api/tilelang.language.allocate.html#tilelang.language.allocate.alloc_var">alloc_var() (in module tilelang.language.allocate)</a>
+</li>
+        <li><a href="api/tilelang.engine.phase.html#tilelang.engine.phase.allow_tma_and_warp_specialized">allow_tma_and_warp_specialized() (in module tilelang.engine.phase)</a>
 </li>
         <li><a href="api/tilelang.tools.Analyzer.html#tilelang.tools.Analyzer.Analyzer.analysis">analysis() (tilelang.tools.Analyzer.Analyzer class method)</a>
 </li>
@@ -472,11 +474,11 @@ <h2>A</h2>
         <li><a href="api/tilelang.language.tir.op.html#tilelang.language.tir.op.anylist_getitem">anylist_getitem() (in module tilelang.language.tir.op)</a>
 </li>
         <li><a href="api/tilelang.language.tir.op.html#tilelang.language.tir.op.anylist_resetitem">anylist_resetitem() (in module tilelang.language.tir.op)</a>
-</li>
-        <li><a href="api/tilelang.language.tir.op.html#tilelang.language.tir.op.anylist_setitem_call_cpacked">anylist_setitem_call_cpacked() (in module tilelang.language.tir.op)</a>
 </li>
     </ul></td>
     <td style="width: 33%; vertical-align: top;"><ul>
+        <li><a href="api/tilelang.language.tir.op.html#tilelang.language.tir.op.anylist_setitem_call_cpacked">anylist_setitem_call_cpacked() (in module tilelang.language.tir.op)</a>
+</li>
         <li><a href="api/tilelang.language.tir.op.html#tilelang.language.tir.op.anylist_setitem_call_packed">anylist_setitem_call_packed() (in module tilelang.language.tir.op)</a>
 </li>
         <li><a href="api/tilelang.transform.simplify.html#tilelang.transform.simplify.apply_simplify">apply_simplify() (in module tilelang.transform.simplify)</a>
@@ -1338,6 +1340,8 @@ <h2>G</h2>
         <li><a href="api/tilelang.carver.roller.node.html#tilelang.carver.roller.node.BlockAnalyzer.get_output_buffers">get_output_buffers() (tilelang.carver.roller.node.BlockAnalyzer method)</a>
 </li>
         <li><a href="api/tilelang.layout.layout.html#tilelang.layout.layout.Layout.get_output_shape">get_output_shape() (tilelang.layout.layout.Layout method)</a>
+</li>
+        <li><a href="api/tilelang.transform.html#tilelang.transform.get_pass_context">get_pass_context() (in module tilelang.transform)</a>
 </li>
         <li><a href="api/tilelang.carver.roller.node.html#tilelang.carver.roller.node.BlockAnalyzer.get_producer_blocks">get_producer_blocks() (tilelang.carver.roller.node.BlockAnalyzer method)</a>
 </li>