Update docs

Author: github-actions[bot]

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

@@ -20,6 +20,11 @@ Module Contents
    Bases: :py:obj:`tilelang.carver.template.base.BaseTemplate`
 
 
+   Base class template for hardware-aware configurations.
+   This serves as an abstract base class (ABC) that defines the structure
+   for subclasses implementing hardware-specific optimizations.
+
+
    .. py:attribute:: batch_size
       :type:  int
       :value: 1

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

@@ -20,6 +20,11 @@ Module Contents
    Bases: :py:obj:`tilelang.carver.template.base.BaseTemplate`
 
 
+   Base class template for hardware-aware configurations.
+   This serves as an abstract base class (ABC) that defines the structure
+   for subclasses implementing hardware-specific optimizations.
+
+
    .. py:attribute:: structure
       :type:  Union[str, List[str]]
       :value: None
@@ -40,6 +45,19 @@ Module Contents
 
    .. py:method:: get_hardware_aware_configs(arch = None, topk = 10)
 
+      Abstract method that must be implemented by subclasses.
+      It should return a list of hardware-aware configurations (hints)
+      based on the specified architecture.
+
+      :param arch: The target architecture. Defaults to None.
+      :type arch: TileDevice, optional
+      :param topk: Number of top configurations to return. Defaults to 10.
+      :type topk: int, optional
+
+      :returns: A list of recommended hardware-aware configurations.
+      :rtype: List[Hint]
+
+
 
    .. py:method:: initialize_function()
 

_sources/autoapi/tilelang/jit/adapter/dlpack/index.rst.txt

@@ -20,8 +20,12 @@ Classes
 Module Contents
 ---------------
 
-.. py:class:: TorchDLPackKernelAdapter
+.. py:class:: TorchDLPackKernelAdapter(mod, params, result_idx)
 
    Bases: :py:obj:`tilelang.jit.adapter.base.BaseKernelAdapter`
 
 
+   Helper class that provides a standard way to create an ABC using
+   inheritance.
+
+

_sources/autoapi/tilelang/jit/adapter/nvrtc/adapter/index.rst.txt

@@ -80,6 +80,8 @@ Module Contents
 
 
    .. py:attribute:: target
+      :value: None
+
 
 
    .. py:attribute:: verbose

_sources/autoapi/tilelang/layout/layout/index.rst.txt

@@ -25,6 +25,9 @@ Module Contents
    Bases: :py:obj:`tvm.ir.Node`
 
 
+   Base class of all IR Nodes.
+
+
    .. py:property:: index
 
       Property to retrieve the forward index of the layout.

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

@@ -466,7 +466,7 @@ <h2>Classes<a class="headerlink" href="#classes" title="Link to this heading">¶
 <table class="autosummary longtable docutils align-default">
 <tbody>
 <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>
-<td><p></p></td>
+<td><p>Base class template for hardware-aware configurations.</p></td>
 </tr>
 </tbody>
 </table>
@@ -478,6 +478,9 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
 <dt class="sig sig-object py" id="tilelang.carver.template.flashattention.FlashAttentionTemplate">
 <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>
 <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>
+<p>Base class template for hardware-aware configurations.
+This serves as an abstract base class (ABC) that defines the structure
+for subclasses implementing hardware-specific optimizations.</p>
 <dl class="py attribute">
 <dt class="sig sig-object py" id="tilelang.carver.template.flashattention.FlashAttentionTemplate.batch_size">
 <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/carver/template/general_reduce/index.html

@@ -466,7 +466,7 @@ <h2>Classes<a class="headerlink" href="#classes" title="Link to this heading">¶
 <table class="autosummary longtable docutils align-default">
 <tbody>
 <tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.carver.template.general_reduce.GeneralReductionTemplate" title="tilelang.carver.template.general_reduce.GeneralReductionTemplate"><code class="xref py py-obj docutils literal notranslate"><span class="pre">GeneralReductionTemplate</span></code></a></p></td>
-<td><p></p></td>
+<td><p>Base class template for hardware-aware configurations.</p></td>
 </tr>
 </tbody>
 </table>
@@ -478,6 +478,9 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
 <dt class="sig sig-object py" id="tilelang.carver.template.general_reduce.GeneralReductionTemplate">
 <em class="property"><span class="pre">class</span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">tilelang.carver.template.general_reduce.</span></span><span class="sig-name descname"><span class="pre">GeneralReductionTemplate</span></span><a class="headerlink" href="#tilelang.carver.template.general_reduce.GeneralReductionTemplate" title="Link to this definition">¶</a></dt>
 <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>
+<p>Base class template for hardware-aware configurations.
+This serves as an abstract base class (ABC) that defines the structure
+for subclasses implementing hardware-specific optimizations.</p>
 <dl class="py attribute">
 <dt class="sig sig-object py" id="tilelang.carver.template.general_reduce.GeneralReductionTemplate.structure">
 <span class="sig-name descname"><span class="pre">structure</span></span><em class="property"><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="pre">str</span><span class="w"> </span><span class="p"><span class="pre">|</span></span><span class="w"> </span><span class="pre">List</span><span class="p"><span class="pre">[</span></span><span class="pre">str</span><span class="p"><span class="pre">]</span></span></em><em class="property"><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="pre">None</span></em><a class="headerlink" href="#tilelang.carver.template.general_reduce.GeneralReductionTemplate.structure" title="Link to this definition">¶</a></dt>
@@ -496,15 +499,21 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
 <dl class="py method">
 <dt class="sig sig-object py" id="tilelang.carver.template.general_reduce.GeneralReductionTemplate.get_hardware_aware_configs">
 <span class="sig-name descname"><span class="pre">get_hardware_aware_configs</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">arch</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">topk</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">10</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.carver.template.general_reduce.GeneralReductionTemplate.get_hardware_aware_configs" title="Link to this definition">¶</a></dt>
-<dd><dl class="field-list simple">
+<dd><p>Abstract method that must be implemented by subclasses.
+It should return a list of hardware-aware configurations (hints)
+based on the specified architecture.</p>
+<dl class="field-list simple">
 <dt class="field-odd">Parameters<span class="colon">:</span></dt>
 <dd class="field-odd"><ul class="simple">
-<li><p><strong>arch</strong> (<em>tilelang.carver.arch.TileDevice</em>)</p></li>
-<li><p><strong>topk</strong> (<em>int</em>)</p></li>
+<li><p><strong>arch</strong> (<a class="reference internal" href="../../arch/arch_base/index.html#tilelang.carver.arch.arch_base.TileDevice" title="tilelang.carver.arch.arch_base.TileDevice"><em>TileDevice</em></a><em>, </em><em>optional</em>) – The target architecture. Defaults to None.</p></li>
+<li><p><strong>topk</strong> (<em>int</em><em>, </em><em>optional</em>) – Number of top configurations to return. Defaults to 10.</p></li>
 </ul>
 </dd>
-<dt class="field-even">Return type<span class="colon">:</span></dt>
-<dd class="field-even"><p>List[tilelang.carver.roller.Hint]</p>
+<dt class="field-even">Returns<span class="colon">:</span></dt>
+<dd class="field-even"><p>A list of recommended hardware-aware configurations.</p>
+</dd>
+<dt class="field-odd">Return type<span class="colon">:</span></dt>
+<dd class="field-odd"><p>List[<a class="reference internal" href="../../roller/hint/index.html#tilelang.carver.roller.hint.Hint" title="tilelang.carver.roller.hint.Hint">Hint</a>]</p>
 </dd>
 </dl>
 </dd></dl>

autoapi/tilelang/jit/adapter/dlpack/index.html

@@ -467,7 +467,7 @@ <h2>Classes<a class="headerlink" href="#classes" title="Link to this heading">¶
 <table class="autosummary longtable docutils align-default">
 <tbody>
 <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>
-<td><p></p></td>
+<td><p>Helper class that provides a standard way to create an ABC using</p></td>
 </tr>
 </tbody>
 </table>
@@ -477,8 +477,18 @@ <h2>Classes<a class="headerlink" href="#classes" title="Link to this heading">¶
 <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to this heading">¶</a></h2>
 <dl class="py class">
 <dt class="sig sig-object py" id="tilelang.jit.adapter.dlpack.TorchDLPackKernelAdapter">
-<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>
+<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>
 <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>
+<p>Helper class that provides a standard way to create an ABC using
+inheritance.</p>
+<dl class="field-list simple">
+<dt class="field-odd">Parameters<span class="colon">:</span></dt>
+<dd class="field-odd"><ul class="simple">
+<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>
+<li><p><strong>result_idx</strong> (<em>List</em><em>[</em><em>int</em><em>]</em>)</p></li>
+</ul>
+</dd>
+</dl>
 </dd></dl>
 
 </section>

autoapi/tilelang/jit/adapter/nvrtc/adapter/index.html

@@ -579,7 +579,7 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
 
 <dl class="py attribute">
 <dt class="sig sig-object py" id="tilelang.jit.adapter.nvrtc.adapter.NVRTCKernelAdapter.target">
-<span class="sig-name descname"><span class="pre">target</span></span><a class="headerlink" href="#tilelang.jit.adapter.nvrtc.adapter.NVRTCKernelAdapter.target" title="Link to this definition">¶</a></dt>
+<span class="sig-name descname"><span class="pre">target</span></span><em class="property"><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="pre">None</span></em><a class="headerlink" href="#tilelang.jit.adapter.nvrtc.adapter.NVRTCKernelAdapter.target" title="Link to this definition">¶</a></dt>
 <dd></dd></dl>
 
 <dl class="py attribute">

autoapi/tilelang/language/index.html

@@ -544,7 +544,7 @@ <h2>Package Contents<a class="headerlink" href="#package-contents" title="Link t
 <dd class="field-odd"><ul class="simple">
 <li><p><strong>panel_size</strong> (<em>int</em>)</p></li>
 <li><p><strong>order</strong> (<em>str</em>)</p></li>
-<li><p><strong>enable</strong> (<em>bool</em>)</p></li>
+<li><p><strong>enable</strong> (<em>tvm.script.parser.tir.bool</em>)</p></li>
 </ul>
 </dd>
 </dl>
@@ -654,7 +654,7 @@ <h2>Package Contents<a class="headerlink" href="#package-contents" title="Link t
 <span class="sig-prename descclassname"><span class="pre">tilelang.language.</span></span><span class="sig-name descname"><span class="pre">import_source</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">source</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.language.import_source" title="Link to this definition">¶</a></dt>
 <dd><dl class="field-list simple">
 <dt class="field-odd">Parameters<span class="colon">:</span></dt>
-<dd class="field-odd"><p><strong>source</strong> (<em>Optional</em><em>[</em><em>str</em><em>]</em>)</p>
+<dd class="field-odd"><p><strong>source</strong> (<em>tvm.script.parser.tir.Optional</em><em>[</em><em>str</em><em>]</em>)</p>
 </dd>
 </dl>
 </dd></dl>