Skip to content

Commit 6ed8cb4

Browse files
committed
Update docs
1 parent c8577df commit 6ed8cb4

File tree

7 files changed

+18
-20
lines changed

7 files changed

+18
-20
lines changed

_sources/autoapi/tilelang/carver/arch/cuda/index.rst.txt

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

9494

95+
Represents the architecture of a computing device, capturing various hardware specifications.
96+
97+
9598
.. py:attribute:: target
9699
97100

_sources/autoapi/tilelang/carver/template/flashattention/index.rst.txt

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,11 @@ 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+
2328
.. py:attribute:: batch_size
2429
:type: int
2530
: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-

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

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -489,7 +489,7 @@ <h2>Classes<a class="headerlink" href="#classes" title="Link to this heading">¶
489489
<td><p></p></td>
490490
</tr>
491491
<tr class="row-even"><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>
492-
<td><p></p></td>
492+
<td><p>Represents the architecture of a computing device, capturing various hardware specifications.</p></td>
493493
</tr>
494494
</tbody>
495495
</table>
@@ -686,6 +686,7 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
686686
<dt class="sig sig-object py" id="tilelang.carver.arch.cuda.CUDA">
687687
<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>
688688
<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>
689+
<p>Represents the architecture of a computing device, capturing various hardware specifications.</p>
689690
<dl class="field-list simple">
690691
<dt class="field-odd">Parameters<span class="colon">:</span></dt>
691692
<dd class="field-odd"><p><strong>target</strong> (<em>Union</em><em>[</em><em>tvm.target.Target</em><em>, </em><em>str</em><em>]</em>)</p>

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

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -465,7 +465,7 @@ <h2>Classes<a class="headerlink" href="#classes" title="Link to this heading">¶
465465
<table class="autosummary longtable docutils align-default">
466466
<tbody>
467467
<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>
468-
<td><p></p></td>
468+
<td><p>Base class template for hardware-aware configurations.</p></td>
469469
</tr>
470470
</tbody>
471471
</table>
@@ -477,6 +477,9 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
477477
<dt class="sig sig-object py" id="tilelang.carver.template.flashattention.FlashAttentionTemplate">
478478
<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>
479479
<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>
480+
<p>Base class template for hardware-aware configurations.
481+
This serves as an abstract base class (ABC) that defines the structure
482+
for subclasses implementing hardware-specific optimizations.</p>
480483
<dl class="py attribute">
481484
<dt class="sig sig-object py" id="tilelang.carver.template.flashattention.FlashAttentionTemplate.batch_size">
482485
<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/jit/adapter/dlpack/index.html

Lines changed: 2 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -466,7 +466,7 @@ <h2>Classes<a class="headerlink" href="#classes" title="Link to this heading">¶
466466
<table class="autosummary longtable docutils align-default">
467467
<tbody>
468468
<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.jit.adapter.dlpack.TorchDLPackKernelAdapter" title="tilelang.jit.adapter.dlpack.TorchDLPackKernelAdapter"><code class="xref py py-obj docutils literal notranslate"><span class="pre">TorchDLPackKernelAdapter</span></code></a></p></td>
469-
<td><p>Helper class that provides a standard way to create an ABC using</p></td>
469+
<td><p></p></td>
470470
</tr>
471471
</tbody>
472472
</table>
@@ -476,18 +476,8 @@ <h2>Classes<a class="headerlink" href="#classes" title="Link to this heading">¶
476476
<h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to this heading"></a></h2>
477477
<dl class="py class">
478478
<dt class="sig sig-object py" id="tilelang.jit.adapter.dlpack.TorchDLPackKernelAdapter">
479-
<em class="property"><span class="pre">class</span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">tilelang.jit.adapter.dlpack.</span></span><span class="sig-name descname"><span class="pre">TorchDLPackKernelAdapter</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">mod</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">params</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">result_idx</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.jit.adapter.dlpack.TorchDLPackKernelAdapter" title="Link to this definition"></a></dt>
479+
<em class="property"><span class="pre">class</span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">tilelang.jit.adapter.dlpack.</span></span><span class="sig-name descname"><span class="pre">TorchDLPackKernelAdapter</span></span><a class="headerlink" href="#tilelang.jit.adapter.dlpack.TorchDLPackKernelAdapter" title="Link to this definition"></a></dt>
480480
<dd><p>Bases: <a class="reference internal" href="../base/index.html#tilelang.jit.adapter.base.BaseKernelAdapter" title="tilelang.jit.adapter.base.BaseKernelAdapter"><code class="xref py py-obj docutils literal notranslate"><span class="pre">tilelang.jit.adapter.base.BaseKernelAdapter</span></code></a></p>
481-
<p>Helper class that provides a standard way to create an ABC using
482-
inheritance.</p>
483-
<dl class="field-list simple">
484-
<dt class="field-odd">Parameters<span class="colon">:</span></dt>
485-
<dd class="field-odd"><ul class="simple">
486-
<li><p><strong>params</strong> (<em>List</em><em>[</em><a class="reference internal" href="../../../engine/param/index.html#tilelang.engine.param.KernelParam" title="tilelang.engine.param.KernelParam"><em>tilelang.engine.param.KernelParam</em></a><em>]</em>)</p></li>
487-
<li><p><strong>result_idx</strong> (<em>List</em><em>[</em><em>int</em><em>]</em>)</p></li>
488-
</ul>
489-
</dd>
490-
</dl>
491481
</dd></dl>
492482

493483
</section>

searchindex.js

Lines changed: 1 addition & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)