Skip to content

Commit 6f36dbc

Browse files
Update docs
1 parent 227af60 commit 6f36dbc

File tree

15 files changed

+22
-161
lines changed

15 files changed

+22
-161
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/cuda/index.rst.txt

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

5050

51-
Represents the architecture of a computing device, capturing various hardware specifications.
52-
53-
5451
.. py:attribute:: target
5552
5653

_sources/autoapi/tilelang/carver/arch/metal/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/roller/policy/tensorcore/index.rst.txt

Lines changed: 1 addition & 56 deletions
Original file line numberDiff line numberDiff line change
@@ -30,15 +30,11 @@ Module Contents
3030

3131
.. py:data:: logger
3232
33-
.. py:class:: TensorCorePolicy(arch, tags = None)
33+
.. py:class:: TensorCorePolicy
3434
3535
Bases: :py:obj:`tilelang.carver.roller.policy.default.DefaultPolicy`
3636

3737

38-
Default Policy for fastdlight, a heuristic plan that tries to
39-
minimize memory traffic and maximize parallelism.for BitBLAS Schedule.
40-
41-
4238
.. py:attribute:: wmma_k
4339
:type: int
4440
:value: 16
@@ -65,67 +61,16 @@ Module Contents
6561

6662
.. py:method:: infer_node_smem_usage(td, node)
6763
68-
Infers the shared memory usage of a node given a TileDict configuration.
69-
70-
:param td: The TileDict object containing the tile configuration.
71-
:type td: TileDict
72-
:param node: The node for which to infer the shared memory usage.
73-
:type node: PrimFuncNode
74-
75-
:returns: The estimated amount of shared memory used by the node.
76-
:rtype: int
77-
78-
7964
8065
.. py:method:: get_node_reduce_step_candidates(node)
8166
82-
Calculates reduction step candidates for each reduction axis in a PrimFuncNode. General idea : use factor first, since it does not require extra boundary check. for large prime number, which is rare case, use power of 2.
83-
84-
:param node: The node for which to calculate reduction step candidates. It contains reduction axes (raxis)
85-
with their domains (dom.extent).
86-
:type node: PrimFuncNode
87-
88-
:returns: A dictionary mapping axis variable names to lists of step candidates. For each axis in the node,
89-
this function calculates possible step sizes. For axes with a large prime domain, it uses powers of 2
90-
as step candidates; for others, it uses all factors of the domain.
91-
:rtype: Dict[str, List[int]]
92-
93-
9467
9568
.. py:method:: check_tile_shape_isvalid(td)
9669
97-
Checks if the tile shapes in the TileDict are valid for the nodes in this context.
98-
99-
Parameters:
100-
- td (TileDict): The TileDict object containing tile shapes and other configurations.
101-
102-
Returns:
103-
- bool: True if all tile shapes are valid, False otherwise.
104-
105-
10670
10771
.. py:method:: compute_node_stride_map(node, td)
10872
109-
Computes the stride map for a given node based on the TileDict configuration.
110-
111-
:param node: The node for which to compute the stride map.
112-
:type node: PrimFuncNode
113-
:param td: The TileDict object containing the tile configuration.
114-
:type td: TileDict
115-
116-
:returns: A tuple of dictionaries containing the output strides and tensor strides.
117-
:rtype: Tuple[Dict, Dict]
118-
119-
12073
12174
.. py:method:: plan_rasterization(td)
12275
123-
Plans the rasterization for the given TileDict. This function is not implemented yet.
124-
125-
:param td: The TileDict object to plan rasterization for.
126-
:type td: TileDict
127-
128-
:raises RasterRationPlan: This function is not implemented yet.
129-
130-
13176

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

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -116,8 +116,6 @@ Module Contents
116116

117117

118118
.. py:attribute:: target
119-
:value: 'auto'
120-
121119
122120
123121
.. py:method:: from_database(func, kernel_global_source, kernel_lib_path, params, target, target_host, out_idx, execution_backend, pass_configs = None, compile_flags = None)

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

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -488,7 +488,7 @@ <h2>Classes<a class="headerlink" href="#classes" title="Link to this heading">¶
488488
<table class="autosummary longtable docutils align-default">
489489
<tbody>
490490
<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>
491-
<td><p>Represents the architecture of a computing device, capturing various hardware specifications.</p></td>
491+
<td><p></p></td>
492492
</tr>
493493
</tbody>
494494
</table>
@@ -525,7 +525,6 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
525525
<dt class="sig sig-object py" id="tilelang.carver.arch.cdna.CDNA">
526526
<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>
527527
<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>
528-
<p>Represents the architecture of a computing device, capturing various hardware specifications.</p>
529528
<dl class="field-list simple">
530529
<dt class="field-odd">Parameters<span class="colon">:</span></dt>
531530
<dd class="field-odd"><p><strong>target</strong> (<em>tvm.target.Target</em><em> | </em><em>str</em>)</p>

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

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -488,7 +488,7 @@ <h2>Classes<a class="headerlink" href="#classes" title="Link to this heading">¶
488488
<table class="autosummary longtable docutils align-default">
489489
<tbody>
490490
<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.carver.arch.cuda.CUDA" title="tilelang.carver.arch.cuda.CUDA"><code class="xref py py-obj docutils literal notranslate"><span class="pre">CUDA</span></code></a></p></td>
491-
<td><p>Represents the architecture of a computing device, capturing various hardware specifications.</p></td>
491+
<td><p></p></td>
492492
</tr>
493493
</tbody>
494494
</table>
@@ -625,7 +625,6 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
625625
<dt class="sig sig-object py" id="tilelang.carver.arch.cuda.CUDA">
626626
<em class="property"><span class="pre">class</span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">tilelang.carver.arch.cuda.</span></span><span class="sig-name descname"><span class="pre">CUDA</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.cuda.CUDA" title="Link to this definition"></a></dt>
627627
<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>
628-
<p>Represents the architecture of a computing device, capturing various hardware specifications.</p>
629628
<dl class="field-list simple">
630629
<dt class="field-odd">Parameters<span class="colon">:</span></dt>
631630
<dd class="field-odd"><p><strong>target</strong> (<em>tvm.target.Target</em><em> | </em><em>str</em>)</p>

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

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -488,7 +488,7 @@ <h2>Classes<a class="headerlink" href="#classes" title="Link to this heading">¶
488488
<table class="autosummary longtable docutils align-default">
489489
<tbody>
490490
<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.carver.arch.metal.METAL" title="tilelang.carver.arch.metal.METAL"><code class="xref py py-obj docutils literal notranslate"><span class="pre">METAL</span></code></a></p></td>
491-
<td><p>Represents the architecture of a computing device, capturing various hardware specifications.</p></td>
491+
<td><p></p></td>
492492
</tr>
493493
</tbody>
494494
</table>
@@ -525,7 +525,6 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
525525
<dt class="sig sig-object py" id="tilelang.carver.arch.metal.METAL">
526526
<em class="property"><span class="pre">class</span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">tilelang.carver.arch.metal.</span></span><span class="sig-name descname"><span class="pre">METAL</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.metal.METAL" title="Link to this definition"></a></dt>
527527
<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>
528-
<p>Represents the architecture of a computing device, capturing various hardware specifications.</p>
529528
<dl class="field-list simple">
530529
<dt class="field-odd">Parameters<span class="colon">:</span></dt>
531530
<dd class="field-odd"><p><strong>target</strong> (<em>tvm.target.Target</em><em> | </em><em>str</em>)</p>

0 commit comments

Comments
 (0)