Skip to content

Commit 3679cf6

Browse files
Update docs
1 parent 763d61b commit 3679cf6

File tree

155 files changed

+1006
-93
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

155 files changed

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

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@ Submodules
1818
/autoapi/tilelang/engine/index
1919
/autoapi/tilelang/env/index
2020
/autoapi/tilelang/intrinsics/index
21+
/autoapi/tilelang/ir/index
2122
/autoapi/tilelang/jit/index
2223
/autoapi/tilelang/language/index
2324
/autoapi/tilelang/layout/index
Lines changed: 94 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,94 @@
1+
tilelang.ir
2+
===========
3+
4+
.. py:module:: tilelang.ir
5+
6+
7+
Classes
8+
-------
9+
10+
.. autoapisummary::
11+
12+
tilelang.ir.Fill
13+
tilelang.ir.AtomicAdd
14+
tilelang.ir.Copy
15+
tilelang.ir.Conv2DIm2ColOp
16+
tilelang.ir.GemmWarpPolicy
17+
tilelang.ir.Gemm
18+
tilelang.ir.GemmSP
19+
tilelang.ir.FinalizeReducerOp
20+
tilelang.ir.ParallelOp
21+
tilelang.ir.ReduceOp
22+
tilelang.ir.CumSumOp
23+
tilelang.ir.RegionOp
24+
tilelang.ir.ReduceType
25+
26+
27+
Module Contents
28+
---------------
29+
30+
.. py:class:: Fill
31+
32+
Bases: :py:obj:`tvm.ir.base.Node`, :py:obj:`tvm.runtime.Scriptable`
33+
34+
35+
.. py:class:: AtomicAdd
36+
37+
Bases: :py:obj:`tvm.ir.base.Node`, :py:obj:`tvm.runtime.Scriptable`
38+
39+
40+
.. py:class:: Copy
41+
42+
Bases: :py:obj:`tvm.ir.base.Node`, :py:obj:`tvm.runtime.Scriptable`
43+
44+
45+
.. py:class:: Conv2DIm2ColOp
46+
47+
Bases: :py:obj:`tvm.ir.base.Node`, :py:obj:`tvm.runtime.Scriptable`
48+
49+
50+
.. py:class:: GemmWarpPolicy
51+
52+
Bases: :py:obj:`tvm.ir.base.Node`, :py:obj:`tvm.runtime.Scriptable`
53+
54+
55+
.. py:class:: Gemm
56+
57+
Bases: :py:obj:`tvm.ir.base.Node`, :py:obj:`tvm.runtime.Scriptable`
58+
59+
60+
.. py:class:: GemmSP
61+
62+
Bases: :py:obj:`tvm.ir.base.Node`, :py:obj:`tvm.runtime.Scriptable`
63+
64+
65+
.. py:class:: FinalizeReducerOp
66+
67+
Bases: :py:obj:`tvm.ir.base.Node`, :py:obj:`tvm.runtime.Scriptable`
68+
69+
70+
.. py:class:: ParallelOp
71+
72+
Bases: :py:obj:`tvm.ir.base.Node`, :py:obj:`tvm.runtime.Scriptable`
73+
74+
75+
.. py:class:: ReduceOp
76+
77+
Bases: :py:obj:`tvm.ir.base.Node`, :py:obj:`tvm.runtime.Scriptable`
78+
79+
80+
.. py:class:: CumSumOp
81+
82+
Bases: :py:obj:`tvm.ir.base.Node`, :py:obj:`tvm.runtime.Scriptable`
83+
84+
85+
.. py:class:: RegionOp
86+
87+
Bases: :py:obj:`tvm.ir.base.Node`, :py:obj:`tvm.runtime.Scriptable`
88+
89+
90+
.. py:class:: ReduceType
91+
92+
Bases: :py:obj:`tvm.ir.base.Node`, :py:obj:`tvm.runtime.Scriptable`
93+
94+

_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/language/proxy/index.rst.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -120,7 +120,7 @@ Module Contents
120120
the tensor should be by default contiguous.
121121

122122

123-
.. py:method:: __call__(shape, dtype = 'float32', data=None)
123+
.. py:method:: __call__(shape, dtype = 'float32', data=None, scope=None)
124124
125125
126126
.. py:class:: StridedTensorProxy
@@ -133,7 +133,7 @@ Module Contents
133133
This class implements the default tensor proxy with global memory scope, with the stride information required.
134134

135135

136-
.. py:method:: __call__(shape, strides, dtype = 'float32')
136+
.. py:method:: __call__(shape, strides, dtype = 'float32', scope=None)
137137
138138
139139
.. py:class:: FragmentBufferProxy

_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.

autoapi/index.html

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -310,6 +310,7 @@
310310
<li class="toctree-l3"><a class="reference internal" href="tilelang/intrinsics/utils/index.html">tilelang.intrinsics.utils</a></li>
311311
</ul>
312312
</li>
313+
<li class="toctree-l2"><a class="reference internal" href="tilelang/ir/index.html">tilelang.ir</a></li>
313314
<li class="toctree-l2 has-children"><a class="reference internal" href="tilelang/jit/index.html">tilelang.jit</a><input class="toctree-checkbox" id="toctree-checkbox-15" name="toctree-checkbox-15" role="switch" type="checkbox"/><label for="toctree-checkbox-15"><div class="visually-hidden">Toggle navigation of tilelang.jit</div><i class="icon"><svg><use href="#svg-arrow-right"></use></svg></i></label><ul>
314315
<li class="toctree-l3 has-children"><a class="reference internal" href="tilelang/jit/adapter/index.html">tilelang.jit.adapter</a><input class="toctree-checkbox" id="toctree-checkbox-16" name="toctree-checkbox-16" role="switch" type="checkbox"/><label for="toctree-checkbox-16"><div class="visually-hidden">Toggle navigation of tilelang.jit.adapter</div><i class="icon"><svg><use href="#svg-arrow-right"></use></svg></i></label><ul>
315316
<li class="toctree-l4"><a class="reference internal" href="tilelang/jit/adapter/base/index.html">tilelang.jit.adapter.base</a></li>
@@ -549,6 +550,7 @@ <h1>API Reference<a class="headerlink" href="#api-reference" title="Link to this
549550
<li class="toctree-l3"><a class="reference internal" href="tilelang/intrinsics/utils/index.html">tilelang.intrinsics.utils</a></li>
550551
</ul>
551552
</li>
553+
<li class="toctree-l2"><a class="reference internal" href="tilelang/ir/index.html">tilelang.ir</a></li>
552554
<li class="toctree-l2"><a class="reference internal" href="tilelang/jit/index.html">tilelang.jit</a><ul>
553555
<li class="toctree-l3"><a class="reference internal" href="tilelang/jit/adapter/index.html">tilelang.jit.adapter</a><ul>
554556
<li class="toctree-l4"><a class="reference internal" href="tilelang/jit/adapter/base/index.html">tilelang.jit.adapter.base</a></li>

autoapi/tilelang/autotuner/capture/index.html

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -310,6 +310,7 @@
310310
<li class="toctree-l3"><a class="reference internal" href="../../intrinsics/utils/index.html">tilelang.intrinsics.utils</a></li>
311311
</ul>
312312
</li>
313+
<li class="toctree-l2"><a class="reference internal" href="../../ir/index.html">tilelang.ir</a></li>
313314
<li class="toctree-l2 has-children"><a class="reference internal" href="../../jit/index.html">tilelang.jit</a><input class="toctree-checkbox" id="toctree-checkbox-15" name="toctree-checkbox-15" role="switch" type="checkbox"/><label for="toctree-checkbox-15"><div class="visually-hidden">Toggle navigation of tilelang.jit</div><i class="icon"><svg><use href="#svg-arrow-right"></use></svg></i></label><ul>
314315
<li class="toctree-l3 has-children"><a class="reference internal" href="../../jit/adapter/index.html">tilelang.jit.adapter</a><input class="toctree-checkbox" id="toctree-checkbox-16" name="toctree-checkbox-16" role="switch" type="checkbox"/><label for="toctree-checkbox-16"><div class="visually-hidden">Toggle navigation of tilelang.jit.adapter</div><i class="icon"><svg><use href="#svg-arrow-right"></use></svg></i></label><ul>
315316
<li class="toctree-l4"><a class="reference internal" href="../../jit/adapter/base/index.html">tilelang.jit.adapter.base</a></li>

0 commit comments

Comments
 (0)