Skip to content

Commit c8577df

Browse files
committed
Update docs
1 parent 2003298 commit c8577df

File tree

23 files changed

+140
-45
lines changed

23 files changed

+140
-45
lines changed

_sources/autoapi/tilelang/carver/arch/cdna/index.rst.txt

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -30,9 +30,6 @@ Module Contents
3030
Bases: :py:obj:`tilelang.carver.arch.arch_base.TileDevice`
3131

3232

33-
Represents the architecture of a computing device, capturing various hardware specifications.
34-
35-
3633
.. py:attribute:: target
3734
3835

_sources/autoapi/tilelang/carver/arch/cpu/index.rst.txt

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -30,9 +30,6 @@ Module Contents
3030
Bases: :py:obj:`tilelang.carver.arch.arch_base.TileDevice`
3131

3232

33-
Represents the architecture of a computing device, capturing various hardware specifications.
34-
35-
3633
.. py:attribute:: target
3734
3835

_sources/autoapi/tilelang/carver/arch/cuda/index.rst.txt

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -92,9 +92,6 @@ Module Contents
9292
Bases: :py:obj:`tilelang.carver.arch.arch_base.TileDevice`
9393

9494

95-
Represents the architecture of a computing device, capturing various hardware specifications.
96-
97-
9895
.. py:attribute:: target
9996
10097

_sources/autoapi/tilelang/carver/arch/driver/cuda_driver/index.rst.txt

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -35,9 +35,6 @@ Module Contents
3535
Bases: :py:obj:`ctypes.Structure`
3636

3737

38-
Structure base class
39-
40-
4138
.. py:function:: get_cuda_device_properties(device_id = 0)
4239
4340
.. py:function:: get_device_name(device_id = 0)

_sources/autoapi/tilelang/carver/roller/hint/index.rst.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -307,12 +307,12 @@ Module Contents
307307
308308
309309
.. py:property:: raxis_order
310-
:type: tilelang.carver.roller.rasterization.List[int]
310+
:type: List[int]
311311

312312

313313

314314
.. py:property:: step
315-
:type: tilelang.carver.roller.rasterization.List[int]
315+
:type: List[int]
316316

317317

318318

_sources/autoapi/tilelang/carver/template/flashattention/index.rst.txt

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -20,11 +20,6 @@ Module Contents
2020
Bases: :py:obj:`tilelang.carver.template.base.BaseTemplate`
2121

2222

23-
Base class template for hardware-aware configurations.
24-
This serves as an abstract base class (ABC) that defines the structure
25-
for subclasses implementing hardware-specific optimizations.
26-
27-
2823
.. py:attribute:: batch_size
2924
:type: int
3025
:value: 1

_sources/autoapi/tilelang/language/tir/entry/index.rst.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ Functions
1616
Module Contents
1717
---------------
1818

19-
.. py:function:: prim_func(func = None, private = False, check_well_formed=True)
19+
.. py:function:: prim_func(func = None, private = False, check_well_formed=False)
2020
2121
The parsing method for tir prim func, by using `@prim_func` as decorator.
2222

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

Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -58,6 +58,10 @@ Functions
5858
tilelang.transform.PersistThreadblock
5959
tilelang.transform.AlignDynamicSharedMemoryAllocations
6060
tilelang.transform.LowerSharedBarrier
61+
tilelang.transform.StorageRewrite
62+
tilelang.transform.LowerOpaqueBlock
63+
tilelang.transform.LowerThreadAllreduce
64+
tilelang.transform.LowerDeviceKernelLaunch
6165

6266

6367
Package Contents
@@ -332,3 +336,29 @@ Package Contents
332336

333337

334338

339+
.. py:function:: StorageRewrite()
340+
341+
StorageRewrite
342+
343+
:returns: **fpass** -- The result pass
344+
:rtype: tvm.transform.Pass
345+
346+
347+
.. py:function:: LowerOpaqueBlock()
348+
349+
LowerOpaqueBlock
350+
351+
352+
353+
.. py:function:: LowerThreadAllreduce()
354+
355+
LowerThreadAllreduce
356+
357+
358+
359+
.. py:function:: LowerDeviceKernelLaunch()
360+
361+
LowerDeviceKernelLaunch
362+
363+
364+

_sources/autoapi/tilelang/utils/language/index.rst.txt

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@ Functions
1717
tilelang.utils.language.get_buffer_elems
1818
tilelang.utils.language.array_reduce
1919
tilelang.utils.language.retrieve_func_from_module
20+
tilelang.utils.language.get_buffer_region_from_load
2021

2122

2223
Module Contents
@@ -108,3 +109,12 @@ Module Contents
108109
:raises AssertionError: If the module contains more than one global function.
109110

110111

112+
.. py:function:: get_buffer_region_from_load(buffer_load)
113+
114+
Get the buffer region from a buffer load.
115+
116+
May encounter buffer load like C[0:128, 0:32], ref to pull request
117+
for buffer wise op: https://github.com/apache/tvm/pull/14693
118+
convert load to region
119+
120+

autoapi/tilelang/carver/arch/cdna/index.html

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -465,7 +465,7 @@ <h2>Classes<a class="headerlink" href="#classes" title="Link to this heading">¶
465465
<table class="autosummary longtable docutils align-default">
466466
<tbody>
467467
<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.carver.arch.cdna.CDNA" title="tilelang.carver.arch.cdna.CDNA"><code class="xref py py-obj docutils literal notranslate"><span class="pre">CDNA</span></code></a></p></td>
468-
<td><p>Represents the architecture of a computing device, capturing various hardware specifications.</p></td>
468+
<td><p></p></td>
469469
</tr>
470470
</tbody>
471471
</table>
@@ -502,7 +502,6 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
502502
<dt class="sig sig-object py" id="tilelang.carver.arch.cdna.CDNA">
503503
<em class="property"><span class="pre">class</span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">tilelang.carver.arch.cdna.</span></span><span class="sig-name descname"><span class="pre">CDNA</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">target</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.carver.arch.cdna.CDNA" title="Link to this definition"></a></dt>
504504
<dd><p>Bases: <a class="reference internal" href="../arch_base/index.html#tilelang.carver.arch.arch_base.TileDevice" title="tilelang.carver.arch.arch_base.TileDevice"><code class="xref py py-obj docutils literal notranslate"><span class="pre">tilelang.carver.arch.arch_base.TileDevice</span></code></a></p>
505-
<p>Represents the architecture of a computing device, capturing various hardware specifications.</p>
506505
<dl class="field-list simple">
507506
<dt class="field-odd">Parameters<span class="colon">:</span></dt>
508507
<dd class="field-odd"><p><strong>target</strong> (<em>Union</em><em>[</em><em>tvm.target.Target</em><em>, </em><em>str</em><em>]</em>)</p>

0 commit comments

Comments
 (0)