Update docs

Author: LeiWang1999

api/tilelang.html

@@ -848,6 +848,7 @@ <h2>Subpackages<a class="headerlink" href="#subpackages" title="Permalink to thi
 </li>
 <li class="toctree-l3"><a class="reference internal" href="tilelang.language.reduce.html">tilelang.language.reduce module</a><ul>
 <li class="toctree-l4"><a class="reference internal" href="tilelang.language.reduce.html#tilelang.language.reduce.reduce"><code class="docutils literal notranslate"><span class="pre">reduce()</span></code></a></li>
+<li class="toctree-l4"><a class="reference internal" href="tilelang.language.reduce.html#tilelang.language.reduce.reduce_absmax"><code class="docutils literal notranslate"><span class="pre">reduce_absmax()</span></code></a></li>
 <li class="toctree-l4"><a class="reference internal" href="tilelang.language.reduce.html#tilelang.language.reduce.reduce_abssum"><code class="docutils literal notranslate"><span class="pre">reduce_abssum()</span></code></a></li>
 <li class="toctree-l4"><a class="reference internal" href="tilelang.language.reduce.html#tilelang.language.reduce.reduce_max"><code class="docutils literal notranslate"><span class="pre">reduce_max()</span></code></a></li>
 <li class="toctree-l4"><a class="reference internal" href="tilelang.language.reduce.html#tilelang.language.reduce.reduce_min"><code class="docutils literal notranslate"><span class="pre">reduce_min()</span></code></a></li>

api/tilelang.language.html

@@ -716,6 +716,7 @@ <h2>Submodules<a class="headerlink" href="#submodules" title="Permalink to this
 </li>
 <li class="toctree-l1"><a class="reference internal" href="tilelang.language.reduce.html">tilelang.language.reduce module</a><ul>
 <li class="toctree-l2"><a class="reference internal" href="tilelang.language.reduce.html#tilelang.language.reduce.reduce"><code class="docutils literal notranslate"><span class="pre">reduce()</span></code></a></li>
+<li class="toctree-l2"><a class="reference internal" href="tilelang.language.reduce.html#tilelang.language.reduce.reduce_absmax"><code class="docutils literal notranslate"><span class="pre">reduce_absmax()</span></code></a></li>
 <li class="toctree-l2"><a class="reference internal" href="tilelang.language.reduce.html#tilelang.language.reduce.reduce_abssum"><code class="docutils literal notranslate"><span class="pre">reduce_abssum()</span></code></a></li>
 <li class="toctree-l2"><a class="reference internal" href="tilelang.language.reduce.html#tilelang.language.reduce.reduce_max"><code class="docutils literal notranslate"><span class="pre">reduce_max()</span></code></a></li>
 <li class="toctree-l2"><a class="reference internal" href="tilelang.language.reduce.html#tilelang.language.reduce.reduce_min"><code class="docutils literal notranslate"><span class="pre">reduce_min()</span></code></a></li>

api/tilelang.language.proxy.html

@@ -455,7 +455,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">0x7f966237b5b0&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">0x7f8b40f69e10&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.reduce.html

@@ -428,6 +428,27 @@
 </dl>
 </dd></dl>
 
+<dl class="py function">
+<dt class="sig sig-object py" id="tilelang.language.reduce.reduce_absmax">
+<span class="sig-prename descclassname"><span class="pre">tilelang.language.reduce.</span></span><span class="sig-name descname"><span class="pre">reduce_absmax</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">buffer</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">Buffer</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">out</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">Buffer</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">dim</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><a class="headerlink" href="#tilelang.language.reduce.reduce_absmax" title="Permalink to this definition">#</a></dt>
+<dd><p>Perform reduce absolute max on input buffer, store the result to output buffer.</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>buffer</strong> (<em>tir.Buffer</em>) – The input buffer</p></li>
+<li><p><strong>out</strong> (<em>tir.Buffer</em>) – The output buffer</p></li>
+<li><p><strong>dim</strong> (<em>int</em>) – The dimension to perform reduce on</p></li>
+</ul>
+</dd>
+<dt class="field-even">Returns<span class="colon">:</span></dt>
+<dd class="field-even"><p>Handle to the reduction operation</p>
+</dd>
+<dt class="field-odd">Return type<span class="colon">:</span></dt>
+<dd class="field-odd"><p>tir.Call</p>
+</dd>
+</dl>
+</dd></dl>
+
 <dl class="py function">
 <dt class="sig sig-object py" id="tilelang.language.reduce.reduce_abssum">
 <span class="sig-prename descclassname"><span class="pre">tilelang.language.reduce.</span></span><span class="sig-name descname"><span class="pre">reduce_abssum</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">buffer</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">Buffer</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">out</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">Buffer</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">dim</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><a class="headerlink" href="#tilelang.language.reduce.reduce_abssum" title="Permalink to this definition">#</a></dt>
@@ -573,6 +594,7 @@
             <ul>
 <li><a class="reference internal" href="#">tilelang.language.reduce module</a><ul>
 <li><a class="reference internal" href="#tilelang.language.reduce.reduce"><code class="docutils literal notranslate"><span class="pre">reduce()</span></code></a></li>
+<li><a class="reference internal" href="#tilelang.language.reduce.reduce_absmax"><code class="docutils literal notranslate"><span class="pre">reduce_absmax()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.language.reduce.reduce_abssum"><code class="docutils literal notranslate"><span class="pre">reduce_abssum()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.language.reduce.reduce_max"><code class="docutils literal notranslate"><span class="pre">reduce_max()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.language.reduce.reduce_min"><code class="docutils literal notranslate"><span class="pre">reduce_min()</span></code></a></li>

genindex.html

@@ -2586,6 +2586,8 @@ <h2>R</h2>
         <li><a href="api/tilelang.carver.template.base.html#tilelang.carver.template.base.BaseTemplate.recommend_hints">recommend_hints() (tilelang.carver.template.base.BaseTemplate method)</a>
 </li>
         <li><a href="api/tilelang.language.reduce.html#tilelang.language.reduce.reduce">reduce() (in module tilelang.language.reduce)</a>
+</li>
+        <li><a href="api/tilelang.language.reduce.html#tilelang.language.reduce.reduce_absmax">reduce_absmax() (in module tilelang.language.reduce)</a>
 </li>
         <li><a href="api/tilelang.language.reduce.html#tilelang.language.reduce.reduce_abssum">reduce_abssum() (in module tilelang.language.reduce)</a>
 </li>
@@ -2610,11 +2612,11 @@ <h2>R</h2>
         <li><a href="api/tilelang.engine.callback.html#tilelang.engine.callback.register_hip_postproc">register_hip_postproc() (in module tilelang.engine.callback)</a>
 </li>
         <li><a href="api/tilelang.engine.callback.html#tilelang.engine.callback.register_hip_postproc_callback">register_hip_postproc_callback() (in module tilelang.engine.callback)</a>
-</li>
-        <li><a href="api/tilelang.carver.arch.driver.cuda_driver.html#tilelang.carver.arch.driver.cuda_driver.cudaDeviceProp.regsPerBlock">regsPerBlock (tilelang.carver.arch.driver.cuda_driver.cudaDeviceProp attribute)</a>
 </li>
     </ul></td>
     <td style="width: 33%; vertical-align: top;"><ul>
+        <li><a href="api/tilelang.carver.arch.driver.cuda_driver.html#tilelang.carver.arch.driver.cuda_driver.cudaDeviceProp.regsPerBlock">regsPerBlock (tilelang.carver.arch.driver.cuda_driver.cudaDeviceProp attribute)</a>
+</li>
         <li><a href="api/tilelang.carver.arch.driver.cuda_driver.html#tilelang.carver.arch.driver.cuda_driver.cudaDeviceProp.regsPerMultiprocessor">regsPerMultiprocessor (tilelang.carver.arch.driver.cuda_driver.cudaDeviceProp attribute)</a>
 </li>
         <li><a href="api/tilelang.language.tir.op.html#tilelang.language.tir.op.reinterpret">reinterpret() (in module tilelang.language.tir.op)</a>