Skip to content

Commit 4c7c128

Browse files
Update docs
1 parent 7094971 commit 4c7c128

File tree

18 files changed

+33
-144
lines changed

18 files changed

+33
-144
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/quantize/lop3/index.rst.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -549,14 +549,14 @@ Module Contents
549549
T3 const scale_r = *(scale + scale_offset);
550550
uint const packed_scales_l = __pack_half2(scale_l, scale_l);
551551
uint const packed_scales_r = __pack_half2(scale_r, scale_r);
552-
552+
553553
const int num_elems_per_storage_dtype = sizeof(T1) * 8 / 4;
554554
555555
T1 const qzeros_l = *qzeros;
556556
T1 const qzeros_r = *(qzeros + qzeros_offset);
557557
int16_t const zero_l = (qzeros_l >> (group_offset * 4) & 0xf);
558558
int16_t const zero_r = (qzeros_r >> (group_offset * 4) & 0xf);
559-
559+
560560
uint median_num_l = ((0xe400 | zero_l) << 16) | (0xe400 | zero_l);
561561
uint median_num_r = ((0xe400 | zero_r) << 16) | (0xe400 | zero_r);
562562

_sources/autoapi/tilelang/quantize/mxfp/index.rst.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ Module Contents
4747
"and.b32 %0, %13, 0b10000001110000001000000111000000;"
4848
"mul.bf16x2 %0, %0, %12;"
4949
"shl.b32 %1, %13, 3;"
50-
"and.b32 %1, %1, 0b10000001110000001000000111000000;"
50+
"and.b32 %1, %1, 0b10000001110000001000000111000000;"
5151
"mul.bf16x2 %1, %1, %12;"
5252
"shl.b32 %2, %13, 6;"
5353
"and.b32 %2, %2, 0b10000001110000001000000111000000;"
@@ -71,7 +71,7 @@ Module Contents
7171
// Pay attention to the big-endianness issue
7272
B_local_decode[(i << 3) + j] = reinterpret_cast<T2*>(&B_dequantize_local_vec[j])[1];
7373
B_local_decode[(i << 3) + j + 4] = reinterpret_cast<T2*>(&B_dequantize_local_vec[j])[0];
74-
}
74+
}
7575
}
7676
// Check if the synchronization is needed
7777
}

_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

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

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -475,7 +475,7 @@ <h2>Classes<a class="headerlink" href="#classes" title="Link to this heading">¶
475475
<table class="autosummary longtable docutils align-default">
476476
<tbody>
477477
<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>
478-
<td><p>Base class template for hardware-aware configurations.</p></td>
478+
<td><p></p></td>
479479
</tr>
480480
</tbody>
481481
</table>
@@ -487,9 +487,6 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
487487
<dt class="sig sig-object py" id="tilelang.carver.template.flashattention.FlashAttentionTemplate">
488488
<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>
489489
<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>
490-
<p>Base class template for hardware-aware configurations.
491-
This serves as an abstract base class (ABC) that defines the structure
492-
for subclasses implementing hardware-specific optimizations.</p>
493490
<dl class="py attribute">
494491
<dt class="sig sig-object py" id="tilelang.carver.template.flashattention.FlashAttentionTemplate.batch_size">
495492
<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)