Update docs

Author: LeiWang1999

api/tilelang.language.html

@@ -498,6 +498,7 @@ <h2>Subpackages<a class="headerlink" href="#subpackages" title="Permalink to thi
 <li class="toctree-l4"><a class="reference internal" href="tilelang.language.tir.op.html#tilelang.language.tir.op.nextafter"><code class="docutils literal notranslate"><span class="pre">nextafter()</span></code></a></li>
 <li class="toctree-l4"><a class="reference internal" href="tilelang.language.tir.op.html#tilelang.language.tir.op.popcount"><code class="docutils literal notranslate"><span class="pre">popcount()</span></code></a></li>
 <li class="toctree-l4"><a class="reference internal" href="tilelang.language.tir.op.html#tilelang.language.tir.op.pow"><code class="docutils literal notranslate"><span class="pre">pow()</span></code></a></li>
+<li class="toctree-l4"><a class="reference internal" href="tilelang.language.tir.op.html#tilelang.language.tir.op.pow_of_int"><code class="docutils literal notranslate"><span class="pre">pow_of_int()</span></code></a></li>
 <li class="toctree-l4"><a class="reference internal" href="tilelang.language.tir.op.html#tilelang.language.tir.op.power"><code class="docutils literal notranslate"><span class="pre">power()</span></code></a></li>
 <li class="toctree-l4"><a class="reference internal" href="tilelang.language.tir.op.html#tilelang.language.tir.op.ptx_arrive_barrier"><code class="docutils literal notranslate"><span class="pre">ptx_arrive_barrier()</span></code></a></li>
 <li class="toctree-l4"><a class="reference internal" href="tilelang.language.tir.op.html#tilelang.language.tir.op.ptx_arrive_barrier_expect_tx"><code class="docutils literal notranslate"><span class="pre">ptx_arrive_barrier_expect_tx()</span></code></a></li>

api/tilelang.language.logical.html

@@ -406,7 +406,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">0x7f5c64c06aa0&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">0x7fa639c23e50&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>
@@ -420,7 +420,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">0x7f5c64c06aa0&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">0x7fa639c23e50&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

@@ -454,7 +454,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">0x7f5c64c06ad0&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">0x7fa639c23ca0&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.language.tir.html

@@ -496,6 +496,7 @@ <h2>Submodules<a class="headerlink" href="#submodules" title="Permalink to this
 <li class="toctree-l2"><a class="reference internal" href="tilelang.language.tir.op.html#tilelang.language.tir.op.nextafter"><code class="docutils literal notranslate"><span class="pre">nextafter()</span></code></a></li>
 <li class="toctree-l2"><a class="reference internal" href="tilelang.language.tir.op.html#tilelang.language.tir.op.popcount"><code class="docutils literal notranslate"><span class="pre">popcount()</span></code></a></li>
 <li class="toctree-l2"><a class="reference internal" href="tilelang.language.tir.op.html#tilelang.language.tir.op.pow"><code class="docutils literal notranslate"><span class="pre">pow()</span></code></a></li>
+<li class="toctree-l2"><a class="reference internal" href="tilelang.language.tir.op.html#tilelang.language.tir.op.pow_of_int"><code class="docutils literal notranslate"><span class="pre">pow_of_int()</span></code></a></li>
 <li class="toctree-l2"><a class="reference internal" href="tilelang.language.tir.op.html#tilelang.language.tir.op.power"><code class="docutils literal notranslate"><span class="pre">power()</span></code></a></li>
 <li class="toctree-l2"><a class="reference internal" href="tilelang.language.tir.op.html#tilelang.language.tir.op.ptx_arrive_barrier"><code class="docutils literal notranslate"><span class="pre">ptx_arrive_barrier()</span></code></a></li>
 <li class="toctree-l2"><a class="reference internal" href="tilelang.language.tir.op.html#tilelang.language.tir.op.ptx_arrive_barrier_expect_tx"><code class="docutils literal notranslate"><span class="pre">ptx_arrive_barrier_expect_tx()</span></code></a></li>

api/tilelang.language.tir.op.html

@@ -1960,6 +1960,20 @@
 </dl>
 </dd></dl>
 
+<dl class="py function">
+<dt class="sig sig-object py" id="tilelang.language.tir.op.pow_of_int">
+<span class="sig-prename descclassname"><span class="pre">tilelang.language.tir.op.</span></span><span class="sig-name descname"><span class="pre">pow_of_int</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">x</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">PrimExpr</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">y</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">int</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">PrimExpr</span></span></span><a class="headerlink" href="#tilelang.language.tir.op.pow_of_int" title="Permalink to this definition">#</a></dt>
+<dd><p>Fast power operation than pow(float, float).</p>
+<dl class="field-list simple">
+<dt class="field-odd">Parameters<span class="colon">:</span></dt>
+<dd class="field-odd"><ul class="simple">
+<li><p><strong>x</strong> (<em>PrimExpr</em>) – Base value</p></li>
+<li><p><strong>y</strong> (<em>int</em>) – Exponent value</p></li>
+</ul>
+</dd>
+</dl>
+</dd></dl>
+
 <dl class="py function">
 <dt class="sig sig-object py" id="tilelang.language.tir.op.power">
 <span class="sig-prename descclassname"><span class="pre">tilelang.language.tir.op.</span></span><span class="sig-name descname"><span class="pre">power</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">x</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">y</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">span</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.tir.op.power" title="Permalink to this definition">#</a></dt>
@@ -3439,6 +3453,7 @@
 <li><a class="reference internal" href="#tilelang.language.tir.op.nextafter"><code class="docutils literal notranslate"><span class="pre">nextafter()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.language.tir.op.popcount"><code class="docutils literal notranslate"><span class="pre">popcount()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.language.tir.op.pow"><code class="docutils literal notranslate"><span class="pre">pow()</span></code></a></li>
+<li><a class="reference internal" href="#tilelang.language.tir.op.pow_of_int"><code class="docutils literal notranslate"><span class="pre">pow_of_int()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.language.tir.op.power"><code class="docutils literal notranslate"><span class="pre">power()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.language.tir.op.ptx_arrive_barrier"><code class="docutils literal notranslate"><span class="pre">ptx_arrive_barrier()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.language.tir.op.ptx_arrive_barrier_expect_tx"><code class="docutils literal notranslate"><span class="pre">ptx_arrive_barrier_expect_tx()</span></code></a></li>

genindex.html

@@ -2596,10 +2596,10 @@ <h2>P</h2>
           <li><a href="api/tilelang.carver.roller.policy.tensorcore.html#tilelang.carver.roller.policy.tensorcore.TensorCorePolicy.plan_rasterization">(tilelang.carver.roller.policy.tensorcore.TensorCorePolicy method)</a>
 </li>
         </ul></li>
-    </ul></td>
-    <td style="width: 33%; vertical-align: top;"><ul>
         <li><a href="api/tilelang.tools.plot_layout.html#tilelang.tools.plot_layout.plot_layout">plot_layout() (in module tilelang.tools.plot_layout)</a>
 </li>
+    </ul></td>
+    <td style="width: 33%; vertical-align: top;"><ul>
         <li><a href="api/tilelang.primitives.gemm.base.html#tilelang.primitives.gemm.base.GemmBaseParams.policy">policy (tilelang.primitives.gemm.base.GemmBaseParams attribute)</a>
 </li>
         <li><a href="api/tilelang.language.frame.html#tilelang.language.frame.FrameStack.pop">pop() (tilelang.language.frame.FrameStack method)</a>
@@ -2611,6 +2611,8 @@ <h2>P</h2>
         <li><a href="api/tilelang.language.tir.op.html#tilelang.language.tir.op.popcount">popcount() (in module tilelang.language.tir.op)</a>
 </li>
         <li><a href="api/tilelang.language.tir.op.html#tilelang.language.tir.op.pow">pow() (in module tilelang.language.tir.op)</a>
+</li>
+        <li><a href="api/tilelang.language.tir.op.html#tilelang.language.tir.op.pow_of_int">pow_of_int() (in module tilelang.language.tir.op)</a>
 </li>
         <li><a href="api/tilelang.language.tir.op.html#tilelang.language.tir.op.power">power() (in module tilelang.language.tir.op)</a>
 </li>