Skip to content

Commit 4b418ea

Browse files
Update docs
1 parent fde941e commit 4b418ea

File tree

22 files changed

+93
-146
lines changed

22 files changed

+93
-146
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/pass_config/index.rst.txt

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,11 +63,22 @@ Module Contents
6363
:value: 'tl.disable_fast_math'
6464

6565

66-
False
66+
True
67+
will be deprecated in the 0.1.7 release
6768

6869
:type: Disable fast math optimization. Default
6970

7071

72+
.. py:attribute:: TL_ENABLE_FAST_MATH
73+
:value: 'tl.enable_fast_math'
74+
75+
76+
False
77+
if enabled, --use_fast_math will be passed to nvcc
78+
79+
:type: Enable fast math optimization. Default
80+
81+
7182
.. py:attribute:: TL_PTXAS_REGISTER_USAGE_LEVEL
7283
:value: 'tl.ptxas_register_usage_level'
7384

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

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,19 +9,28 @@ Functions
99

1010
.. autoapisummary::
1111

12+
tilelang.utils.deprecated.deprecated_warning
1213
tilelang.utils.deprecated.deprecated
1314

1415

1516
Module Contents
1617
---------------
1718

18-
.. py:function:: deprecated(method_name, new_method_name)
19+
.. py:function:: deprecated_warning(method_name, new_method_name, phaseout_version = None)
20+
21+
A function to indicate that a method is deprecated
22+
23+
24+
25+
.. py:function:: deprecated(method_name, new_method_name, phaseout_version = None)
1926
2027
A decorator to indicate that a method is deprecated
2128

2229
:param method_name: The name of the method to deprecate
2330
:type method_name: str
2431
:param new_method_name: The name of the new method to use instead
2532
:type new_method_name: str
33+
:param phaseout_version: The version to phase out the method
34+
:type phaseout_version: str
2635

2736

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>

0 commit comments

Comments
 (0)