Skip to content

Commit d64c6da

Browse files
Update docs
1 parent 02b2cce commit d64c6da

File tree

20 files changed

+33
-431
lines changed

20 files changed

+33
-431
lines changed

_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/carver/template/general_reduce/index.rst.txt

Lines changed: 0 additions & 18 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:: structure
2924
:type: Union[str, List[str]]
3025
:value: None
@@ -45,19 +40,6 @@ Module Contents
4540

4641
.. py:method:: get_hardware_aware_configs(arch = None, topk = 10)
4742
48-
Abstract method that must be implemented by subclasses.
49-
It should return a list of hardware-aware configurations (hints)
50-
based on the specified architecture.
51-
52-
:param arch: The target architecture. Defaults to None.
53-
:type arch: TileDevice, optional
54-
:param topk: Number of top configurations to return. Defaults to 10.
55-
:type topk: int, optional
56-
57-
:returns: A list of recommended hardware-aware configurations.
58-
:rtype: List[Hint]
59-
60-
6143
6244
.. py:method:: initialize_function()
6345

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

Lines changed: 0 additions & 39 deletions
Original file line numberDiff line numberDiff line change
@@ -32,41 +32,26 @@ Module Contents
3232
Bases: :py:obj:`tvm.ir.base.Node`, :py:obj:`tvm.runtime.Scriptable`
3333

3434

35-
Base class of all IR Nodes.
36-
37-
3835
.. py:class:: AtomicAdd
3936
4037
Bases: :py:obj:`tvm.ir.base.Node`, :py:obj:`tvm.runtime.Scriptable`
4138

4239

43-
Base class of all IR Nodes.
44-
45-
4640
.. py:class:: Copy
4741
4842
Bases: :py:obj:`tvm.ir.base.Node`, :py:obj:`tvm.runtime.Scriptable`
4943

5044

51-
Base class of all IR Nodes.
52-
53-
5445
.. py:class:: Conv2DIm2ColOp
5546
5647
Bases: :py:obj:`tvm.ir.base.Node`, :py:obj:`tvm.runtime.Scriptable`
5748

5849

59-
Base class of all IR Nodes.
60-
61-
6250
.. py:class:: GemmWarpPolicy
6351
6452
Bases: :py:obj:`tvm.ir.base.Node`, :py:obj:`tvm.runtime.Scriptable`
6553

6654

67-
Base class of all IR Nodes.
68-
69-
7055
.. py:attribute:: policy_type
7156
:type: int
7257

@@ -87,62 +72,38 @@ Module Contents
8772
Bases: :py:obj:`tvm.ir.base.Node`, :py:obj:`tvm.runtime.Scriptable`
8873

8974

90-
Base class of all IR Nodes.
91-
92-
9375
.. py:class:: GemmSP
9476
9577
Bases: :py:obj:`tvm.ir.base.Node`, :py:obj:`tvm.runtime.Scriptable`
9678

9779

98-
Base class of all IR Nodes.
99-
100-
10180
.. py:class:: FinalizeReducerOp
10281
10382
Bases: :py:obj:`tvm.ir.base.Node`, :py:obj:`tvm.runtime.Scriptable`
10483

10584

106-
Base class of all IR Nodes.
107-
108-
10985
.. py:class:: ParallelOp
11086
11187
Bases: :py:obj:`tvm.ir.base.Node`, :py:obj:`tvm.runtime.Scriptable`
11288

11389

114-
Base class of all IR Nodes.
115-
116-
11790
.. py:class:: ReduceOp
11891
11992
Bases: :py:obj:`tvm.ir.base.Node`, :py:obj:`tvm.runtime.Scriptable`
12093

12194

122-
Base class of all IR Nodes.
123-
124-
12595
.. py:class:: CumSumOp
12696
12797
Bases: :py:obj:`tvm.ir.base.Node`, :py:obj:`tvm.runtime.Scriptable`
12898

12999

130-
Base class of all IR Nodes.
131-
132-
133100
.. py:class:: RegionOp
134101
135102
Bases: :py:obj:`tvm.ir.base.Node`, :py:obj:`tvm.runtime.Scriptable`
136103

137104

138-
Base class of all IR Nodes.
139-
140-
141105
.. py:class:: ReduceType
142106
143107
Bases: :py:obj:`tvm.ir.base.Node`, :py:obj:`tvm.runtime.Scriptable`
144108

145109

146-
Base class of all IR Nodes.
147-
148-

_sources/autoapi/tilelang/jit/adapter/dlpack/index.rst.txt

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -20,12 +20,8 @@ Classes
2020
Module Contents
2121
---------------
2222

23-
.. py:class:: TorchDLPackKernelAdapter(mod, params, result_idx)
23+
.. py:class:: TorchDLPackKernelAdapter
2424
2525
Bases: :py:obj:`tilelang.jit.adapter.base.BaseKernelAdapter`
2626

2727

28-
Helper class that provides a standard way to create an ABC using
29-
inheritance.
30-
31-

_sources/autoapi/tilelang/jit/adapter/nvrtc/adapter/index.rst.txt

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -80,8 +80,6 @@ Module Contents
8080
8181
8282
.. py:attribute:: target
83-
:value: None
84-
8583
8684
8785
.. py:attribute:: verbose

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

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -25,9 +25,6 @@ Module Contents
2525
Bases: :py:obj:`tvm.ir.Node`
2626

2727

28-
Base class of all IR Nodes.
29-
30-
3128
.. py:property:: index
3229
3330
Property to retrieve the forward index of the layout.

_sources/autoapi/tilelang/tileop/gemm/index.rst.txt

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -43,9 +43,6 @@ Package Contents
4343
Bases: :py:obj:`tvm.ir.base.Node`, :py:obj:`tvm.runtime.Scriptable`
4444

4545

46-
Base class of all IR Nodes.
47-
48-
4946
.. py:attribute:: A
5047
:type: tvm.tir.Buffer
5148

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

Lines changed: 0 additions & 109 deletions
Original file line numberDiff line numberDiff line change
@@ -4,15 +4,6 @@ tilelang.transform.add_bufstore_wrapper
44
.. py:module:: tilelang.transform.add_bufstore_wrapper
55
66
7-
Classes
8-
-------
9-
10-
.. autoapisummary::
11-
12-
tilelang.transform.add_bufstore_wrapper.FindVarUse
13-
tilelang.transform.add_bufstore_wrapper.AddWrapperForSingleStoreMutator
14-
15-
167
Functions
178
---------
189

@@ -24,105 +15,5 @@ Functions
2415
Module Contents
2516
---------------
2617

27-
.. py:class:: FindVarUse
28-
29-
Bases: :py:obj:`tvm.tir.PyStmtExprVisitor`
30-
31-
32-
A Python StmtExprVisitor to define custom visitor for both Stmt and PrimExpr.
33-
34-
Users can customize any of the visit function.
35-
36-
37-
.. py:attribute:: used_var
38-
39-
40-
.. py:method:: visit_var_(op)
41-
42-
Visit Var.
43-
44-
Users can customize this function to overwrite VisitVar_(const VarNode* op)
45-
on the C++ side.
46-
47-
:param op: The Var to be visited.
48-
:type op: Var
49-
50-
51-
52-
.. py:class:: AddWrapperForSingleStoreMutator
53-
54-
Bases: :py:obj:`tvm.tir.PyStmtExprMutator`
55-
56-
57-
Add a dummy parallel for loop to wrap the single buffer store
58-
Condition:
59-
1. not inside a parallel for loop
60-
2. no custom thread binding, i.e. threadIdx.x, blockIdx.x
61-
62-
63-
.. py:attribute:: inside_pfor
64-
:value: 0
65-
66-
67-
68-
.. py:attribute:: thread_binding_var
69-
70-
71-
.. py:method:: visit_block_(op)
72-
73-
Visit Block.
74-
Users can customize this function to overwrite VisitStmt_(const BlockNode* op)
75-
on the C++ side.
76-
77-
:param op: The Block to be visited.
78-
:type op: Block
79-
80-
:returns: **result** -- The mutated Stmt.
81-
:rtype: Stmt
82-
83-
84-
85-
.. py:method:: visit_attr_stmt_(op)
86-
87-
Visit AttrStmt.
88-
Users can customize this function to overwrite VisitStmt_(const AttrStmtNode* op)
89-
on the C++ side.
90-
91-
:param op: The AttrStmt to be visited.
92-
:type op: AttrStmt
93-
94-
:returns: **result** -- The mutated Stmt.
95-
:rtype: Stmt
96-
97-
98-
99-
.. py:method:: visit_for_(op)
100-
101-
Visit For.
102-
Users can customize this function to overwrite VisitStmt_(const ForNode* op)
103-
on the C++ side.
104-
105-
:param op: The For to be visited.
106-
:type op: For
107-
108-
:returns: **result** -- The mutated Stmt.
109-
:rtype: Stmt
110-
111-
112-
113-
.. py:method:: visit_buffer_store_(op)
114-
115-
Visit BufferStore.
116-
Users can customize this function to overwrite VisitStmt_(const BufferStoreNode* op)
117-
on the C++ side.
118-
119-
:param op: The BufferStore to be visited.
120-
:type op: BufferStore
121-
122-
:returns: **result** -- The mutated Stmt.
123-
:rtype: Stmt
124-
125-
126-
12718
.. py:function:: AddWrapperForSingleBufStore()
12819

autoapi/tilelang/carver/template/flashattention/index.html

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -476,7 +476,7 @@ <h2>Classes<a class="headerlink" href="#classes" title="Link to this heading">¶
476476
<table class="autosummary longtable docutils align-default">
477477
<tbody>
478478
<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.carver.template.flashattention.FlashAttentionTemplate" title="tilelang.carver.template.flashattention.FlashAttentionTemplate"><code class="xref py py-obj docutils literal notranslate"><span class="pre">FlashAttentionTemplate</span></code></a></p></td>
479-
<td><p>Base class template for hardware-aware configurations.</p></td>
479+
<td><p></p></td>
480480
</tr>
481481
</tbody>
482482
</table>
@@ -488,9 +488,6 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
488488
<dt class="sig sig-object py" id="tilelang.carver.template.flashattention.FlashAttentionTemplate">
489489
<em class="property"><span class="pre">class</span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">tilelang.carver.template.flashattention.</span></span><span class="sig-name descname"><span class="pre">FlashAttentionTemplate</span></span><a class="headerlink" href="#tilelang.carver.template.flashattention.FlashAttentionTemplate" title="Link to this definition"></a></dt>
490490
<dd><p>Bases: <a class="reference internal" href="../base/index.html#tilelang.carver.template.base.BaseTemplate" title="tilelang.carver.template.base.BaseTemplate"><code class="xref py py-obj docutils literal notranslate"><span class="pre">tilelang.carver.template.base.BaseTemplate</span></code></a></p>
491-
<p>Base class template for hardware-aware configurations.
492-
This serves as an abstract base class (ABC) that defines the structure
493-
for subclasses implementing hardware-specific optimizations.</p>
494491
<dl class="py attribute">
495492
<dt class="sig sig-object py" id="tilelang.carver.template.flashattention.FlashAttentionTemplate.batch_size">
496493
<span class="sig-name descname"><span class="pre">batch_size</span></span><em class="property"><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="pre">int</span></em><em class="property"><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="pre">1</span></em><a class="headerlink" href="#tilelang.carver.template.flashattention.FlashAttentionTemplate.batch_size" title="Link to this definition"></a></dt>

autoapi/tilelang/carver/template/general_reduce/index.html

Lines changed: 6 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -476,7 +476,7 @@ <h2>Classes<a class="headerlink" href="#classes" title="Link to this heading">¶
476476
<table class="autosummary longtable docutils align-default">
477477
<tbody>
478478
<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.carver.template.general_reduce.GeneralReductionTemplate" title="tilelang.carver.template.general_reduce.GeneralReductionTemplate"><code class="xref py py-obj docutils literal notranslate"><span class="pre">GeneralReductionTemplate</span></code></a></p></td>
479-
<td><p>Base class template for hardware-aware configurations.</p></td>
479+
<td><p></p></td>
480480
</tr>
481481
</tbody>
482482
</table>
@@ -488,9 +488,6 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
488488
<dt class="sig sig-object py" id="tilelang.carver.template.general_reduce.GeneralReductionTemplate">
489489
<em class="property"><span class="pre">class</span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">tilelang.carver.template.general_reduce.</span></span><span class="sig-name descname"><span class="pre">GeneralReductionTemplate</span></span><a class="headerlink" href="#tilelang.carver.template.general_reduce.GeneralReductionTemplate" title="Link to this definition"></a></dt>
490490
<dd><p>Bases: <a class="reference internal" href="../base/index.html#tilelang.carver.template.base.BaseTemplate" title="tilelang.carver.template.base.BaseTemplate"><code class="xref py py-obj docutils literal notranslate"><span class="pre">tilelang.carver.template.base.BaseTemplate</span></code></a></p>
491-
<p>Base class template for hardware-aware configurations.
492-
This serves as an abstract base class (ABC) that defines the structure
493-
for subclasses implementing hardware-specific optimizations.</p>
494491
<dl class="py attribute">
495492
<dt class="sig sig-object py" id="tilelang.carver.template.general_reduce.GeneralReductionTemplate.structure">
496493
<span class="sig-name descname"><span class="pre">structure</span></span><em class="property"><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="pre">str</span><span class="w"> </span><span class="p"><span class="pre">|</span></span><span class="w"> </span><span class="pre">List</span><span class="p"><span class="pre">[</span></span><span class="pre">str</span><span class="p"><span class="pre">]</span></span></em><em class="property"><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="pre">None</span></em><a class="headerlink" href="#tilelang.carver.template.general_reduce.GeneralReductionTemplate.structure" title="Link to this definition"></a></dt>
@@ -509,21 +506,15 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
509506
<dl class="py method">
510507
<dt class="sig sig-object py" id="tilelang.carver.template.general_reduce.GeneralReductionTemplate.get_hardware_aware_configs">
511508
<span class="sig-name descname"><span class="pre">get_hardware_aware_configs</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">arch</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">topk</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">10</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.carver.template.general_reduce.GeneralReductionTemplate.get_hardware_aware_configs" title="Link to this definition"></a></dt>
512-
<dd><p>Abstract method that must be implemented by subclasses.
513-
It should return a list of hardware-aware configurations (hints)
514-
based on the specified architecture.</p>
515-
<dl class="field-list simple">
509+
<dd><dl class="field-list simple">
516510
<dt class="field-odd">Parameters<span class="colon">:</span></dt>
517511
<dd class="field-odd"><ul class="simple">
518-
<li><p><strong>arch</strong> (<a class="reference internal" href="../../arch/arch_base/index.html#tilelang.carver.arch.arch_base.TileDevice" title="tilelang.carver.arch.arch_base.TileDevice"><em>TileDevice</em></a><em>, </em><em>optional</em>) – The target architecture. Defaults to None.</p></li>
519-
<li><p><strong>topk</strong> (<em>int</em><em>, </em><em>optional</em>) – Number of top configurations to return. Defaults to 10.</p></li>
512+
<li><p><strong>arch</strong> (<em>tilelang.carver.arch.TileDevice</em>)</p></li>
513+
<li><p><strong>topk</strong> (<em>int</em>)</p></li>
520514
</ul>
521515
</dd>
522-
<dt class="field-even">Returns<span class="colon">:</span></dt>
523-
<dd class="field-even"><p>A list of recommended hardware-aware configurations.</p>
524-
</dd>
525-
<dt class="field-odd">Return type<span class="colon">:</span></dt>
526-
<dd class="field-odd"><p>List[<a class="reference internal" href="../../roller/hint/index.html#tilelang.carver.roller.hint.Hint" title="tilelang.carver.roller.hint.Hint">Hint</a>]</p>
516+
<dt class="field-even">Return type<span class="colon">:</span></dt>
517+
<dd class="field-even"><p>List[tilelang.carver.roller.Hint]</p>
527518
</dd>
528519
</dl>
529520
</dd></dl>

0 commit comments

Comments
 (0)