Skip to content

Commit 4ae16bf

Browse files
committed
Update docs
1 parent c50b4da commit 4ae16bf

8 files changed

+16
-6
lines changed

api/tilelang.html

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -889,6 +889,7 @@ <h2>Subpackages<a class="headerlink" href="#subpackages" title="Permalink to thi
889889
</ul>
890890
</li>
891891
<li class="toctree-l3"><a class="reference internal" href="tilelang.language.reduce.html">tilelang.language.reduce module</a><ul>
892+
<li class="toctree-l4"><a class="reference internal" href="tilelang.language.reduce.html#tilelang.language.reduce.cumsum"><code class="docutils literal notranslate"><span class="pre">cumsum()</span></code></a></li>
892893
<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>
893894
<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>
894895
<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>

api/tilelang.language.html

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -718,6 +718,7 @@ <h2>Submodules<a class="headerlink" href="#submodules" title="Permalink to this
718718
</ul>
719719
</li>
720720
<li class="toctree-l1"><a class="reference internal" href="tilelang.language.reduce.html">tilelang.language.reduce module</a><ul>
721+
<li class="toctree-l2"><a class="reference internal" href="tilelang.language.reduce.html#tilelang.language.reduce.cumsum"><code class="docutils literal notranslate"><span class="pre">cumsum()</span></code></a></li>
721722
<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>
722723
<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>
723724
<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>

api/tilelang.language.logical.html

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -404,7 +404,7 @@
404404
<p>The language interface for tl programs.</p>
405405
<dl class="py function">
406406
<dt class="sig sig-object py" id="tilelang.language.logical.all_of">
407-
<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">0x7f3042cb45e0&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>
407+
<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">0x7f9fc08ade40&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>
408408
<dd><p>Check if all elements in the buffer are true.</p>
409409
<dl class="field-list simple">
410410
<dt class="field-odd">Parameters<span class="colon">:</span></dt>
@@ -418,7 +418,7 @@
418418

419419
<dl class="py function">
420420
<dt class="sig sig-object py" id="tilelang.language.logical.any_of">
421-
<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">0x7f3042cb45e0&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>
421+
<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">0x7f9fc08ade40&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>
422422
<dd><p>Check if any element in the buffer is true.</p>
423423
<dl class="field-list simple">
424424
<dt class="field-odd">Parameters<span class="colon">:</span></dt>

api/tilelang.language.proxy.html

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -452,7 +452,7 @@
452452
<p>Buffer proxy class for constructing tir buffer.</p>
453453
<dl class="py method">
454454
<dt class="sig sig-object py" id="tilelang.language.proxy.BufferProxy.from_ptr">
455-
<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">0x7f3042cb45b0&gt;</span></span></span><a class="headerlink" href="#tilelang.language.proxy.BufferProxy.from_ptr" title="Permalink to this definition">#</a></dt>
455+
<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">0x7f9fc08ade10&gt;</span></span></span><a class="headerlink" href="#tilelang.language.proxy.BufferProxy.from_ptr" title="Permalink to this definition">#</a></dt>
456456
<dd><p>Create a buffer from a pointer, shape, and data type.</p>
457457
<dl class="field-list simple">
458458
<dt class="field-odd">Parameters<span class="colon">:</span></dt>

api/tilelang.language.reduce.html

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -402,6 +402,11 @@
402402
<section id="module-tilelang.language.reduce">
403403
<span id="tilelang-language-reduce-module"></span><h1>tilelang.language.reduce module<a class="headerlink" href="#module-tilelang.language.reduce" title="Permalink to this heading">#</a></h1>
404404
<p>The language interface for tl programs.</p>
405+
<dl class="py function">
406+
<dt class="sig sig-object py" id="tilelang.language.reduce.cumsum">
407+
<span class="sig-prename descclassname"><span class="pre">tilelang.language.reduce.</span></span><span class="sig-name descname"><span class="pre">cumsum</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">src</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">dst</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">Optional</span><span class="p"><span class="pre">[</span></span><span class="pre">Buffer</span><span class="p"><span class="pre">]</span></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">None</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><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">0</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">reverse</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">bool</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">False</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.language.reduce.cumsum" title="Permalink to this definition">#</a></dt>
408+
<dd></dd></dl>
409+
405410
<dl class="py function">
406411
<dt class="sig sig-object py" id="tilelang.language.reduce.reduce">
407412
<span class="sig-prename descclassname"><span class="pre">tilelang.language.reduce.</span></span><span class="sig-name descname"><span class="pre">reduce</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">reduce_type</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">str</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>, <em class="sig-param"><span class="n"><span class="pre">clear</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">bool</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.language.reduce.reduce" title="Permalink to this definition">#</a></dt>
@@ -590,6 +595,7 @@
590595
<div class="toc-tree">
591596
<ul>
592597
<li><a class="reference internal" href="#">tilelang.language.reduce module</a><ul>
598+
<li><a class="reference internal" href="#tilelang.language.reduce.cumsum"><code class="docutils literal notranslate"><span class="pre">cumsum()</span></code></a></li>
593599
<li><a class="reference internal" href="#tilelang.language.reduce.reduce"><code class="docutils literal notranslate"><span class="pre">reduce()</span></code></a></li>
594600
<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>
595601
<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>

genindex.html

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -733,10 +733,10 @@ <h2>C</h2>
733733
</li>
734734
<li><a href="api/tilelang.language.tir.op.html#tilelang.language.tir.op.clz">clz() (in module tilelang.language.tir.op)</a>
735735
</li>
736-
</ul></td>
737-
<td style="width: 33%; vertical-align: top;"><ul>
738736
<li><a href="api/tilelang.carver.roller.policy.common.html#tilelang.carver.roller.policy.common.coalesced_factor">coalesced_factor() (in module tilelang.carver.roller.policy.common)</a>
739737
</li>
738+
</ul></td>
739+
<td style="width: 33%; vertical-align: top;"><ul>
740740
<li><a href="api/tilelang.carver.roller.policy.common.html#tilelang.carver.roller.policy.common.coalesced_tensor_shape">coalesced_tensor_shape() (in module tilelang.carver.roller.policy.common)</a>
741741
</li>
742742
<li><a href="api/tilelang.carver.analysis.html#tilelang.carver.analysis.collect_block_iter_vars_used_in_access_region">collect_block_iter_vars_used_in_access_region() (in module tilelang.carver.analysis)</a>
@@ -826,6 +826,8 @@ <h2>C</h2>
826826
<li><a href="api/tilelang.carver.arch.cuda.html#tilelang.carver.arch.cuda.CUDA">CUDA (class in tilelang.carver.arch.cuda)</a>
827827
</li>
828828
<li><a href="api/tilelang.carver.arch.driver.cuda_driver.html#tilelang.carver.arch.driver.cuda_driver.cudaDeviceProp">cudaDeviceProp (class in tilelang.carver.arch.driver.cuda_driver)</a>
829+
</li>
830+
<li><a href="api/tilelang.language.reduce.html#tilelang.language.reduce.cumsum">cumsum() (in module tilelang.language.reduce)</a>
829831
</li>
830832
<li><a href="api/tilelang.language.frame.html#tilelang.language.frame.LetFrame.Current">Current() (tilelang.language.frame.LetFrame class method)</a>
831833

objects.inv

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